State views

* replaces function based state views with class based

ema/views/state.py

@@ -6,85 +6,42 @@ Created on: 19.08.22
 
 """
 from django.contrib.auth.decorators import login_required
-from django.http import HttpRequest
-from django.shortcuts import get_object_or_404
-from django.urls import reverse
+from django.utils.decorators import method_decorator
 
-from compensation.forms.modals.state import NewCompensationStateModalForm, RemoveCompensationStateModalForm, \
-    EditCompensationStateModalForm
-from compensation.models import CompensationState
 from ema.models import Ema
 from konova.decorators import conservation_office_group_required, shared_access_required
-from konova.utils.message_templates import COMPENSATION_STATE_ADDED, COMPENSATION_STATE_REMOVED, \
-    COMPENSATION_STATE_EDITED
+from konova.views.state import AbstractNewCompensationStateView, AbstractEditCompensationStateView, \
+    AbstractRemoveCompensationStateView
 
 
-@login_required
-@conservation_office_group_required
-@shared_access_required(Ema, "id")
-def state_new_view(request: HttpRequest, id: str):
-    """ Renders a form for adding new states for an EMA
+class NewEmaStateView(AbstractNewCompensationStateView):
+    model = Ema
+    redirect_url = "ema:detail"
 
-    Args:
-        request (HttpRequest): The incoming request
-        id (str): The EMA's id to which the new state will be related
-
-    Returns:
-
-    """
-    ema = get_object_or_404(Ema, id=id)
-    form = NewCompensationStateModalForm(request.POST or None, instance=ema, request=request)
-    return form.process_request(
-        request,
-        msg_success=COMPENSATION_STATE_ADDED,
-        redirect_url=reverse("ema:detail", args=(id,)) + "#related_data"
-    )
+    @method_decorator(login_required)
+    @method_decorator(conservation_office_group_required)
+    @method_decorator(shared_access_required(Ema, "id"))
+    def dispatch(self, request, *args, **kwargs):
+        return super().dispatch(request, *args, **kwargs)
 
 
-@login_required
-@conservation_office_group_required
-@shared_access_required(Ema, "id")
-def state_remove_view(request: HttpRequest, id: str, state_id: str):
-    """ Renders a form for removing an EMA state
+class EditEmaStateView(AbstractEditCompensationStateView):
+    model = Ema
+    redirect_url = "ema:detail"
 
-    Args:
-        request (HttpRequest): The incoming request
-        id (str): The ema id
-        state_id (str): The state's id
-
-    Returns:
-
-    """
-    ema = get_object_or_404(Ema, id=id)
-    state = get_object_or_404(CompensationState, id=state_id)
-    form = RemoveCompensationStateModalForm(request.POST or None, instance=ema, state=state, request=request)
-    return form.process_request(
-        request,
-        msg_success=COMPENSATION_STATE_REMOVED,
-        redirect_url=reverse("ema:detail", args=(id,)) + "#related_data"
-    )
+    @method_decorator(login_required)
+    @method_decorator(conservation_office_group_required)
+    @method_decorator(shared_access_required(Ema, "id"))
+    def dispatch(self, request, *args, **kwargs):
+        return super().dispatch(request, *args, **kwargs)
 
 
-@login_required
-@conservation_office_group_required
-@shared_access_required(Ema, "id")
-def state_edit_view(request: HttpRequest, id: str, state_id: str):
-    """ Renders a form for editing an EMA state
-
-    Args:
-        request (HttpRequest): The incoming request
-        id (str): The ema id
-        state_id (str): The state's id
-
-    Returns:
-
-    """
-    ema = get_object_or_404(Ema, id=id)
-    state = get_object_or_404(CompensationState, id=state_id)
-    form = EditCompensationStateModalForm(request.POST or None, instance=ema, state=state, request=request)
-    return form.process_request(
-        request,
-        msg_success=COMPENSATION_STATE_EDITED,
-        redirect_url=reverse("ema:detail", args=(id,)) + "#related_data"
-    )
+class RemoveEmaStateView(AbstractRemoveCompensationStateView):
+    model = Ema
+    redirect_url = "ema:detail"
 
+    @method_decorator(login_required)
+    @method_decorator(conservation_office_group_required)
+    @method_decorator(shared_access_required(Ema, "id"))
+    def dispatch(self, request, *args, **kwargs):
+        return super().dispatch(request, *args, **kwargs)