Add an explicit profile for user (refs #4046)

1 files changed, 44 insertions, 0 deletions

diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index b92614e08..7c2069309 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -2470,6 +2470,24 @@ class Person(Address, Merge, OwnPerms, ValueGetter):
             [unicode(getattr(self, attr))
              for attr in ['title', 'salutation'] if getattr(self, attr)])
 
+    @property
+    def default_profile(self):
+        q = self.profiles.filter(default=True)
+        if q.count():
+            return q.all()[0]
+        q = self.profiles
+        nb = q.count()
+        if not nb:
+            return
+        if nb > 1:
+            # arbitrary return the first one
+            return q.all()[0]
+        # if there is only one it is the default
+        profile = q.all()[0]
+        profile.default = True
+        profile.save()
+        return profile
+
     def simple_lbl(self):
         values = [unicode(getattr(self, attr)) for attr in ('surname', 'name')
                   if getattr(self, attr)]
@@ -2630,6 +2648,32 @@ class Person(Address, Merge, OwnPerms, ValueGetter):
               =user.ishtaruser)
 
 
+class ProfileType(GeneralType):
+    groups = models.ManyToManyField(Group, verbose_name=_(u"Groups"),
+                                    blank=True)
+
+    class Meta:
+        verbose_name = _(u"Profile type")
+        verbose_name_plural = _(u"Profile types")
+        ordering = ('label',)
+
+
+post_save.connect(post_save_cache, sender=ProfileType)
+post_delete.connect(post_save_cache, sender=ProfileType)
+
+
+class UserProfile(models.Model):
+    profile_type = models.ForeignKey(
+        ProfileType, verbose_name=_(u"Profile type"))
+    default = models.BooleanField(_(u"Default profile"), default=False)
+    person = models.ForeignKey(
+        Person, verbose_name=_(u"Person"), related_name='profiles')
+
+    class Meta:
+        verbose_name = _(u"User profile")
+        verbose_name_plural = _(u"User profiles")
+
+
 class IshtarUser(FullSearch):
     TABLE_COLS = ('username', 'person__name', 'person__surname',
                   'person__email', 'person__person_types_list',