From c897b5be0815eb41149a5144dacfc49675fa6220 Mon Sep 17 00:00:00 2001 From: Patrick Mayr Date: Mon, 27 Jan 2025 01:04:55 +0100 Subject: [PATCH] replace string with enum --- fet2020/finance/admin.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/fet2020/finance/admin.py b/fet2020/finance/admin.py index 82af7512..70807e23 100644 --- a/fet2020/finance/admin.py +++ b/fet2020/finance/admin.py @@ -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'{obj.get_status_display()}' @@ -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(