diff options
author | Étienne Loks <etienne.loks@peacefrogs.net> | 2013-12-14 23:51:21 +0100 |
---|---|---|
committer | Étienne Loks <etienne.loks@peacefrogs.net> | 2013-12-14 23:51:21 +0100 |
commit | 9b3ede3e6d5d2b076975c188454a39c2b45d155d (patch) | |
tree | 859f6860e524944946fe5b7a23245c82c4fa8953 | |
parent | 494a8df3727831241c7b1e9993d1ef55d255befb (diff) | |
download | Ishtar-9b3ede3e6d5d2b076975c188454a39c2b45d155d.tar.bz2 Ishtar-9b3ede3e6d5d2b076975c188454a39c2b45d155d.zip |
Set Non Precised by default (if available) for permit types
-rw-r--r-- | archaeological_files/forms.py | 3 | ||||
-rw-r--r-- | ishtar_common/models.py | 22 |
2 files changed, 18 insertions, 7 deletions
diff --git a/archaeological_files/forms.py b/archaeological_files/forms.py index 9affc0901..ec9409ef6 100644 --- a/archaeological_files/forms.py +++ b/archaeological_files/forms.py @@ -182,7 +182,8 @@ class FileFormPreventive(forms.Form): super(FileFormPreventive, self).__init__(*args, **kwargs) self.fields['saisine_type'].choices = models.SaisineType.get_types() self.fields['saisine_type'].help_text = models.SaisineType.get_help() - self.fields['permit_type'].choices = models.PermitType.get_types() + self.fields['permit_type'].choices = models.PermitType.get_types( + default='NP') self.fields['permit_type'].help_text = models.PermitType.get_help() class FinalFileClosingForm(FinalForm): diff --git a/ishtar_common/models.py b/ishtar_common/models.py index 8d153f120..aacc9f2d0 100644 --- a/ishtar_common/models.py +++ b/ishtar_common/models.py @@ -230,20 +230,30 @@ class GeneralType(models.Model): return u"" @classmethod - def get_types(cls, dct={}, instances=False, exclude=[], empty_first=True): + def get_types(cls, dct={}, instances=False, exclude=[], empty_first=True, + default=None): base_dct = dct.copy() if hasattr(cls, 'parent'): return cls._get_parent_types(base_dct, instances, exclude=exclude, - empty_first=empty_first) + empty_first=empty_first, default=default) return cls._get_types(base_dct, instances, exclude=exclude, - empty_first=empty_first) + empty_first=empty_first, default=default) @classmethod - def _get_types(cls, dct={}, instances=False, exclude=[], empty_first=True): + def _get_types(cls, dct={}, instances=False, exclude=[], empty_first=True, + default=None): dct['available'] = True - if not instances and empty_first: + if not instances and empty_first and not default: yield ('', '--') + if default: + try: + default = cls.objects.get(txt_idx=default) + yield(default.pk, _(unicode(default))) + except cls.DoesNotExist: + pass items = cls.objects.filter(**dct) + if default: + exclude.append(default.txt_idx) if exclude: items = items.exclude(txt_idx__in=exclude) for item in items.order_by(*cls._meta.ordering).all(): @@ -277,7 +287,7 @@ class GeneralType(models.Model): @classmethod def _get_parent_types(cls, dct={}, instances=False, exclude=[], - empty_first=True): + empty_first=True, default=None): dct['available'] = True if not instances and empty_first: yield ('', '--') |