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(