Slanje dokumenta u PDF fajlu

backend/core/views.py

@@ -3,12 +3,11 @@ import yaml
 
 from django.shortcuts import render, redirect , get_object_or_404
 from .forms import OrganizationForm
-from .models import Organization,Document,Risk, DocumentTemplate,DocumentRiskControl
-from backend.core.utils import get_top_risk
-from django.urls import reverse
+from .models import Organization,Document, DocumentTemplate,DocumentRiskControl
 from backend.accounts.utils import send_confirmation_email, send_document_email
 from django.contrib.admin.views.decorators import staff_member_required
 from django.template import Template, Context
+from .utils import generate_pdf
 
 
 
@@ -101,15 +100,18 @@ def template_preview(request, name):
     parsed_template = template.to_dict()
     return render(request, 'template_preview.html', {'template': parsed_template})
 
+def pdf_view(request, document_id):
+    document = get_object_or_404(Document, id=document_id)
+    return generate_pdf(document)
+
 def payment_page(request):
     email = request.GET.get("email")
-    organization = Organization.objects.get(email=email)
-    document = Document.objects.get(organization=organization)
-    document_link = f"http://127.0.0.1:8000/document/{document.id}/"
-
+    organization = get_object_or_404(Organization, email=email)
+    document = get_object_or_404(Document, organization=organization)
+    
     if request.method == "POST":
-        send_document_email(email, document_link)  
-        return redirect(document_link)
-
-    return render(request, "payment.html", {"email": email})
-
+        pdf_url = f"http://127.0.0.1:8000/pdf/{document.id}"
+        send_document_email(email, pdf_url, document)  
+        return redirect(pdf_url)
+    
+    return render(request, "payment.html", {"email": email})