From cee9fea8ec22dfc24b6db4c914e220a59fbb996b Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Sat, 14 Dec 2013 23:38:52 +0100 Subject: Script to fix bad permit type --- archaeological_files/management/__init__.py | 0 .../management/commands/__init__.py | 0 .../management/commands/fix_permittype.py | 32 ++++++++++++++++++++++ 3 files changed, 32 insertions(+) create mode 100644 archaeological_files/management/__init__.py create mode 100644 archaeological_files/management/commands/__init__.py create mode 100644 archaeological_files/management/commands/fix_permittype.py diff --git a/archaeological_files/management/__init__.py b/archaeological_files/management/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/archaeological_files/management/commands/__init__.py b/archaeological_files/management/commands/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/archaeological_files/management/commands/fix_permittype.py b/archaeological_files/management/commands/fix_permittype.py new file mode 100644 index 000000000..c7013266d --- /dev/null +++ b/archaeological_files/management/commands/fix_permittype.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +import csv +import datetime, time + +from django.conf import settings +from django.core.management.base import BaseCommand, CommandError + +from archaeological_files.models import File, PermitType + +replaced = (('cu', 'CU'), + ('pc', 'PC'), + ('', 'NP') + ) + + +class Command(BaseCommand): + help = "Update file types" + + def handle(self, *args, **options): + self.stdout.write("* Update file types\n") + for origin, dest in replaced: + permit_type = PermitType.objects.get(txt_idx=dest) + if origin: + File.objects.filter(permit_type__txt_idx=origin).update( + permit_type=permit_type) + PermitType.objects.filter(txt_idx=origin).update(available=False) + else: + File.objects.filter(permit_type__isnull=True).update( + permit_type=permit_type) + -- cgit v1.2.3 From c6f301b76291d2f416f6a641fe92b03cd10076a7 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Sat, 14 Dec 2013 23:51:21 +0100 Subject: Set Non Precised by default (if available) for permit types --- archaeological_files/forms.py | 3 ++- 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 ('', '--') -- cgit v1.2.3 From 370082be320249538600dfedf35779055a7ce3b4 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Sat, 14 Dec 2013 23:58:42 +0100 Subject: Fix permit types - 2 --- archaeological_files/management/commands/fix_permittype.py | 1 + 1 file changed, 1 insertion(+) diff --git a/archaeological_files/management/commands/fix_permittype.py b/archaeological_files/management/commands/fix_permittype.py index c7013266d..909952634 100644 --- a/archaeological_files/management/commands/fix_permittype.py +++ b/archaeological_files/management/commands/fix_permittype.py @@ -11,6 +11,7 @@ from archaeological_files.models import File, PermitType replaced = (('cu', 'CU'), ('pc', 'PC'), + ('ei', 'EI'), ('', 'NP') ) -- cgit v1.2.3