from django.db.models.signals import post_save, post_delete
from django.dispatch import receiver


@receiver(post_save, sender=DiarioOficial)
def update_document(sender, instance, **kwargs):
    """Atualizar documento no Elasticsearch quando o objeto for salvo"""
    DiarioOficialDocument.update_document(instance)


@receiver(post_delete, sender=DiarioOficial)
def delete_document(sender, instance, **kwargs):
    """Deletar documento do Elasticsearch quando o objeto for deletado"""
    document = DiarioOficialDocument.get(id=instance.id)
    document.delete()