GeriLife/caregiving

import uuid
 
from django.db import transaction
from django.contrib.auth.mixins import LoginRequiredMixin
from django.urls import reverse_lazy
from django.views.generic import ListView, FormView
 
from metrics.models import ResidentActivity
from residents.models import Residency, Resident
from metrics.forms import ResidentActivityForm
 
 
class ResidentActivityListView(LoginRequiredMixin, ListView):
    template_name = "activities/resident_activity_list.html"
    queryset = ResidentActivity.objects.all()
    context_object_name = "activities"
    paginate_by = 100
    ordering = ["-activity_date"]
 
 
class ResidentActivityFormView(LoginRequiredMixin, FormView):
    template_name = "activities/resident_activity_form.html"
    form_class = ResidentActivityForm
    success_url = reverse_lazy("activity-list-view")
 
    # Check whether request user is authorized to view this page
    def dispatch(self, request, *args, **kwargs):
        if not request.user.can_add_activity:
            return self.handle_no_permission()
 
        return super().dispatch(request, *args, **kwargs)
 
    def get_form_kwargs(self):
        """Override the get_form_kwargs method to pass the user to the form.
 
        This will allow the form to filter the residents by the user's
        homes or the superuser to filter by all homes.
        """
 
        kwargs = super().get_form_kwargs()
 
        kwargs["user"] = self.request.user
 
        return kwargs
 
Cyclomatic complexity is too high in method post. (8)
    @transaction.atomic
Function `post` has a Cognitive Complexity of 14 (exceeds 5 allowed). Consider refactoring.
    def post(self, request, *args, **kwargs):
        """Override the post method to add the resident activity in the same
        transaction as the activity."""
        form = self.get_form()
        is_form_valid = form.is_valid()
 
        if is_form_valid:
            # create a resident activity for each resident in the form
            residents_field_errors = []
 
            resident_ids = form.cleaned_data["residents"]
 
            # generate group activity ID based on current epoch time
            group_activity_id = uuid.uuid4()
 
            if not request.user.can_manage_residents(resident_ids):
                return self.handle_no_permission()
 
            for resident_id in resident_ids:
                try:
                    resident = Resident.objects.get(id=resident_id)
                    _create_resident_activity(resident, group_activity_id, form)
                except Resident.DoesNotExist:
                    transaction.set_rollback(True)
 
                    error_message = f"Resident ID {resident_id} does not exist."
 
                    is_form_valid = False
                    residents_field_errors.append(error_message)
                except Residency.DoesNotExist:
                    transaction.set_rollback(True)
 
                    error_message = (
                        f"Resident {resident} is not currently residing in a home."
                    )
 
                    is_form_valid = False
                    residents_field_errors.append(error_message)
 
            if residents_field_errors:
                form.add_error("residents", residents_field_errors)
 
        return self.form_valid(form) if is_form_valid else self.form_invalid(form)
 
 
def _create_resident_activity(
    resident: Resident,
    group_activity_id: uuid.UUID,
    form: ResidentActivityForm,
):
    """Create a resident activity for the given resident and form."""
 
    try:
        residency = Residency.objects.get(
            resident=resident,
            move_out__isnull=True,
        )
    except Residency.DoesNotExist:
        raise
 
    home = residency.home
    activity_type = form.cleaned_data["activity_type"]
    activity_minutes = form.cleaned_data["activity_minutes"]
    caregiver_role = form.cleaned_data["caregiver_role"]
    activity_date = form.cleaned_data["activity_date"]
 
    resident_activity = ResidentActivity.objects.create(
        resident=resident,
        residency=residency,
        home=home,
        activity_type=activity_type,
        activity_minutes=activity_minutes,
        caregiver_role=caregiver_role,
        activity_date=activity_date,
        group_activity_id=group_activity_id,
    )
 
    resident_activity.save()