changed payment code to demo code
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user