from django.db import models
from django.contrib.auth import get_user_model

from .helpers import generate_voucher_code

from payment.const import PAYMENT_STATUS_SUCCESS, PAYMENT_STATUS_FAILED, PAYMENT_STATUS_REFUNDED
from clients.models import ClientProfile

class Voucher(models.Model):
    code = models.IntegerField(default=generate_voucher_code)
    value = models.DecimalField(max_digits=10, decimal_places=2)
    created_at = models.DateTimeField(auto_now_add=True)

    @property
    def is_active(self):
        try:
            self.voucheractivation
            return True
        except VoucherActivation.DoesNotExist:
            return False

class VoucherActivation(models.Model):
    voucher = models.OneToOneField(Voucher, models.PROTECT)
    user = models.ForeignKey(get_user_model(), models.PROTECT)
    activated_at = models.DateTimeField(auto_now_add=True)
    data = models.CharField(max_length=256)

class VoucherApplication(models.Model):
    voucher = models.OneToOneField(Voucher, models.PROTECT)
    client = models.ForeignKey(get_user_model(), models.PROTECT)
    applied_at = models.DateTimeField(auto_now_add=True)

class VoucherInvoicePayment(InvoicePayment):
    success = models.BooleanField(default=True)

    @property
    def gateway(self):
        return "Voucher"

    @classmethod
    def initiate(cls, invoice, amount=None):
        amount = amount or (invoice.balance * -1)

        payment = cls.objects.create(invoice=invoice, amount=amount, gateway_id=uuid.uuid4())

        if invoice.user.balance <= 0:
            payment.success = False
            payment.save()
        
        invoice.finalize()
        return reverse_lazy("payment:status", args=[payment.uuid])

    @property
    def status(self):
        return PAYMENT_STATUS_SUCCESS if self.success else PAYMENT_STATUS_FAILED