Initial

intervention/models.py

@@ -0,0 +1,91 @@
+"""
+Author: Michel Peltriaux
+Organization: Struktur- und Genehmigungsdirektion Nord, Rhineland-Palatinate, Germany
+Contact: michel.peltriaux@sgdnord.rlp.de
+Created on: 17.11.20
+
+"""
+from django.contrib.auth.models import User
+from django.contrib.gis.db import models
+from django.utils.timezone import now
+
+from intervention.settings import INTERVENTION_IDENTIFIER_LENGTH, INTERVENTION_IDENTIFIER_TEMPLATE
+from konova.models import BaseObject
+from konova.utils.generators import generate_random_string
+from organisation.enums import RoleTypeEnum
+from process.models import Process
+
+
+class Intervention(BaseObject):
+    """
+    Interventions are e.g. construction sites where nature used to be.
+
+    A process consists of exactly one intervention and one or more compensation
+    """
+    type = models.CharField(max_length=500, null=True, blank=True)
+    law = models.CharField(max_length=500, null=True, blank=True)
+    handler = models.CharField(max_length=500, null=True, blank=True)
+    data_provider = models.ForeignKey("organisation.Organisation", on_delete=models.SET_NULL, null=True, blank=True)
+    data_provider_detail = models.CharField(max_length=500, null=True, blank=True)
+    geometry = models.MultiPolygonField(null=True, blank=True)
+    process = models.OneToOneField("process.Process", on_delete=models.CASCADE, null=True, blank=True, related_name="intervention")
+    documents = models.ManyToManyField("konova.Document", blank=True)
+
+    def __str__(self):
+        return "{} by {}".format(self.type, self.handler)
+
+    def delete(self, *args, **kwargs):
+        if self.process is not None:
+            self.process.delete()
+        super().delete(*args, **kwargs)
+
+    @staticmethod
+    def __generate_new_identifier() -> str:
+        """ Generates a new identifier for the intervention object
+
+        Returns:
+            str
+        """
+        curr_month = str(now().month)
+        curr_year = str(now().year)
+        rand_str = generate_random_string(
+            length=INTERVENTION_IDENTIFIER_LENGTH,
+            only_numbers=True,
+        )
+        _str = "{}{}{}".format(curr_month, curr_year, rand_str)
+        return INTERVENTION_IDENTIFIER_TEMPLATE.format(_str)
+
+    def save(self, *args, **kwargs):
+        if self.identifier is None or len(self.identifier) == 0:
+            # Create new identifier
+            new_id = self.__generate_new_identifier()
+            while Intervention.objects.filter(identifier=new_id).exists():
+                new_id = self.__generate_new_identifier()
+            self.identifier = new_id
+        super().save(*args, **kwargs)
+
+    @staticmethod
+    def get_role_objects(user: User, order_by: str = "-created_on") -> list:
+        """ Returns objects depending on the currently selected role of the user
+
+            * REGISTRATIONOFFICE
+                * User can see the processes where registration_office is set to the organisation of the currently selected role
+                * User can see self-created processes
+            * LICENSINGOFFICE
+                * same
+            * DATAPROVIDER
+                * User can see only self-created processes
+
+        Args:
+            user (User): The performing user
+            order_by (str): Order by which Process attribute
+
+        Returns:
+
+        """
+        role = user.current_role
+        if role is None:
+            return Intervention.objects.none()
+        processes = Process.get_role_objects(user, order_by)
+        interventions = [process.intervention for process in processes]
+        return interventions