"""
Author: Michel Peltriaux
Organization: Struktur- und Genehmigungsdirektion Nord, Rhineland-Palatinate, Germany
Contact: ksp-servicestelle@sgdnord.rlp.de
Created on: 18.08.22

"""
from intervention.models import InterventionDocument
from konova.forms.modals import NewDocumentModalForm, EditDocumentModalForm, RemoveDocumentModalForm


class NewInterventionDocumentModalForm(NewDocumentModalForm):
    _DOCUMENT_CLS = InterventionDocument

    def save(self, *args, **kwargs):
        """ Extension of regular NewDocumentModalForm

        Checks whether payments exist on the intervention and sends the data to EGON

        Args:
            *args ():
            **kwargs ():

        Returns:

        """
        doc = super().save(*args, **kwargs)
        self.instance.send_data_to_egon()

        return doc

class EditInterventionDocumentModalForm(EditDocumentModalForm):
    _DOCUMENT_CLS = InterventionDocument

    def save(self, *args, **kwargs):
        """ Extension of regular EditDocumentModalForm

        Checks whether payments exist on the intervention and sends the data to EGON

        Args:
            *args ():
            **kwargs ():

        Returns:

        """
        doc = super().save(*args, **kwargs)
        self.instance.send_data_to_egon()

        return doc


class RemoveInterventionDocumentModalForm(RemoveDocumentModalForm):
    _DOCUMENT_CLS = InterventionDocument

    def save(self, *args, **kwargs):
        super().save(*args, **kwargs)
        self.instance.send_data_to_egon()