55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
|
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
|