User profile form: duplicate, delete and edit

1 files changed, 62 insertions, 10 deletions

diff --git a/ishtar_common/forms_common.py b/ishtar_common/forms_common.py
index af16e6deb..edea0cde2 100644
--- a/ishtar_common/forms_common.py
+++ b/ishtar_common/forms_common.py
@@ -771,6 +771,7 @@ class ProfileForm(ManageOldType):
     profile_type = forms.ChoiceField(label=_(u"Type"), choices=[],
                                      required=False)
     area = widgets.Select2MultipleField(label=_(u"Areas"), required=False)
+    name = forms.CharField(label=_(u"Name"), required=False)
     pk = forms.IntegerField(label=" ", widget=forms.HiddenInput, required=False)
 
     TYPES = [
@@ -798,37 +799,88 @@ class FinalAccountForm(forms.Form):
 
 
 class ProfilePersonForm(forms.Form):
+    """
+    Edit the current profile
+    """
     current_profile = forms.ChoiceField(label=_(u"Current profile"),
                                         choices=[])
+    name = forms.CharField(label=_(u"Name"), required=False)
+    profile_type = forms.ChoiceField(label=_(u"Profile type"), required=False,
+                                     disabled=True, choices=[])
+    duplicate_profile = forms.BooleanField(
+        label=_(u"Duplicate this profile"), required=False)
+    delete_profile = forms.BooleanField(
+        label=_(u"Delete this profile"), required=False,
+    )
 
     def __init__(self, *args, **kwargs):
         self.user = kwargs.pop('user')
         choices, initial = [], kwargs.get('initial', {})
-        for profile in self.user.ishtaruser.person.profiles.all():
+        current_profile = None
+        for profile in self.user.ishtaruser.person.profiles.order_by(
+                'name', 'profile_type__label').all():
             if profile.current:
+                current_profile = profile
                 initial['current_profile'] = profile.pk
             choices.append((profile.pk, unicode(profile)))
+        if current_profile:
+            initial['name'] = current_profile.name or \
+                              current_profile.profile_type
+            initial['profile_type'] = current_profile.profile_type.pk
         kwargs['initial'] = initial
         super(ProfilePersonForm, self).__init__(*args, **kwargs)
         self.fields['current_profile'].choices = choices
 
-    def save(self, session):
-        q = models.UserProfile.objects.filter(
-            person__ishtaruser=self.user.ishtaruser, current=True)
-        for profile in q.all():
-            profile.current = False
-            profile.save()
+        if not current_profile or \
+                not self.user.ishtaruser.person.profiles.filter(
+                    profile_type=current_profile.profile_type).exclude(
+                    pk=current_profile.pk).count():
+            # cannot delete the current profile if no profile of this type is
+            # available
+            self.fields.pop('delete_profile')
+
+        if not current_profile:
+            return
+        self.fields['profile_type'].choices = [
+            (current_profile.profile_type.pk, current_profile.profile_type.name)
+        ]
 
+    def save(self, session):
         q = models.UserProfile.objects.filter(
             person__ishtaruser=self.user.ishtaruser,
-            pk=int(self.cleaned_data['current_profile'])
-        )
+            pk=self.cleaned_data['current_profile'])
         if not q.count():
             return
         profile = q.all()[0]
+
+        # manage deletion
+        if self.cleaned_data.get('delete_profile', None):
+            q = self.user.ishtaruser.person.profiles.filter(
+                    profile_type=profile.profile_type).exclude(
+                    pk=profile.pk)
+            if not q.count():
+                # cannot delete the current profile if no profile of this type
+                # is available
+                return
+            new_current = q.all()[0]
+            new_current.current = True
+            new_current.save()
+            profile.delete()
+            return
+
+        name = self.cleaned_data['name']
+
+        # manage duplication
+        if self.cleaned_data.get('duplicate_profile', None):
+            profile_name = profile.name or profile.profile_type.label
+            if name == profile_name:
+                name += unicode(_(u" (duplicate)"))
+            profile.duplicate(name=name)
+            return
+
         profile.current = True
+        profile.name = name
         profile.save()
-
         clean_session_cache(session)