path: root/archaeological_files/models.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# Copyright (C) 2012 Étienne Loks  <etienne.loks_AT_peacefrogsDOTnet>

# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.

# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

# See the file COPYING for details.

import datetime

from django.conf import settings
from django.contrib.gis.db import models
from django.utils.translation import ugettext_lazy as _, ugettext

from ishtar_common.models import GeneralType, BaseHistorizedItem, \
    HistoricalRecords, OwnPerms, Person, Organization, Department, Town, \
    Dashboard

class FileType(GeneralType):
    class Meta:
        verbose_name = _(u"Archaeological file type")
        verbose_name_plural = _(u"Archaeological file types")

    @classmethod
    def is_preventive(cls, file_type_id, key=''):
        key = key or 'preventive'
        try:
            preventive = FileType.objects.get(txt_idx=key).pk
            return file_type_id == preventive
        except ObjectDoesNotExist:
            return False

class PermitType(GeneralType):
    class Meta:
        verbose_name = _(u"Permit type")
        verbose_name_plural = _(u"Permit types")

if settings.COUNTRY == 'fr':
    class SaisineType(GeneralType):
        delay = models.IntegerField(_(u"Delay (in days)"))
        class Meta:
            verbose_name = u"Type Saisine"
            verbose_name_plural = u"Types Saisine"

class File(BaseHistorizedItem, OwnPerms):
    TABLE_COLS = ['numeric_reference', 'year', 'internal_reference',
                  'file_type', 'saisine_type', 'towns', ]
    year = models.IntegerField(_(u"Year"),
                               default=lambda:datetime.datetime.now().year)
    numeric_reference = models.IntegerField(_(u"Numeric reference"))
    internal_reference = models.CharField(_(u"Internal reference"),
                                          max_length=60, unique=True)
    file_type = models.ForeignKey(FileType, verbose_name=_(u"File type"))
    in_charge = models.ForeignKey(Person, related_name='+',
                                  verbose_name=_(u"Person in charge"))
    general_contractor = models.ForeignKey(Person, related_name='+',
               verbose_name=_(u"General contractor"), blank=True, null=True)
    town_planning_service = models.ForeignKey(Organization, related_name='+',
               verbose_name=_(u"Town planning service"), blank=True, null=True)
    permit_type = models.ForeignKey(PermitType, verbose_name=_(u"Permit type"),
                                                          blank=True, null=True)
    permit_reference = models.CharField(_(u"Permit reference"),
                                          max_length=60, blank=True, null=True)
    end_date = models.DateField(_(u"Closing date"), null=True, blank=True)
    towns = models.ManyToManyField(Town, verbose_name=_(u"Towns"),
                                   related_name='file')
    creation_date = models.DateField(_(u"Creation date"),
                                     default=datetime.date.today)
    reception_date = models.DateField(_(u'Reception date'), blank=True,
                                      null=True)
    related_file = models.ForeignKey("File", verbose_name=_(u"Related file"),
                                     blank=True, null=True)
    if settings.COUNTRY == 'fr':
        saisine_type = models.ForeignKey(SaisineType, blank=True, null=True,
                                         verbose_name= u"Type de saisine")
        reference_number = models.IntegerField(_(u"Reference number"),
                                               blank=True, null=True)
    total_surface = models.IntegerField(_(u"Total surface (m²)"),
                                        blank=True, null=True)
    total_developed_surface = models.IntegerField(
            _(u"Total developed surface (m²)"), blank=True, null=True)
    address = models.TextField(_(u"Main address"), null=True, blank=True)
    address_complement = models.TextField(_(u"Main address - complement"),
                                          null=True, blank=True)
    postal_code = models.CharField(_(u"Main address - postal code"),
                                   max_length=10, null=True, blank=True)
    comment = models.TextField(_(u"Comment"), null=True, blank=True)
    history = HistoricalRecords()

    class Meta:
        verbose_name = _(u"Archaeological file")
        verbose_name_plural = _(u"Archaeological files")
        permissions = (
            ("view_own_file", ugettext(u"Can view own Archaelogical file")),
            ("add_own_file", ugettext(u"Can add own Archaelogical file")),
            ("change_own_file", ugettext(u"Can change own Archaelogical file")),
            ("delete_own_file", ugettext(u"Can delete own Archaelogical file")),
        )
        ordering = ['-year', '-numeric_reference']

    @classmethod
    def get_years(cls):
        return [res['year'] for res in list(cls.objects.values('year').annotate(
                                             Count("id")).order_by())]

    @classmethod
    def get_by_year(cls, year):
        return cls.objects.filter(year=year)

    @classmethod
    def get_total_number(cls):
        return cls.objects.count()

    def __unicode__(self):
        items = [unicode(_('Intercommunal'))]
        if self.towns.count() == 1:
            items[0] = unicode(self.towns.all()[0])
        items.append("-".join((unicode(self.year),
                               unicode(self.numeric_reference))))
        items += [unicode(getattr(self, k))[:36]
                  for k in ['internal_reference',] if getattr(self, k)]
        return JOINT.join(items)

    @classmethod
    def get_query_owns(cls, user):
        return Q(history_modifier=user) & Q(end_date__isnull=True)

    def is_active(self):
        return not bool(self.end_date)

    def closing(self):
        if self.is_active():
            return
        for item in self.history.all():
            if not item.end_date:
                break
        return {'date':item.history_date,
                'user':IshtarUser.objects.get(pk=item.history_modifier_id)}

    def total_surface_ha(self):
        if self.total_surface:
            return self.total_surface/10000.0

    def total_developed_surface_ha(self):
        if self.total_developed_surface:
            return self.total_developed_surface/10000.0

    def operation_acts(self):
        acts = []
        for ope in self.operations.all():
            for act in ope.administrative_act.all():
                acts.append(act)
        return acts

    def is_preventive(self):
        return FileType.is_preventive(self.file_type.pk)

class FileByDepartment(models.Model):
    '''
    Database view: don't forget to create it

    create view file_department (id, department_id, file_id) as
        select town."id", town."departement_id", file_towns."file_id"
            from ishtar_base_town town
        inner join ishtar_base_file_towns file_towns on
            file_towns."town_id"=town."id" order by town."departement_id";
    CREATE RULE file_department_delete
                            AS ON DELETE TO file_department DO INSTEAD();
    '''
    file = models.ForeignKey(File, verbose_name=_(u"File"))
    department = models.ForeignKey(Department, verbose_name=_(u"Department"),
                                   blank=True, null=True)
    class Meta:
        managed = False
        db_table = 'file_department'

class FileDashboard:
    def __init__(self):
        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')

        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')

        # research
        self.research = {}
        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['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]

        # rescue
        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()

        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', 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']:
            if idx > len(by_year_operationnal):
                break
            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')