diff --git a/backend/core/admin.py b/backend/core/admin.py index e9c1e27..9d34b80 100644 --- a/backend/core/admin.py +++ b/backend/core/admin.py @@ -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) diff --git a/backend/core/migrations/0018_rename_paymentcode_democode.py b/backend/core/migrations/0018_rename_paymentcode_democode.py new file mode 100644 index 0000000..f59d4ee --- /dev/null +++ b/backend/core/migrations/0018_rename_paymentcode_democode.py @@ -0,0 +1,17 @@ +# Generated by Django 5.1.3 on 2025-07-13 22:45 + +from django.db import migrations + + +class Migration(migrations.Migration): + + dependencies = [ + ('core', '0017_document_key_findings'), + ] + + operations = [ + migrations.RenameModel( + old_name='PaymentCode', + new_name='DemoCode', + ), + ] diff --git a/backend/core/models.py b/backend/core/models.py index aa21de9..3ef45cd 100644 --- a/backend/core/models.py +++ b/backend/core/models.py @@ -175,7 +175,7 @@ class DocumentRiskControl(models.Model): unique_together = ('document', 'risk', 'control') -class PaymentCode(models.Model): +class DemoCode(models.Model): code = models.CharField(max_length=10, unique=True) created_at = models.DateTimeField(auto_now_add=True) used = models.BooleanField(default=False) diff --git a/backend/core/templates/admin/democode_changelist.html b/backend/core/templates/admin/democode_changelist.html new file mode 100644 index 0000000..04a4e3c --- /dev/null +++ b/backend/core/templates/admin/democode_changelist.html @@ -0,0 +1,14 @@ +{% extends "admin/change_list.html" %} +{% block object-tools %} +