replace string with enum

This commit is contained in:
2025-01-27 01:04:55 +01:00
parent fcf63d8bf9
commit c897b5be08

View File

@@ -308,7 +308,7 @@ class BillAdmin(admin.ModelAdmin):
@admin.display(description="QR Code")
def get_qrcode(self, obj):
if obj.status != "C":
if obj.status != Bill.Status.CLEARED:
return "-"
try:
@@ -335,10 +335,10 @@ class BillAdmin(admin.ModelAdmin):
def status_colored(self, obj):
# TODO: if there is a status without color, set nothing.
colors = {
"S": "red",
"C": "darkorange",
"F": "green",
"I": "blue",
Bill.Status.SUBMITTED: "red",
Bill.Status.CLEARED: "darkorange",
Bill.Status.FINISHED: "green",
Bill.Status.INCOMPLETED: "blue",
}
return format_html(
f'<b style="background:{colors[obj.status]};">{obj.get_status_display()}</b>'
@@ -346,7 +346,7 @@ class BillAdmin(admin.ModelAdmin):
@admin.action(description="Als 'Abgerechnet' markieren.")
def make_cleared(self, request, queryset):
updated = queryset.update(status="C")
updated = queryset.update(status=Bill.Status.CLEARED)
self.message_user(
request,
ngettext(
@@ -360,7 +360,7 @@ class BillAdmin(admin.ModelAdmin):
@admin.action(description="Als 'Abgeschlossen' markieren.")
def make_finished(self, request, queryset):
updated = queryset.update(status="F")
updated = queryset.update(status=Bill.Status.FINISHED)
self.message_user(
request,
ngettext(