Issue#4: Staviti izabrane rizike u bazu kako treba.

backend/core/views.py

@@ -1,9 +1,10 @@
 import logging
 
-from django.shortcuts import render, redirect
+from django.shortcuts import render, redirect , get_object_or_404
 from .forms import OrganizationForm
 from .models import Organization,Document,Risk
 from backend.core.utils import get_top_risk
+from django.urls import reverse
 # @login_required
 # def index(request):
 #     return HttpResponse('<h1>Django</h1><p>Página simples.</p>')
@@ -24,17 +25,28 @@ def signup(request):
             top_risk_ids = get_top_risk(organization)
             top_risks = Risk.objects.filter(risk_id__in = top_risk_ids)
 
+            organization.risks.set(top_risks)
+
             document = Document.objects.create(organization=organization)
             document.add_segment('h1', "Top 10 Risk Identified")
 
-            for risk in top_risks:
-                document.add_segment('h2',f"Risk: {risk.risk_id}:{risk.risk_name}")
-                document.add_segment('body',f"Category: {risk.category} \n Primary Impact: {risk.primary_impact} \n Secondary Impact: {risk.secondary_impact} \n Tertiary Impact: {risk.tretiary_impact} \n Detection Difficulty: {risk.detection_difficulty} \n Recovery Complexity: {risk.recovery_complexity} \n Business Impact Severity: {risk.businnes_impact_severity} ")
+            risk_content = "\n\n".join([
+                f"Risk: {risk.risk_id} : {risk.risk_name} \n"
+                f"Category: {risk.category}\n"
+                f"Primary Impaact: {risk.primary_impact} \n"
+                f"Secondary Impact: {risk.secondary_impact}\n"
+                f"Tertiary Impact: {risk.tretiary_impact} \n"
+                f"Detection Difficulty: {risk.detection_difficulty} \n"
+                f"Recovery Complexity: {risk.recovery_complexity} \n"
+                f"Business Impact Severity: {risk.businnes_impact_severity}\n"
+                for risk in top_risks
+            ])
+            
+            document.add_segment('body',f"Identified Risks: \n\n{risk_content}")
 
             return render(request, 'thankyou.html', {
                 'email': form.data['email'],
-                'top_risks':top_risks,
-                'document':document
+                'document_link': reverse('core:document', args=[str(document.id)])
                 })
         else:
             logging.error(form.errors)
@@ -48,5 +60,12 @@ def signup(request):
 def thankyou(request):
     return render(request, 'thankyou.html')
 
-def document(request):
-    return render(request, 'document.html')
+def document(request, document_id):  
+    print(f"Document ID received: {document_id}")
+    doc = get_object_or_404(Document, id=document_id)
+
+    return render(request, 'document.html', {
+        'document': doc,
+        'organization': doc.organization,
+        'segments': doc.segments.all(),
+    })