models.py 4.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106
  1. from django.db import models
  2. from django.urls import reverse
  3. from django.contrib.auth.models import User
  4. from django.contrib.contenttypes.fields import GenericForeignKey, GenericRelation
  5. from django.contrib.contenttypes.models import ContentType
  6. from djadhere.utils import get_active_filter
  7. from banking.models import Payment
  8. # Terminologie : une « adhésion » désgine une instance de ce modèle
  9. # tandis qu’un « adhérent » désigne un user ou une corporation.
  10. class Adhesion(models.Model):
  11. limit = models.Q(app_label='auth', model='user') \
  12. | models.Q(app_label='adhesions', model='corporation')
  13. id = models.AutoField(verbose_name='Numéro d’adhérent', primary_key=True, editable=True)
  14. adherent_type = models.ForeignKey(ContentType, on_delete=models.CASCADE,
  15. limit_choices_to=limit, verbose_name='Type d’adhérent')
  16. adherent_id = models.PositiveIntegerField(verbose_name='ID')
  17. adherent = GenericForeignKey('adherent_type', 'adherent_id')
  18. contributions = GenericRelation(Payment,
  19. content_type_field='reason_type',
  20. object_id_field='reason_id',
  21. related_query_name='adhesion')
  22. created = models.DateTimeField(null=True, blank=True, auto_now_add=True)
  23. class Meta:
  24. verbose_name = 'adhésion'
  25. unique_together = ("adherent_type", "adherent_id")
  26. ordering = ('id',)
  27. @property
  28. def contribution(self):
  29. try:
  30. return self.contributions.exclude(period=0).filter(get_active_filter()).get()
  31. except Payment.DoesNotExist:
  32. return None
  33. # MultipleObjectsReturned non catché volontairement, le filtrage par la méthode clean est censé
  34. # empêcher cette possibilité, si cette exception est levé on veut recevoir un mail avec l’erreur !
  35. def is_physical(self):
  36. return self.adherent_type.app_label == 'auth' and self.adherent_type.model == 'user'
  37. def is_moral(self):
  38. return self.adherent_type.app_label == 'adhesions' and self.adherent_type.model == 'corporation'
  39. @property
  40. def type(self):
  41. if self.is_physical():
  42. return 'Personne physique'
  43. else:
  44. return 'Personne morale'
  45. def get_adherent_name(self):
  46. if self.is_physical():
  47. return str(self.adherent.profile)
  48. else:
  49. return str(self.adherent)
  50. get_adherent_name.short_description = 'Nom ou raison sociale'
  51. def __str__(self):
  52. return 'ADT%d' % self.id
  53. if self.id is None:
  54. return '?'
  55. else:
  56. return 'ADT%d (%s)' % (self.id, self.get_adherent_name())
  57. class ProxyUser(User):
  58. class Meta:
  59. proxy = True
  60. verbose_name = 'personne physique'
  61. verbose_name_plural = 'personnes physiques'
  62. def get_model_perms(self, request):
  63. return {}
  64. def __str__(self):
  65. return str(self.profile)
  66. class Corporation(models.Model):
  67. social_reason = models.CharField(max_length=256, verbose_name='Raison sociale', unique=True)
  68. description = models.TextField(blank=True, default='')
  69. members = models.ManyToManyField(User, blank=True, verbose_name='Membres',
  70. related_name='corporations', related_query_name='corporation')
  71. email = models.EmailField(verbose_name='Adresse e-mail', blank=True)
  72. phone_number = models.CharField(max_length=16, blank=True, default='',
  73. verbose_name='Numéro de téléphone')
  74. address = models.TextField(blank=True, default='', verbose_name='Adresse')
  75. notes = models.TextField(blank=True, default='')
  76. @property
  77. def adhesion(self):
  78. ctype = ContentType.objects.get_for_model(self)
  79. try:
  80. return Adhesion.objects.get(adherent_type=ctype, adherent_id=self.pk)
  81. except Adhesion.DoesNotExist:
  82. return None
  83. class Meta:
  84. verbose_name = 'personne morale'
  85. verbose_name_plural = 'personnes morales'
  86. def __str__(self):
  87. return self.social_reason