Pormenjen naicn rendovanja podatak iz template, popunjeni ceo dokument

2025-04-18 16:29:07 +02:00
parent fc29671331
commit 31d679d9cf
7 changed files with 753 additions and 142 deletions
--- a/backend/core/tables.py
+++ b/backend/core/tables.py
@@ -0,0 +1,108 @@
+from backend.core.models import DocumentRiskControl
+from backend.core.utils import calculate_aggregate_likelihood, calculate_aggregate_weight, map_weight_to_impact_likelihood
+
+
+def risk_matrix_table():
+    likelihood_labels = [
+        "Certain (90-100%)",
+        "Almost Certain (80-89%)",
+        "Very Probable (70-79%)",
+        "Probable (60-69%)",
+        "Highly Likely (50-59%)",
+        "Likely (40-49%)",
+        "Occasional (30-39%)",
+        "Possible (20-29%)",
+        "Unlikely (10-19%)",
+        "Rare (0-9%)"
+    ]
+
+    impact_labels = [
+        "Insignificant",
+        "Minor",
+        "Moderate",
+        "Major",
+        "Severe",
+        "Catastrophic",
+        "Critical",
+        "Extreme",
+        "Disastrous",
+        "Unrecoverable"
+    ]
+
+    color_mapping = {
+        "Very Low": "green",
+        "Low": "lightgreen",
+        "Medium": "yellow",
+        "High": "orange",
+        "Critical": "red"
+    }
+
+    table_matrix_risk = [["Impact ↓ / Likelihood →"] + impact_labels]
+
+    for likelihood_index, likelihood_label in enumerate(likelihood_labels, start=1):
+        reversed_index = 11 - likelihood_index  
+        row = [likelihood_label]
+        for impact_index in range(1, 11):
+            score = reversed_index * impact_index
+            if score <= 20:
+                label = "Very Low"
+            elif score <= 40:
+                label = "Low"
+            elif score <= 60:
+                label = "Medium"
+            elif score <= 80:
+                label = "High"
+            else:
+                label = "Critical"
+            color_class = color_mapping[label]
+            row.append((score, label, color_class))
+        table_matrix_risk.append(row)
+
+    return table_matrix_risk
+
+def get_risk_table(document):
+    risks = (
+        DocumentRiskControl.objects
+        .filter(document=document)
+        .values('risk', 'risk__risk_name')
+        .distinct()
+    )
+
+    risks_with_controls = []
+
+    for risk_entry in risks:
+        risk = {
+            'id': risk_entry['risk'],
+            'name': risk_entry['risk__risk_name']
+        }
+
+        controls = (
+            DocumentRiskControl.objects
+            .filter(document=document, risk_id=risk['id'])
+            .values('control', 'control__name', 'weight', 'likelihood')
+            .distinct()
+        )
+        max_weight = 10*10
+        total_weight = calculate_aggregate_weight(controls)
+        total_likelihood = calculate_aggregate_likelihood(controls)
+        impact, likelihood = map_weight_to_impact_likelihood(total_weight, total_likelihood, max_weight)
+        r_impact = round(impact)
+        r_likelihood = round(likelihood)
+        residua_impact = r_impact - 1 if r_impact > 2 else r_impact
+        residual_likelihood = r_likelihood - 1 if r_likelihood > 2 else r_likelihood
+        risks_with_controls.append({
+            'risk': risk,
+            'controls': list(controls),
+            'total_weight': total_weight,
+            'impact': impact,
+            'likelihood': likelihood,
+            'r_impact': r_impact,
+            'r_likelihood': r_likelihood,
+            'risk_score': r_impact * r_likelihood,
+            'residual_impact': residua_impact,
+            'residual_likelihood': residual_likelihood,
+            'residual_risk_score': residua_impact * residual_likelihood,
+        })
+    risks_with_controls.sort(key=lambda x: x['risk_score'], reverse=True)
+
+    return risks_with_controls