changed payment code to demo code

This commit is contained in:
2025-07-14 11:29:11 +02:00
parent 549f3a7633
commit 253db898a4
11 changed files with 57 additions and 40 deletions

View File

@@ -1,8 +1,8 @@
from django.contrib import admin
from .models import Document, DocumentSegment, Organization, Risk, Control, DocumentTemplate, DocumentRiskControl, PaymentCode
from .models import Document, DocumentSegment, Organization, Risk, Control, DocumentTemplate, DocumentRiskControl, DemoCode
from django.urls import reverse
from django.utils.html import format_html
from .utils import generate_payment_code
from .utils import generate_demo_code
from django.urls import path
from django.shortcuts import render, redirect
from .forms import GenerateCodesForm
@@ -44,10 +44,10 @@ class ControlAdmin(admin.ModelAdmin):
class DocumentRiskControlAdmin(admin.ModelAdmin):
list_display = ('document', 'risk', 'control', 'weight','likelihood')
class PaymentCodeAdmin(admin.ModelAdmin):
class DemoCodeAdmin(admin.ModelAdmin):
list_display = ('code', 'created_at', 'used', 'company', 'used_at')
change_list_template = "admin/paymentcode_changelist.html"
change_list_template = "admin/democode_changelist.html"
def get_urls(self):
urls = super().get_urls()
@@ -64,9 +64,9 @@ class PaymentCodeAdmin(admin.ModelAdmin):
created = 0
for _ in range(count):
while True:
code = generate_payment_code()
if not PaymentCode.objects.filter(code=code).exists():
PaymentCode.objects.create(code=code)
code = generate_demo_code()
if not DemoCode.objects.filter(code=code).exists():
DemoCode.objects.create(code=code)
created += 1
break
self.message_user(request, f"{created} codes generated.")
@@ -82,4 +82,4 @@ admin.site.register(Risk ,RiskAdmin)
admin.site.register(Control, ControlAdmin)
admin.site.register(DocumentTemplate, DocumentTemplateAdmin)
admin.site.register(DocumentRiskControl, DocumentRiskControlAdmin)
admin.site.register(PaymentCode, PaymentCodeAdmin)
admin.site.register(DemoCode, DemoCodeAdmin)