Promene u dizajnu dokumenata, controla, residual graph, residual tabele...

backend/core/views.py

@@ -6,8 +6,8 @@ from .forms import OrganizationForm
 from .models import Organization,Document, DocumentTemplate
 from backend.accounts.utils import send_confirmation_email, send_document_email
 from django.contrib.admin.views.decorators import staff_member_required
-from .utils import generate_pdf, generate_risk_graph
-from .tables import risk_matrix_table ,get_risk_table
+from .utils import generate_pdf, generate_risk_graph, generate_residual_risk_graph
+from .tables import risk_matrix_table ,get_risk_table, get_safeguard_summary_table
 from django.conf import settings
 site_domain = settings.SITE_DOMAIN
 from .processors import render_template
@@ -74,8 +74,9 @@ def document(request, document_id):
     document = get_object_or_404(Document, id=document_id)
     risks_with_controls = get_risk_table(document)
     table_risk_matrix = risk_matrix_table()
+    safeguard_summary_table = get_safeguard_summary_table(risks_with_controls)
     graph_base64 = generate_risk_graph(risks_with_controls)
-
+    residual_graph_base64 = generate_residual_risk_graph(risks_with_controls)
     template_obj = get_object_or_404(DocumentTemplate, name="Default Template")
     template_content = template_obj.content 
 
@@ -88,9 +89,15 @@ def document(request, document_id):
         'risks_with_controls': risks_with_controls,
         'graph': graph_base64,
         'table_risk_matrix': table_risk_matrix,
+        'residual_graph': residual_graph_base64,
+        'safeguard_summary_table': safeguard_summary_table,
+        'table_risk_matrix_header' : table_risk_matrix[0],
+        'table_risk_matrix_rows': table_risk_matrix[1:],
     }
     rendered_content = render_template(template_segments, context)
-    return render(request, 'document.html', {'rendered_html': rendered_content})
+    return render(request, 'document.html',
+                   {'rendered_html': rendered_content,
+                    'document': document,})
 
 
 @staff_member_required