diff --git a/aleksis/apps/tezor/models/invoice.py b/aleksis/apps/tezor/models/invoice.py index d5d4ab4b39f315c02aa4e68e506cbcb2c3127605..d8e9073529e5f58717daae79abe8b63dc12e7ccb 100644 --- a/aleksis/apps/tezor/models/invoice.py +++ b/aleksis/apps/tezor/models/invoice.py @@ -1,3 +1,4 @@ +from django.conf import settings from django.contrib.contenttypes.fields import GenericForeignKey from django.contrib.contenttypes.models import ContentType from django.db import models @@ -50,6 +51,13 @@ class Invoice(BasePayment, PureDjangoModel): for_object_id = models.PositiveIntegerField() for_object = GenericForeignKey("for_content_type", "for_object_id") + @classmethod + def get_variant_choices(cls): + choices = [] + for variant in settings.PAYMENT_VARIANTS.keys(): + choices.append((variant, cls.VARIANT_DISPLAY[self.variant][0])) + return choices + def get_variant_name(self): return self.__class__.VARIANT_DISPLAY[self.variant][0]