konova/api/views/v1/intervention.py

"""
Author: Michel Peltriaux
Organization: Struktur- und Genehmigungsdirektion Nord, Rhineland-Palatinate, Germany
Contact: michel.peltriaux@sgdnord.rlp.de
Created on: 21.01.22

"""
import json

from django.db.models import QuerySet
from django.http import HttpRequest, JsonResponse

from api.views.v1.general import AbstractModelAPIViewV1
from intervention.models import Intervention


class APIInterventionViewV1(AbstractModelAPIViewV1):
    model = Intervention

    def get(self, request: HttpRequest, id):
        self.lookup["id"] = id
        self.lookup["users__in"] = [self.user]

        data = self.fetch_and_serialize()
        return JsonResponse(data)

    def compensations_to_json(self, qs: QuerySet):
        return list(
            qs.values(
                "id", "identifier", "title"
            )
        )

    def model_to_json(self, entry: Intervention):
        entry_json = {
            "identifier": entry.identifier,
            "title": entry.title,
            "responsible": self.responsible_to_json(entry.responsible),
            "legal": self.legal_to_json(entry.legal),
            "compensations": self.compensations_to_json(entry.compensations.all()),
            "payments": self.payments_to_json(entry.payments.all()),
            "deductions": self.deductions_to_json(entry.deductions.all()),
            "modified_on": self.modified_on_to_json(entry),
            "created_on": self.created_on_to_json(entry),
        }
        geom = entry.geometry.geom.geojson
        geo_json = json.loads(geom)
        geo_json["properties"] = entry_json
        return geo_json