mirror of
https://github.com/gmfricke/cmmc_audit.git
synced 2026-06-28 19:53:04 -06:00
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
from django.shortcuts import get_object_or_404
|
|
|
|
from coldfront.core.allocation.models import Allocation
|
|
|
|
from .models import AuditEvent
|
|
from .resolvers import allocation_label
|
|
from .utils import log_event
|
|
|
|
|
|
PATCH_MARKER = "_cmmc_audit_allocation_workflow_patch_installed"
|
|
|
|
|
|
def _date_value(value):
|
|
return value.isoformat() if value else None
|
|
|
|
|
|
def _allocation_summary(allocation):
|
|
return allocation_label(allocation)
|
|
|
|
|
|
def install_allocation_workflow_audit_patch():
|
|
from coldfront.core.allocation import views as allocation_views
|
|
|
|
if getattr(allocation_views, PATCH_MARKER, False):
|
|
return
|
|
|
|
original_renew_post = allocation_views.AllocationRenewView.post
|
|
|
|
def renew_post(self, request, *args, **kwargs):
|
|
allocation = get_object_or_404(
|
|
Allocation.objects.select_related("project", "status"),
|
|
pk=self.kwargs.get("pk"),
|
|
)
|
|
old_status = allocation.status.name
|
|
old_end_date = allocation.end_date
|
|
|
|
response = original_renew_post(self, request, *args, **kwargs)
|
|
|
|
allocation.refresh_from_db()
|
|
allocation = Allocation.objects.select_related("project", "status").get(pk=allocation.pk)
|
|
new_status = allocation.status.name
|
|
if old_status != new_status and new_status == "Renewal Requested":
|
|
log_event(
|
|
AuditEvent.Action.RENEWAL_REQUESTED,
|
|
allocation,
|
|
old_values={
|
|
"status": old_status,
|
|
"end_date": _date_value(old_end_date),
|
|
"project_id": allocation.project_id,
|
|
},
|
|
new_values={
|
|
"status": new_status,
|
|
"end_date": _date_value(allocation.end_date),
|
|
"project_id": allocation.project_id,
|
|
},
|
|
message=f"Renewal requested for {_allocation_summary(allocation)}",
|
|
actor=getattr(request, "user", None),
|
|
request=request,
|
|
source=AuditEvent.Source.COLDFRONT_WORKFLOW,
|
|
target_repr=_allocation_summary(allocation),
|
|
)
|
|
|
|
return response
|
|
|
|
allocation_views.AllocationRenewView.post = renew_post
|
|
setattr(allocation_views, PATCH_MARKER, True)
|