from django.dispatch import receiver from django.db.models.signals import post_save, post_delete from .models import User, Corporation, Adhesion @receiver(post_save, sender=User, dispatch_uid='create_user_adhesion') def create_user_adhesion(sender, instance, created, **kwargs): if created and not hasattr(instance, 'skip_adhesion_creation'): Adhesion.objects.create(user=instance) @receiver(post_save, sender=Corporation, dispatch_uid='create_corporation_adhesion') def create_corporation_adhesion(sender, instance, created, **kwargs): if created and not hasattr(instance, 'skip_adhesion_creation'): Adhesion.objects.create(corporation=instance) @receiver(post_delete, sender=Adhesion, dispatch_uid='delete_adhesion') def delete_adhesion(sender, instance, **kwargs): if instance.is_physical(): instance.user.delete() if instance.is_moral(): instance.corporation.delete() instance.membership.delete()