senaduka/old-riskletpy
1
0
Fork 0
Code Issues 5 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

changed payment code to demo code

Browse Source
This commit is contained in:
Amir
2025-07-14 11:29:11 +02:00
parent 549f3a7633
commit 253db898a4
11 changed files with 57 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
backend/core/views.py
View File

@@ -6,7 +6,7 @@ import time
from django.shortcuts import render, redirect , get_object_or_404
from .forms import OrganizationForm
from .models import Organization,Document, DocumentTemplate, PaymentCode
from .models import Organization,Document, DocumentTemplate, DemoCode
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, generate_residual_risk_graph
@@ -123,7 +123,7 @@ def payment_page(request):
import re
code = re.sub(r'\s+', '', request.POST.get('code', '')).upper()[:10]
try:
payment_code = PaymentCode.objects.get(code=code)
payment_code = DemoCode.objects.get(code=code)
if payment_code.used:
error = "CODE INVALID"
else:
@@ -136,7 +136,7 @@ def payment_page(request):
url = f"{site_domain}/pdf/{document.id}/"
send_document_email(email, url, document)
return redirect(url)
except PaymentCode.DoesNotExist:
except DemoCode.DoesNotExist:
error = "❌ Invalid code"
return render(request, 'payment.html', {'error': error})
@@ -152,18 +152,18 @@ def download_example_pdf(request):
@staff_member_required
def payment_codes_pdf_view(request):
def demo_codes_pdf_view(request):
filter_by = request.GET.get('filter_by', 'all')
if filter_by == 'used':
codes = PaymentCode.objects.filter(used=True)
codes = DemoCode.objects.filter(used=True)
elif filter_by == 'available':
codes = PaymentCode.objects.filter(used=False)
codes = DemoCode.objects.filter(used=False)
else:
codes = PaymentCode.objects.all()
html_string = render_to_string('payment_code_report.html', {'codes': codes})
codes = DemoCode.objects.all()
html_string = render_to_string('demo_code_report.html', {'codes': codes})
pdf_content = HTML(string=html_string, base_url=request.build_absolute_uri('/')).write_pdf()
response = HttpResponse(pdf_content, content_type='application/pdf')
response['Content-Disposition'] = f'inline; filename=payment_codes_{timezone.now().strftime("%Y%m%d_%H%M%S")}.pdf'
response['Content-Disposition'] = f'inline; filename=demo_codes_{timezone.now().strftime("%Y%m%d_%H%M%S")}.pdf'
return response
@csrf_exempt
@@ -172,8 +172,8 @@ def validate_code(request):
try:
data = json.loads(request.body)
code = data.get("code", "").strip().upper()
from .models import PaymentCode
valid = PaymentCode.objects.filter(code=code, used=False).exists()
from .models import DemoCode
valid = DemoCode.objects.filter(code=code, used=False).exists()
time.sleep(3)
return JsonResponse({"valid": valid})
except Exception: