diff options
Diffstat (limited to 'archaeological_files/models.py')
| -rw-r--r-- | archaeological_files/models.py | 948 |
1 files changed, 544 insertions, 404 deletions
diff --git a/archaeological_files/models.py b/archaeological_files/models.py index b897911eb..88326d405 100644 --- a/archaeological_files/models.py +++ b/archaeological_files/models.py @@ -1,4 +1,4 @@ -#!/usr/bin/env python +#!/usr/bin/env python3 # -*- coding: utf-8 -*- # Copyright (C) 2012-2017 Étienne Loks <etienne.loks_AT_peacefrogsDOTnet> @@ -30,30 +30,53 @@ from django.db.models.signals import post_save, m2m_changed, post_delete from django.core.urlresolvers import reverse from ishtar_common.utils import ugettext_lazy as _, pgettext_lazy -from ishtar_common.utils import cached_label_changed, get_cache, \ - get_current_year, m2m_historization_changed - -from ishtar_common.models import GeneralType, BaseHistorizedItem, \ - OwnPerms, Person, Organization, Town, \ - Dashboard, DashboardFormItem, ValueGetter, ShortMenuItem, \ - OperationType, get_generated_id, post_save_cache, Document, HistoryModel, \ - SearchAltName, SearchVectorConfig, DocumentItem, CompleteIdentifierItem +from ishtar_common.utils import ( + cached_label_changed, + get_cache, + get_current_year, + m2m_historization_changed, +) + +from ishtar_common.models import ( + GeneralType, + BaseHistorizedItem, + OwnPerms, + Person, + Organization, + Town, + Dashboard, + DashboardFormItem, + ValueGetter, + ShortMenuItem, + OperationType, + get_generated_id, + post_save_cache, + Document, + HistoryModel, + SearchAltName, + SearchVectorConfig, + DocumentItem, + CompleteIdentifierItem, +) from ishtar_common.models_common import HistoricalRecords, Department -from archaeological_operations.models import get_values_town_related, \ - ClosedItem, ParcelItem +from archaeological_operations.models import ( + get_values_town_related, + ClosedItem, + ParcelItem, +) class FileType(GeneralType): class Meta: verbose_name = _("Archaeological file type") verbose_name_plural = _("Archaeological file types") - ordering = ('label',) + ordering = ("label",) @classmethod - def is_preventive(cls, file_type_id, key=''): - key = key or 'preventive' + def is_preventive(cls, file_type_id, key=""): + key = key or "preventive" try: preventive = FileType.get_cache(key).pk return file_type_id == preventive @@ -69,314 +92,375 @@ class PermitType(GeneralType): class Meta: verbose_name = _("Permit type") verbose_name_plural = _("Permit types") - ordering = ('label',) + ordering = ("label",) post_save.connect(post_save_cache, sender=PermitType) post_delete.connect(post_save_cache, sender=PermitType) -if settings.COUNTRY == 'fr': +if settings.COUNTRY == "fr": + class SaisineType(GeneralType, ValueGetter): delay = models.IntegerField(_("Delay (in days)"), default=30) class Meta: verbose_name = "Type de saisine" verbose_name_plural = "Types de saisine" - ordering = ('label',) + ordering = ("label",) + post_save.connect(post_save_cache, sender=SaisineType) post_delete.connect(post_save_cache, sender=SaisineType) -class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, - OwnPerms, ValueGetter, ShortMenuItem, DashboardFormItem, ParcelItem): - SLUG = 'file' - SHOW_URL = 'show-file' - DELETE_URL = 'delete-file' +class File( + ClosedItem, + DocumentItem, + BaseHistorizedItem, + CompleteIdentifierItem, + OwnPerms, + ValueGetter, + ShortMenuItem, + DashboardFormItem, + ParcelItem, +): + SLUG = "file" + SHOW_URL = "show-file" + DELETE_URL = "delete-file" APP = "archaeological-files" MODEL = "file" - TABLE_COLS = ['numeric_reference', 'year', 'internal_reference', - 'file_type', 'saisine_type', 'towns_label', ] + TABLE_COLS = [ + "numeric_reference", + "year", + "internal_reference", + "file_type", + "saisine_type", + "towns_label", + ] # statistics - STATISTIC_MODALITIES_OPTIONS = OrderedDict([ - ('year', _("Year")), - ("file_type__label", _("File type")), - ("towns__areas__label", _("Area")), - ("towns__areas__parent__label", _("Extended area")), - ("saisine_type__label", "Type de saisine"), - ("permit_type__label", _("Permit type")), - ("requested_operation_type__label", _("File type")), - ]) - STATISTIC_MODALITIES = [ - key for key, lbl in STATISTIC_MODALITIES_OPTIONS.items()] + STATISTIC_MODALITIES_OPTIONS = OrderedDict( + [ + ("year", _("Year")), + ("file_type__label", _("File type")), + ("towns__areas__label", _("Area")), + ("towns__areas__parent__label", _("Extended area")), + ("saisine_type__label", "Type de saisine"), + ("permit_type__label", _("Permit type")), + ("requested_operation_type__label", _("File type")), + ] + ) + STATISTIC_MODALITIES = [key for key, lbl in STATISTIC_MODALITIES_OPTIONS.items()] STATISTIC_SUM_VARIABLE = OrderedDict( ( ("pk", (_("Number"), 1)), ("total_surface", (_("Total surface (km2)"), 0.000001)), - ("total_developed_surface", ( - _("Total developed surface (km2)"), 0.000001)), + ("total_developed_surface", (_("Total developed surface (km2)"), 0.000001)), ) ) # search parameters - BOOL_FIELDS = ['end_date__isnull'] + BOOL_FIELDS = ["end_date__isnull"] EXTRA_REQUEST_KEYS = { - 'parcel_0': ('parcels__section', - 'operations__parcels__section'), - 'parcel_1': ('parcels__parcel_number', - 'operations__parcels__parcel_number'), - 'parcel_2': ('operations__parcels__public_domain', - 'parcels__public_domain'), - 'end_date': 'end_date__isnull', - 'towns__numero_insee__startswith': - 'towns__numero_insee__startswith', - 'name': 'name__icontains', - 'cached_label': 'cached_label__icontains', - 'comment': 'comment__icontains', - 'permit_reference': 'permit_reference__icontains', - 'general_contractor__attached_to': - 'general_contractor__attached_to__pk', - 'history_creator': 'history_creator__ishtaruser__person__pk', - 'history_modifier': 'history_modifier__ishtaruser__person__pk', - 'towns_label': 'towns', - 'general_contractor__pk': 'general_contractor__pk', - 'responsible_town_planning_service__pk': - 'responsible_town_planning_service__pk', - 'in_charge__pk': 'in_charge__pk', + "parcel_0": ("parcels__section", "operations__parcels__section"), + "parcel_1": ("parcels__parcel_number", "operations__parcels__parcel_number"), + "parcel_2": ("operations__parcels__public_domain", "parcels__public_domain"), + "end_date": "end_date__isnull", + "towns__numero_insee__startswith": "towns__numero_insee__startswith", + "name": "name__icontains", + "cached_label": "cached_label__icontains", + "comment": "comment__icontains", + "permit_reference": "permit_reference__icontains", + "general_contractor__attached_to": "general_contractor__attached_to__pk", + "history_creator": "history_creator__ishtaruser__person__pk", + "history_modifier": "history_modifier__ishtaruser__person__pk", + "towns_label": "towns", + "general_contractor__pk": "general_contractor__pk", + "responsible_town_planning_service__pk": "responsible_town_planning_service__pk", + "in_charge__pk": "in_charge__pk", } BASE_SEARCH_VECTORS = [ - SearchVectorConfig('name'), - SearchVectorConfig('internal_reference'), - SearchVectorConfig('file_type__label'), - SearchVectorConfig('saisine_type__label'), - SearchVectorConfig('permit_type__label'), - SearchVectorConfig('permit_reference'), - SearchVectorConfig('comment', 'local'), - SearchVectorConfig('research_comment', 'local') + SearchVectorConfig("name"), + SearchVectorConfig("internal_reference"), + SearchVectorConfig("file_type__label"), + SearchVectorConfig("saisine_type__label"), + SearchVectorConfig("permit_type__label"), + SearchVectorConfig("permit_reference"), + SearchVectorConfig("comment", "local"), + SearchVectorConfig("research_comment", "local"), ] - INT_SEARCH_VECTORS = [SearchVectorConfig('numeric_reference'), - SearchVectorConfig('year')] - M2M_SEARCH_VECTORS = [SearchVectorConfig('towns__name')] + INT_SEARCH_VECTORS = [ + SearchVectorConfig("numeric_reference"), + SearchVectorConfig("year"), + ] + M2M_SEARCH_VECTORS = [SearchVectorConfig("towns__name")] PARENT_SEARCH_VECTORS = [ - 'in_charge', 'general_contractor', 'corporation_general_contractor', - 'responsible_town_planning_service', 'planning_service', 'organization', - 'scientist' + "in_charge", + "general_contractor", + "corporation_general_contractor", + "responsible_town_planning_service", + "planning_service", + "organization", + "scientist", ] COL_LABELS = { - 'towns_label': _("Towns"), + "towns_label": _("Towns"), } REVERSED_BOOL_FIELDS = [ - 'documents__image__isnull', - 'documents__associated_file__isnull', - 'documents__associated_url__isnull', + "documents__image__isnull", + "documents__associated_file__isnull", + "documents__associated_url__isnull", ] # alternative names of fields for searches ALT_NAMES = { - 'year': SearchAltName( - pgettext_lazy("key for text search", "year"), - 'year' - ), - 'numeric_reference': SearchAltName( - pgettext_lazy("key for text search", "reference"), - 'numeric_reference' + "year": SearchAltName(pgettext_lazy("key for text search", "year"), "year"), + "numeric_reference": SearchAltName( + pgettext_lazy("key for text search", "reference"), "numeric_reference" ), - 'internal_reference': SearchAltName( + "internal_reference": SearchAltName( pgettext_lazy("key for text search", "other-reference"), - 'internal_reference__iexact' + "internal_reference__iexact", ), - 'towns': SearchAltName( - pgettext_lazy("key for text search", "town"), - 'towns__cached_label__iexact' + "towns": SearchAltName( + pgettext_lazy("key for text search", "town"), "towns__cached_label__iexact" ), - 'parcel': SearchAltName( + "parcel": SearchAltName( pgettext_lazy("key for text search", "parcel"), - 'parcels__cached_label__iexact' + "parcels__cached_label__iexact", ), - 'towns__numero_insee__startswith': SearchAltName( + "towns__numero_insee__startswith": SearchAltName( pgettext_lazy("key for text search", "department"), - 'towns__numero_insee__startswith' + "towns__numero_insee__startswith", ), - 'name': SearchAltName( - pgettext_lazy("key for text search", "name"), - 'name__iexact' + "name": SearchAltName( + pgettext_lazy("key for text search", "name"), "name__iexact" ), - 'file_type': SearchAltName( - pgettext_lazy("key for text search", "type"), - 'file_type__label__iexact' + "file_type": SearchAltName( + pgettext_lazy("key for text search", "type"), "file_type__label__iexact" ), - 'end_date': SearchAltName( - pgettext_lazy("key for text search", "active"), - 'end_date__isnull' + "end_date": SearchAltName( + pgettext_lazy("key for text search", "active"), "end_date__isnull" ), - 'saisine_type': SearchAltName( + "saisine_type": SearchAltName( pgettext_lazy("key for text search", "saisine-type"), - 'saisine_type__label__iexact' + "saisine_type__label__iexact", ), - 'permit_type': SearchAltName( + "permit_type": SearchAltName( pgettext_lazy("key for text search", "permit-type"), - 'permit_type__label__iexact' + "permit_type__label__iexact", ), - 'permit_reference': SearchAltName( + "permit_reference": SearchAltName( pgettext_lazy("key for text search", "permit-reference"), - 'permit_reference__iexact' + "permit_reference__iexact", ), - 'comment': SearchAltName( - pgettext_lazy("key for text search", "comment"), - 'comment__iexact' + "comment": SearchAltName( + pgettext_lazy("key for text search", "comment"), "comment__iexact" ), - 'in_charge': SearchAltName( + "in_charge": SearchAltName( pgettext_lazy("key for text search", "in-charge"), - 'in_charge__cached_label__iexact' + "in_charge__cached_label__iexact", ), - 'general_contractor': SearchAltName( + "general_contractor": SearchAltName( pgettext_lazy("key for text search", "general-contractor"), - 'general_contractor__cached_label__iexact' + "general_contractor__cached_label__iexact", ), - 'general_contractor__attached_to': SearchAltName( - pgettext_lazy("key for text search", - "general-contractor-organization"), - 'general_contractor__attached_to__cached_label__iexact' + "general_contractor__attached_to": SearchAltName( + pgettext_lazy("key for text search", "general-contractor-organization"), + "general_contractor__attached_to__cached_label__iexact", ), } ALT_NAMES.update(BaseHistorizedItem.ALT_NAMES) ALT_NAMES.update(DocumentItem.ALT_NAMES) POST_PROCESS_REQUEST = { - 'towns__numero_insee__startswith': '_get_department_code', + "towns__numero_insee__startswith": "_get_department_code", } - HISTORICAL_M2M = ['towns', 'departments'] + HISTORICAL_M2M = ["towns", "departments"] # fields year = models.IntegerField(_("Year"), default=get_current_year) numeric_reference = models.IntegerField( - _("Numeric reference"), blank=True, null=True) - internal_reference = models.CharField(_("Internal reference"), blank=True, - null=True, max_length=60) - external_id = models.CharField(_("External ID"), blank=True, null=True, - max_length=120) + _("Numeric reference"), blank=True, null=True + ) + internal_reference = models.CharField( + _("Internal reference"), blank=True, null=True, max_length=60 + ) + external_id = models.CharField( + _("External ID"), blank=True, null=True, max_length=120 + ) auto_external_id = models.BooleanField( - _("External ID is set automatically"), default=False) + _("External ID is set automatically"), default=False + ) name = models.TextField(_("Name"), blank=True, default="") file_type = models.ForeignKey(FileType, verbose_name=_("File type")) - in_charge = models.ForeignKey(Person, related_name='file_responsability', - verbose_name=_("Person in charge"), - on_delete=models.SET_NULL, - blank=True, null=True) + in_charge = models.ForeignKey( + Person, + related_name="file_responsability", + verbose_name=_("Person in charge"), + on_delete=models.SET_NULL, + blank=True, + null=True, + ) general_contractor = models.ForeignKey( - Person, related_name='general_contractor_files', - verbose_name=_("General contractor"), blank=True, null=True, - on_delete=models.SET_NULL,) # aménageur - personne - raw_general_contractor = models.CharField(_("General contractor (raw)"), - max_length=200, blank=True, null=True) + Person, + related_name="general_contractor_files", + verbose_name=_("General contractor"), + blank=True, + null=True, + on_delete=models.SET_NULL, + ) # aménageur - personne + raw_general_contractor = models.CharField( + _("General contractor (raw)"), max_length=200, blank=True, null=True + ) corporation_general_contractor = models.ForeignKey( Organization, - related_name='general_contractor_files', - verbose_name=_("General contractor organization"), blank=True, - null=True, on_delete=models.SET_NULL,) # aménageur + related_name="general_contractor_files", + verbose_name=_("General contractor organization"), + blank=True, + null=True, + on_delete=models.SET_NULL, + ) # aménageur responsible_town_planning_service = models.ForeignKey( - Person, related_name='responsible_town_planning_service_files', - blank=True, null=True, + Person, + related_name="responsible_town_planning_service_files", + blank=True, + null=True, verbose_name=_("Responsible for planning service"), - on_delete=models.SET_NULL,) # service instructeur - personne + on_delete=models.SET_NULL, + ) # service instructeur - personne raw_town_planning_service = models.CharField( - _("Planning service (raw)"), max_length=200, - blank=True, null=True) + _("Planning service (raw)"), max_length=200, blank=True, null=True + ) planning_service = models.ForeignKey( Organization, - related_name='planning_service_files', - blank=True, null=True, + related_name="planning_service_files", + blank=True, + null=True, verbose_name=_("Planning service organization"), - on_delete=models.SET_NULL,) # service instructeur + on_delete=models.SET_NULL, + ) # service instructeur permit_type = models.ForeignKey( - PermitType, verbose_name=_("Permit type"), blank=True, null=True, - on_delete=models.SET_NULL + PermitType, + verbose_name=_("Permit type"), + blank=True, + null=True, + on_delete=models.SET_NULL, ) - permit_reference = models.TextField( - _("Permit reference"), blank=True, default="") + permit_reference = models.TextField(_("Permit reference"), blank=True, default="") end_date = models.DateField(_("Closing date"), null=True, blank=True) main_town = models.ForeignKey( - Town, verbose_name=_("Main town"), null=True, blank=True, - related_name='file_main', - on_delete=models.SET_NULL + Town, + verbose_name=_("Main town"), + null=True, + blank=True, + related_name="file_main", + on_delete=models.SET_NULL, + ) + towns = models.ManyToManyField( + Town, verbose_name=_("Towns"), related_name="file", blank=True ) - towns = models.ManyToManyField(Town, verbose_name=_("Towns"), - related_name='file', blank=True) creation_date = models.DateField( - _("Creation date"), default=datetime.date.today, blank=True, - null=True) - reception_date = models.DateField(_('Reception date'), blank=True, - null=True) + _("Creation date"), default=datetime.date.today, blank=True, null=True + ) + reception_date = models.DateField(_("Reception date"), blank=True, null=True) planning_service_date = models.DateField( - _("Date of planning service file"), null=True, blank=True) + _("Date of planning service file"), null=True, blank=True + ) related_file = models.ForeignKey( - "File", verbose_name=_("Related file"), blank=True, null=True, - on_delete=models.SET_NULL + "File", + verbose_name=_("Related file"), + blank=True, + null=True, + on_delete=models.SET_NULL, ) - if settings.COUNTRY == 'fr': + if settings.COUNTRY == "fr": saisine_type = models.ForeignKey( - SaisineType, blank=True, null=True, + SaisineType, + blank=True, + null=True, on_delete=models.SET_NULL, - verbose_name="Type de saisine") - instruction_deadline = models.DateField(_('Instruction deadline'), - blank=True, null=True) - total_surface = models.FloatField(_("Total surface (m2)"), - blank=True, null=True) + verbose_name="Type de saisine", + ) + instruction_deadline = models.DateField( + _("Instruction deadline"), blank=True, null=True + ) + total_surface = models.FloatField(_("Total surface (m2)"), blank=True, null=True) total_developed_surface = models.FloatField( - _("Total developed surface (m2)"), blank=True, null=True) - locality = models.CharField(_("Locality"), - max_length=100, null=True, blank=True) + _("Total developed surface (m2)"), blank=True, null=True + ) + locality = models.CharField(_("Locality"), max_length=100, null=True, blank=True) address = models.TextField(_("Main address"), blank=True, default="") - postal_code = models.CharField(_("Main address - postal code"), - max_length=10, null=True, blank=True) + postal_code = models.CharField( + _("Main address - postal code"), max_length=10, null=True, blank=True + ) comment = models.TextField(_("Comment"), blank=True, default="") # research archaeology --> departments = models.ManyToManyField( - Department, verbose_name=_("Departments"), blank=True) + Department, verbose_name=_("Departments"), blank=True + ) requested_operation_type = models.ForeignKey( - OperationType, related_name='+', + OperationType, + related_name="+", on_delete=models.SET_NULL, - null=True, blank=True, verbose_name=_("Requested operation type")) + null=True, + blank=True, + verbose_name=_("Requested operation type"), + ) organization = models.ForeignKey( - Organization, blank=True, null=True, verbose_name=_("Organization"), - related_name='files', on_delete=models.SET_NULL) + Organization, + blank=True, + null=True, + verbose_name=_("Organization"), + related_name="files", + on_delete=models.SET_NULL, + ) scientist = models.ForeignKey( - Person, blank=True, null=True, related_name='scientist', - on_delete=models.SET_NULL, verbose_name=_("Scientist in charge")) + Person, + blank=True, + null=True, + related_name="scientist", + on_delete=models.SET_NULL, + verbose_name=_("Scientist in charge"), + ) research_comment = models.TextField( - _("Research archaeology comment"), blank=True, default="") + _("Research archaeology comment"), blank=True, default="" + ) classified_area = models.NullBooleanField( - _("Classified area"), blank=True, null=True) - protected_area = models.NullBooleanField( - _("Protected area"), blank=True, null=True) - if settings.COUNTRY == 'fr': - cira_advised = models.NullBooleanField( - "Passage en CIRA", blank=True, null=True) + _("Classified area"), blank=True, null=True + ) + protected_area = models.NullBooleanField(_("Protected area"), blank=True, null=True) + if settings.COUNTRY == "fr": + cira_advised = models.NullBooleanField("Passage en CIRA", blank=True, null=True) mh_register = models.NullBooleanField( - "Sur Monument Historique classé", blank=True, null=True) + "Sur Monument Historique classé", blank=True, null=True + ) mh_listing = models.NullBooleanField( - "Sur Monument Historique inscrit", blank=True, null=True) + "Sur Monument Historique inscrit", blank=True, null=True + ) # <-- research archaeology documents = models.ManyToManyField( - Document, related_name="files", verbose_name=_("Documents"), - blank=True) + Document, related_name="files", verbose_name=_("Documents"), blank=True + ) cached_label = models.TextField( - _("Cached name"), blank=True, default="", db_index=True, + _("Cached name"), + blank=True, + default="", + db_index=True, help_text=_("Generated automatically - do not edit"), ) imported_line = models.TextField(_("Imported line"), blank=True, default="") history = HistoricalRecords(bases=[HistoryModel]) GET_VALUES_EXTRA = ValueGetter.GET_VALUES_EXTRA + [ - 'general_contractor_address_1', - 'general_contractor_address_2', - 'general_contractor_address_3', - 'get_locality', + "general_contractor_address_1", + "general_contractor_address_2", + "general_contractor_address_3", + "get_locality", ] class Meta: @@ -390,9 +474,9 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, ("delete_own_file", "Can delete own Archaeological file"), ("close_file", "Can close File"), ) - ordering = ('cached_label',) + ordering = ("cached_label",) indexes = [ - GinIndex(fields=['data']), + GinIndex(fields=["data"]), ] @classmethod @@ -413,33 +497,29 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, @property def full_internal_ref(self): - return "{}{}".format(settings.ISHTAR_FILE_PREFIX or '', - self.external_id or '') + return "{}{}".format(settings.ISHTAR_FILE_PREFIX or "", self.external_id or "") @property def delay_date(self): - cache_key, val = get_cache(self.__class__, [self.pk, 'delay_date']) + cache_key, val = get_cache(self.__class__, [self.pk, "delay_date"]) if val: return val return self.update_delay_date(cache_key) def update_delay_date(self, cache_key=None): if not cache_key: - cache_key, val = get_cache(self.__class__, - [self.pk, 'delay_date']) + cache_key, val = get_cache(self.__class__, [self.pk, "delay_date"]) date = self.reception_date if not date: date = datetime.date(2500, 1, 1) - elif settings.COUNTRY == 'fr' and self.saisine_type \ - and self.saisine_type.delay: + elif settings.COUNTRY == "fr" and self.saisine_type and self.saisine_type.delay: date += datetime.timedelta(days=self.saisine_type.delay) cache.set(cache_key, date, settings.CACHE_TIMEOUT) return date @property def has_adminact(self): - cache_key, val = get_cache(self.__class__, [self.pk, - 'has_adminact']) + cache_key, val = get_cache(self.__class__, [self.pk, "has_adminact"]) if val: return val return self.update_has_admin_act(cache_key) @@ -447,54 +527,71 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, @property def get_locality(self): return " - ".join( - [getattr(self, k) for k in ('locality', 'address') - if getattr(self, k)]) + [getattr(self, k) for k in ("locality", "address") if getattr(self, k)] + ) @property def general_contractor_address_1(self): - address = '' + address = "" if self.general_contractor: if self.general_contractor.name: - address = " ".join([ - str(getattr(self.general_contractor, key)) - for key in ('title', 'surname', 'name') - if getattr(self.general_contractor, key)]) + address = " ".join( + [ + str(getattr(self.general_contractor, key)) + for key in ("title", "surname", "name") + if getattr(self.general_contractor, key) + ] + ) elif self.general_contractor.raw_name: address = self.general_contractor.raw_name - if not address and self.corporation_general_contractor and\ - self.corporation_general_contractor.name: + if ( + not address + and self.corporation_general_contractor + and self.corporation_general_contractor.name + ): address = self.corporation_general_contractor.name return address @property def general_contractor_address_2(self): - address = '' + address = "" if self.general_contractor and self.general_contractor.address: address = self.general_contractor.address if self.general_contractor.address_complement: address += " " + self.general_contractor.address_complement - if not address and self.corporation_general_contractor and\ - self.corporation_general_contractor.address: + if ( + not address + and self.corporation_general_contractor + and self.corporation_general_contractor.address + ): address = self.corporation_general_contractor.address if self.corporation_general_contractor.address_complement: - address += " " + \ - self.corporation_general_contractor.address_complement + address += " " + self.corporation_general_contractor.address_complement return address @property def general_contractor_address_3(self): - address = '' + address = "" if self.general_contractor and self.general_contractor.postal_code: - address = " ".join([ - getattr(self.general_contractor, key) - for key in ('postal_code', 'town') - if getattr(self.general_contractor, key)]) - if not address and self.corporation_general_contractor and\ - self.corporation_general_contractor.address: - address = " ".join([ - getattr(self.corporation_general_contractor, key) - for key in ('postal_code', 'town') - if getattr(self.corporation_general_contractor, key)]) + address = " ".join( + [ + getattr(self.general_contractor, key) + for key in ("postal_code", "town") + if getattr(self.general_contractor, key) + ] + ) + if ( + not address + and self.corporation_general_contractor + and self.corporation_general_contractor.address + ): + address = " ".join( + [ + getattr(self.corporation_general_contractor, key) + for key in ("postal_code", "town") + if getattr(self.corporation_general_contractor, key) + ] + ) return address @classmethod @@ -503,9 +600,9 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, similar = set() for parcel in parcels: q = cls.objects.filter( - parcels__town__pk=parcel['town'], - parcels__section=parcel['section'], - parcels__parcel_number=parcel['parcel_number'] + parcels__town__pk=parcel["town"], + parcels__section=parcel["section"], + parcels__parcel_number=parcel["parcel_number"], ) if q.count(): for fle in q.all(): @@ -514,53 +611,55 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, def update_has_admin_act(self, cache_key=None): if not cache_key: - cache_key, val = get_cache(self.__class__, [self.pk, - 'has_adminact']) - has_adminact = self.administrative_act.exclude( - act_type__txt_idx='a_receipt').count() or self.operations.count() + cache_key, val = get_cache(self.__class__, [self.pk, "has_adminact"]) + has_adminact = ( + self.administrative_act.exclude(act_type__txt_idx="a_receipt").count() + or self.operations.count() + ) cache.set(cache_key, has_adminact, settings.CACHE_TIMEOUT) return has_adminact @classmethod def get_short_menu_class(cls, pk): - cache_key, val = get_cache(cls, [pk, 'short_class_name']) + cache_key, val = get_cache(cls, [pk, "short_class_name"]) if val: return val q = cls.objects.filter(pk=pk) if not q.count(): - return '' + return "" item = q.all()[0] return item.update_short_menu_class(cache_key) def update_short_menu_class(self, cache_key=None): if not cache_key: - cache_key, val = get_cache(self.__class__, [self.pk, - 'short_class_name']) - cls = 'normal' - if not self.file_type.txt_idx == 'preventive': + cache_key, val = get_cache(self.__class__, [self.pk, "short_class_name"]) + cls = "normal" + if not self.file_type.txt_idx == "preventive": cls = "blue" elif not self.has_adminact and self.reception_date: delta = datetime.date.today() - self.reception_date - cls = 'red' + cls = "red" if self.saisine_type and self.saisine_type.delay: if delta.days <= (self.saisine_type.delay * 2 / 3): - cls = 'green' + cls = "green" elif delta.days <= self.saisine_type.delay: - cls = 'orange' + cls = "orange" cache.set(cache_key, cls, settings.CACHE_TIMEOUT) return cls @classmethod - def get_owns(cls, user, menu_filtr=None, limit=None, values=None, - get_short_menu_class=False): + def get_owns( + cls, user, menu_filtr=None, limit=None, values=None, get_short_menu_class=False + ): owns = super(File, cls).get_owns( - user, limit=limit, values=values, - get_short_menu_class=get_short_menu_class) + user, limit=limit, values=values, get_short_menu_class=get_short_menu_class + ) return cls._return_get_owns(owns, values, get_short_menu_class) - def get_values(self, prefix='', no_values=False, filtr=None, **kwargs): + def get_values(self, prefix="", no_values=False, filtr=None, **kwargs): values = super(File, self).get_values( - prefix=prefix, no_values=no_values, filtr=filtr, **kwargs) + prefix=prefix, no_values=no_values, filtr=filtr, **kwargs + ) return get_values_town_related(self, prefix, values, filtr=filtr) def render_parcels(self): @@ -580,16 +679,20 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, def _generate_cached_label(self): items = [self.get_town_label(), self.reference] - items += [str(getattr(self, k)) - for k in ['internal_reference', 'name'] if getattr(self, k)] + items += [ + str(getattr(self, k)) + for k in ["internal_reference", "name"] + if getattr(self, k) + ] return settings.JOINT.join(items) def grouped_parcels(self): from archaeological_operations.models import Parcel + return Parcel.grouped_parcels(list(self.parcels.all())) def get_town_label(self): - lbl = str(_('Multi-town')) + lbl = str(_("Multi-town")) if self.main_town: lbl = self.main_town.name elif self.towns.count() == 1: @@ -600,7 +703,7 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, def get_department(self): if not self.towns.count(): - return '00' + return "00" return self.towns.all()[0].numero_insee[:2] @classmethod @@ -608,24 +711,19 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, profile = ishtaruser.current_profile town_ids = [] if profile: - town_ids = [town['pk'] - for town in profile.query_towns.values('pk').all()] + town_ids = [town["pk"] for town in profile.query_towns.values("pk").all()] return [ { - 'in_charge': ishtaruser.person, - 'history_creator': ishtaruser.user_ptr, - 'towns__pk__in': town_ids + "in_charge": ishtaruser.person, + "history_creator": ishtaruser.user_ptr, + "towns__pk__in": town_ids, }, - { - 'end_date__isnull': True - } + {"end_date__isnull": True}, ] @classmethod def get_query_owns(cls, ishtaruser): - return cls._construct_query_own( - '', cls._get_query_owns_dicts(ishtaruser) - ) + return cls._construct_query_own("", cls._get_query_owns_dicts(ishtaruser)) def is_active(self): return not bool(self.end_date) @@ -650,9 +748,10 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, return acts def update_raw_town_planning_service(self): - if (self.raw_town_planning_service and not - self.responsible_town_planning_service) or \ - not self.responsible_town_planning_service: + if ( + self.raw_town_planning_service + and not self.responsible_town_planning_service + ) or not self.responsible_town_planning_service: return False current_lbl = "" if self.raw_town_planning_service: @@ -664,28 +763,30 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, return current_lbl != self.raw_town_planning_service def update_planning_service(self): - if not self.responsible_town_planning_service or \ - not self.responsible_town_planning_service.attached_to or \ - self.planning_service: + if ( + not self.responsible_town_planning_service + or not self.responsible_town_planning_service.attached_to + or self.planning_service + ): return False - self.planning_service = \ - self.responsible_town_planning_service.attached_to + self.planning_service = self.responsible_town_planning_service.attached_to return True def update_resp_planning_service(self): - if not self.responsible_town_planning_service or \ - self.responsible_town_planning_service.attached_to or \ - not self.planning_service: + if ( + not self.responsible_town_planning_service + or self.responsible_town_planning_service.attached_to + or not self.planning_service + ): return False - self.responsible_town_planning_service.attached_to = \ - self.planning_service + self.responsible_town_planning_service.attached_to = self.planning_service self.responsible_town_planning_service.save() return True def update_raw_general_contractor(self): - if (self.raw_general_contractor and not - self.general_contractor) or \ - not self.general_contractor: + if ( + self.raw_general_contractor and not self.general_contractor + ) or not self.general_contractor: return False current_lbl = "" if self.raw_general_contractor: @@ -697,45 +798,53 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, return current_lbl != self.raw_general_contractor def update_corpo_general_contractor(self): - if not self.general_contractor or \ - self.general_contractor.attached_to \ - == self.corporation_general_contractor: + if ( + not self.general_contractor + or self.general_contractor.attached_to + == self.corporation_general_contractor + ): return False if self.general_contractor.attached_to: - self.corporation_general_contractor = \ - self.general_contractor.attached_to + self.corporation_general_contractor = self.general_contractor.attached_to else: - self.general_contractor.attached_to = \ - self.corporation_general_contractor + self.general_contractor.attached_to = self.corporation_general_contractor self.general_contractor.save() return True def get_extra_actions(self, request): # url, base_text, icon, extra_text, extra css class, is a quick action actions = super(File, self).get_extra_actions(request) - if self.can_do(request, 'add_administrativeact'): + if self.can_do(request, "add_administrativeact"): actions += [ - (reverse('file-add-adminact', args=[self.pk]), - _("Add associated administrative act"), "fa fa-plus", - _("admin. act"), "", False), + ( + reverse("file-add-adminact", args=[self.pk]), + _("Add associated administrative act"), + "fa fa-plus", + _("admin. act"), + "", + False, + ), ] - if self.can_do(request, 'add_operation'): + if self.can_do(request, "add_operation"): actions += [ ( - reverse('file-add-operation', args=[self.pk]), + reverse("file-add-operation", args=[self.pk]), _("Add operation"), "fa fa-plus", _("operation"), "", - False + False, ) ] return actions def save(self, *args, **kwargs): returned = super(File, self).save(*args, **kwargs) - if not getattr(self, '_no_new_add', None) and self.main_town and \ - self.main_town not in list(self.towns.all()): + if ( + not getattr(self, "_no_new_add", None) + and self.main_town + and self.main_town not in list(self.towns.all()) + ): self._no_new_add = True self.towns.add(self.main_town) updated = self.update_raw_town_planning_service() @@ -745,7 +854,7 @@ class File(ClosedItem, DocumentItem, BaseHistorizedItem, CompleteIdentifierItem, updated += self.update_corpo_general_contractor() if not self.external_id or self.auto_external_id: - external_id = get_generated_id('file_external_id', self) + external_id = get_generated_id("file_external_id", self) if external_id != self.external_id: updated = True self.auto_external_id = True @@ -768,14 +877,14 @@ m2m_changed.connect(cached_label_changed, sender=File.towns.through) post_save.connect(cached_label_changed, sender=File) for attr in File.HISTORICAL_M2M: - m2m_changed.connect(m2m_historization_changed, - sender=getattr(File, attr).through) + m2m_changed.connect(m2m_historization_changed, sender=getattr(File, attr).through) class FileByDepartment(models.Model): """ Database view for dashboard """ + CREATE_SQL = """ CREATE VIEW file_department (id, department_id, file_id) as select town."id", town."departement_id", file_towns."file_id" @@ -790,13 +899,17 @@ class FileByDepartment(models.Model): DROP VIEW IF EXISTS file_department; """ file = models.ForeignKey(File, verbose_name=_("File")) - department = models.ForeignKey(Department, verbose_name=_("Department"), - on_delete=models.DO_NOTHING, - blank=True, null=True) + department = models.ForeignKey( + Department, + verbose_name=_("Department"), + on_delete=models.DO_NOTHING, + blank=True, + null=True, + ) class Meta: managed = False - db_table = 'file_department' + db_table = "file_department" def __str__(self): return "{} - {}".format(self.file, self.department) @@ -805,119 +918,146 @@ class FileByDepartment(models.Model): class FileDashboard: def __init__(self): from archaeological_operations.models import AdministrativeAct + main_dashboard = Dashboard(File) self.total_number = main_dashboard.total_number - types = File.objects.values('file_type', 'file_type__label') - self.types = types.annotate(number=Count('pk')).order_by('file_type') + types = File.objects.values("file_type", "file_type__label") + self.types = types.annotate(number=Count("pk")).order_by("file_type") - by_year = File.objects.extra( - {'date': "date_trunc('year', creation_date)"}) - self.by_year = by_year.values('date')\ - .annotate(number=Count('pk')).order_by('-date') + by_year = File.objects.extra({"date": "date_trunc('year', creation_date)"}) + self.by_year = ( + by_year.values("date").annotate(number=Count("pk")).order_by("-date") + ) now = datetime.date.today() limit = datetime.date(now.year, now.month, 1) - datetime.timedelta(365) by_month = File.objects.filter(creation_date__gt=limit).extra( - {'date': "date_trunc('month', creation_date)"}) - self.by_month = by_month.values('date')\ - .annotate(number=Count('pk')).order_by('-date') + {"date": "date_trunc('month', creation_date)"} + ) + self.by_month = ( + by_month.values("date").annotate(number=Count("pk")).order_by("-date") + ) # research self.research = {} - prog_type = FileType.objects.get(txt_idx='prog') + prog_type = FileType.objects.get(txt_idx="prog") researchs = File.objects.filter(file_type=prog_type) - self.research['total_number'] = researchs.count() - by_year = researchs.extra( - {'date': "date_trunc('year', creation_date)"} + self.research["total_number"] = researchs.count() + by_year = researchs.extra({"date": "date_trunc('year', creation_date)"}) + self.research["by_year"] = ( + by_year.values("date").annotate(number=Count("pk")).order_by("-date") + ) + by_month = researchs.filter(creation_date__gt=limit).extra( + {"date": "date_trunc('month', creation_date)"} + ) + self.research["by_month"] = ( + by_month.values("date").annotate(number=Count("pk")).order_by("-date") + ) + + self.research["by_dpt"] = ( + FileByDepartment.objects.filter( + file__file_type=prog_type, department__isnull=False + ) + .values("department__label") + .annotate(number=Count("file")) + .order_by("department__label") ) - self.research['by_year'] = by_year.values('date')\ - .annotate(number=Count('pk'))\ - .order_by('-date') - by_month = researchs.filter(creation_date__gt=limit)\ - .extra({'date': "date_trunc('month', creation_date)"}) - self.research['by_month'] = by_month.values('date')\ - .annotate(number=Count('pk'))\ - .order_by('-date') - - self.research['by_dpt'] = FileByDepartment.objects\ - .filter(file__file_type=prog_type, department__isnull=False)\ - .values('department__label')\ - .annotate(number=Count('file'))\ - .order_by('department__label') FileTown = File.towns.through - self.research['towns'] = FileTown.objects\ - .filter(file__file_type=prog_type)\ - .values('town__name')\ - .annotate(number=Count('file'))\ - .order_by('-number', 'town__name')[:10] + self.research["towns"] = ( + FileTown.objects.filter(file__file_type=prog_type) + .values("town__name") + .annotate(number=Count("file")) + .order_by("-number", "town__name")[:10] + ) # rescue - rescue_type = FileType.objects.get(txt_idx='preventive') + rescue_type = FileType.objects.get(txt_idx="preventive") rescues = File.objects.filter(file_type=rescue_type) self.rescue = {} - self.rescue['total_number'] = rescues.count() - self.rescue['saisine'] = rescues.values('saisine_type__label')\ - .annotate(number=Count('pk')).order_by('saisine_type__label') - self.rescue['administrative_act'] = AdministrativeAct.objects\ - .filter(associated_file__isnull=False)\ - .values('act_type__label')\ - .annotate(number=Count('pk'))\ - .order_by('act_type__pk') - - by_year = rescues.extra({'date': "date_trunc('year', creation_date)"}) - self.rescue['by_year'] = by_year.values('date')\ - .annotate(number=Count('pk')).order_by('-date') - by_month = rescues.filter(creation_date__gt=limit)\ - .extra({'date': "date_trunc('month', creation_date)"}) - self.rescue['by_month'] = by_month.values('date')\ - .annotate(number=Count('pk'))\ - .order_by('-date') - - self.rescue['by_dpt'] = FileByDepartment.objects\ - .filter(file__file_type=rescue_type, department__isnull=False)\ - .values('department__label')\ - .annotate(number=Count('file'))\ - .order_by('department__label') - self.rescue['towns'] = FileTown.objects\ - .filter(file__file_type=rescue_type)\ - .values('town__name')\ - .annotate(number=Count('file'))\ - .order_by('-number', 'town__name')[:10] - - self.rescue['with_associated_operation'] = rescues\ - .filter(operations__isnull=False).count() - - if self.rescue['total_number']: - self.rescue['with_associated_operation_percent'] = round( - float(self.rescue['with_associated_operation']) - / self.rescue['total_number'] * 100, 2) - - by_year_operationnal = rescues.filter(operations__isnull=False)\ - .extra({'date': 'date_trunc(\'year\', ' - '"archaeological_files_file".creation_date)'}) - by_year_operationnal = by_year_operationnal.values('date')\ - .annotate(number=Count('pk')).order_by('-date') + self.rescue["total_number"] = rescues.count() + self.rescue["saisine"] = ( + rescues.values("saisine_type__label") + .annotate(number=Count("pk")) + .order_by("saisine_type__label") + ) + self.rescue["administrative_act"] = ( + AdministrativeAct.objects.filter(associated_file__isnull=False) + .values("act_type__label") + .annotate(number=Count("pk")) + .order_by("act_type__pk") + ) + + by_year = rescues.extra({"date": "date_trunc('year', creation_date)"}) + self.rescue["by_year"] = ( + by_year.values("date").annotate(number=Count("pk")).order_by("-date") + ) + by_month = rescues.filter(creation_date__gt=limit).extra( + {"date": "date_trunc('month', creation_date)"} + ) + self.rescue["by_month"] = ( + by_month.values("date").annotate(number=Count("pk")).order_by("-date") + ) + + self.rescue["by_dpt"] = ( + FileByDepartment.objects.filter( + file__file_type=rescue_type, department__isnull=False + ) + .values("department__label") + .annotate(number=Count("file")) + .order_by("department__label") + ) + self.rescue["towns"] = ( + FileTown.objects.filter(file__file_type=rescue_type) + .values("town__name") + .annotate(number=Count("file")) + .order_by("-number", "town__name")[:10] + ) + + self.rescue["with_associated_operation"] = rescues.filter( + operations__isnull=False + ).count() + + if self.rescue["total_number"]: + self.rescue["with_associated_operation_percent"] = round( + float(self.rescue["with_associated_operation"]) + / self.rescue["total_number"] + * 100, + 2, + ) + + by_year_operationnal = rescues.filter(operations__isnull=False).extra( + {"date": "date_trunc('year', " '"archaeological_files_file".creation_date)'} + ) + by_year_operationnal = ( + by_year_operationnal.values("date") + .annotate(number=Count("pk")) + .order_by("-date") + ) percents, idx = [], 0 - for dct in self.rescue['by_year']: + for dct in self.rescue["by_year"]: if idx >= len(by_year_operationnal): break - if by_year_operationnal[idx]['date'] != dct['date'] or\ - not dct['number']: + if by_year_operationnal[idx]["date"] != dct["date"] or not dct["number"]: continue - val = round(float(by_year_operationnal[idx]['number']) / - dct['number'] * 100, 2) - percents.append({'date': dct['date'], 'number': val}) - self.rescue['operational_by_year'] = percents - - self.rescue['surface_by_town'] = FileTown.objects\ - .filter(file__file_type=rescue_type)\ - .values('town__name')\ - .annotate(number=Sum('file__total_surface'))\ - .order_by('-number', 'town__name')[:10] - self.rescue['surface_by_dpt'] = FileByDepartment.objects\ - .filter(file__file_type=rescue_type, department__isnull=False)\ - .values('department__label')\ - .annotate(number=Sum('file__total_surface'))\ - .order_by('department__label') + val = round( + float(by_year_operationnal[idx]["number"]) / dct["number"] * 100, 2 + ) + percents.append({"date": dct["date"], "number": val}) + self.rescue["operational_by_year"] = percents + + self.rescue["surface_by_town"] = ( + FileTown.objects.filter(file__file_type=rescue_type) + .values("town__name") + .annotate(number=Sum("file__total_surface")) + .order_by("-number", "town__name")[:10] + ) + self.rescue["surface_by_dpt"] = ( + FileByDepartment.objects.filter( + file__file_type=rescue_type, department__isnull=False + ) + .values("department__label") + .annotate(number=Sum("file__total_surface")) + .order_by("department__label") + ) |