from django.db import models
from django.utils.translation import gettext_lazy as _


class BankType(models.TextChoices):
    BMI = "BMI", _("BMI")
    SEP = "SEP", _("SEP")
    ZARINPAL = "ZARINPAL", _("Zarinpal")
    ZIBAL = "ZIBAL", _("Zibal")
    BAHAMTA = "BAHAMTA", _("Bahamta")
    MELLAT = "MELLAT", _("Mellat")
    IRANDARGAH = "IRANDARGAH", _("IranDargah")
    ASANPARDAKHT = "ASANPARDAKHT", _("AsanPardakht")


class CurrencyEnum(models.TextChoices):
    IRR = "IRR", _("Rial")
    IRT = "IRT", _("Toman")

    @classmethod
    def rial_to_toman(cls, amount):
        return amount / 10

    @classmethod
    def toman_to_rial(cls, amount):
        return amount * 10


class PaymentStatus(models.TextChoices):
    WAITING = "WAITING", _("Waiting")
    REDIRECT_TO_BANK = "REDIRECT_TO_BANK", _("Redirect to bank")
    RETURN_FROM_BANK = "RETURN_FROM_BANK", _("Return from bank")
    CANCEL_BY_USER = "CANCEL_BY_USER", _("Cancel by user")
    EXPIRE_GATEWAY_TOKEN = "EXPIRE_GATEWAY_TOKEN", _("Expire gateway token")
    EXPIRE_VERIFY_PAYMENT = "EXPIRE_VERIFY_PAYMENT", _("Expire verify payment")
    COMPLETE = "COMPLETE", _("Complete")
    ERROR = "ERROR", _("Unknown error acquired")