#31 API PUT/POST Ema

* adds support for PUT and POST of Ema
* moves set_responsibility() and set_legal() from Intervention API Serializer into proper Mixins where they belong to

api/utils/serializer/v1/ema.py

@@ -5,10 +5,15 @@ Contact: michel.peltriaux@sgdnord.rlp.de
 Created on: 24.01.22
 
 """
+from django.db import transaction
+
 from api.utils.serializer.v1.serializer import AbstractModelAPISerializerV1, AbstractCompensationAPISerializerV1Mixin, \
     ResponsibilityAPISerializerV1Mixin
 from ema.models import Ema
 from intervention.models import Responsibility
+from konova.models import Geometry
+from konova.tasks import celery_update_parcels
+from user.models import UserActionLogEntry
 
 
 class EmaAPISerializerV1(AbstractModelAPISerializerV1, AbstractCompensationAPISerializerV1Mixin, ResponsibilityAPISerializerV1Mixin):
@@ -27,3 +32,103 @@ class EmaAPISerializerV1(AbstractModelAPISerializerV1, AbstractCompensationAPISe
         self.properties_data["after_states"] = self.compensation_state_to_json(entry.after_states.all())
         self.properties_data["actions"] = self.compensation_actions_to_json(entry.actions.all())
         self.properties_data["deadlines"] = self.deadlines_to_json(entry.deadlines.all())
+
+    def initialize_objects(self, json_model, user):
+        """ Initializes all needed objects from the json_model data
+
+        Does not persist data to the DB!
+
+        Args:
+            json_model (dict): The json data
+            user (User): The API user
+
+        Returns:
+            obj (Compensation)
+        """
+        create_action = UserActionLogEntry.get_created_action(user, comment="API Import")
+        # Create geometry
+        json_geom = self.create_geometry_from_json(json_model)
+        geometry = Geometry()
+        geometry.geom = json_geom
+        geometry.created = create_action
+
+        # Create linked objects
+        obj = Ema()
+        obj.responsible = Responsibility()
+        created = create_action
+        obj.created = created
+        obj.geometry = geometry
+        return obj
+
+    def create_model_from_json(self, json_model, user):
+        """ Creates a new entry for the model based on the contents of json_model
+
+        Args:
+            json_model (dict): The json containing data
+            user (User): The API user
+
+        Returns:
+            created_id (str): The id of the newly created Ema entry
+        """
+        with transaction.atomic():
+            obj = self.initialize_objects(json_model, user)
+
+            # Fill in data to objects
+            properties = json_model["properties"]
+            obj.identifier = obj.generate_new_identifier()
+            obj.title = properties["title"]
+            obj = self.set_responsibility(obj, properties["responsible"])
+
+            obj.geometry.save()
+            obj.responsible.save()
+            obj.save()
+
+            obj = self.set_compensation_actions(obj, properties["actions"])
+            obj = self.set_compensation_states(obj, properties["before_states"], obj.before_states)
+            obj = self.set_compensation_states(obj, properties["after_states"], obj.after_states)
+            obj = self.set_deadlines(obj, properties["deadlines"])
+
+            obj.log.add(obj.created)
+            obj.users.add(user)
+
+            celery_update_parcels.delay(obj.geometry.id)
+
+            return obj.id
+
+    def update_model_from_json(self, id, json_model, user):
+        """ Updates an entry for the model based on the contents of json_model
+
+        Args:
+            id (str): The object's id
+            json_model (dict): The json containing data
+            user (User): The API user
+
+        Returns:
+            created_id (str): The id of the newly created Ema entry
+        """
+        with transaction.atomic():
+            update_action = UserActionLogEntry.get_edited_action(user, "API update")
+            obj = self.get_obj_from_db(id, user)
+
+            # Fill in data to objects
+            properties = json_model["properties"]
+            obj.title = properties["title"]
+            obj.modified = update_action
+            obj.geometry.geom = self.create_geometry_from_json(json_model)
+            obj.geometry.modified = update_action
+            obj = self.set_responsibility(obj, properties["responsible"])
+
+            obj.geometry.save()
+            obj.responsible.save()
+            obj.save()
+
+            obj = self.set_compensation_actions(obj, properties["actions"])
+            obj = self.set_compensation_states(obj, properties["before_states"], obj.before_states)
+            obj = self.set_compensation_states(obj, properties["after_states"], obj.after_states)
+            obj = self.set_deadlines(obj, properties["deadlines"])
+
+            obj.log.add(update_action)
+
+            celery_update_parcels.delay(obj.geometry.id)
+
+            return obj.id