105 lines
4.3 KiB
Python
105 lines
4.3 KiB
Python
from decimal import Decimal
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
from accounts.models import BaseModel, IAmPrincipalType
|
|
from django.core.exceptions import ValidationError
|
|
|
|
|
|
class Coupon(BaseModel):
|
|
title = models.CharField(max_length=255)
|
|
coupon_code = models.CharField(max_length=50, unique=True)
|
|
coupon_id = models.CharField(max_length=255, blank=True, null=True)
|
|
no_of_redeems = models.IntegerField(default=0)
|
|
description = models.TextField(null=True, blank=True)
|
|
image = models.ImageField(upload_to="coupon_img", null=True, blank=True)
|
|
discount_amount = models.DecimalField(
|
|
max_digits=10, decimal_places=2, null=True, blank=True, help_text="Representing the amount to subtract from an invoice total (required if discount_percentage is not passed)"
|
|
)
|
|
discount_percentage = models.DecimalField(
|
|
max_digits=5, decimal_places=2, null=True, blank=True, help_text="A positive float larger than 0, and smaller or equal to 100, that represents the discount the coupon will apply (required if discount_amount is not passed)."
|
|
)
|
|
valid_from = models.DateTimeField()
|
|
valid_to = models.DateTimeField(help_text="Datetime for the last redeemable date. After this, the coupon is invalid for new customers.")
|
|
max_redeems = models.IntegerField(default=1)
|
|
|
|
class Meta:
|
|
db_table = "coupon"
|
|
|
|
def __str__(self):
|
|
return self.coupon_code
|
|
|
|
def clean(self):
|
|
"""
|
|
Validate the Coupon instance. Ensure that the `max_redeems` is greater than 0,
|
|
that either `discount_amount` or `discount_percentage` is set, and that
|
|
`valid_from` is earlier than `valid_to`.
|
|
"""
|
|
if self.max_redeems < 1:
|
|
raise ValidationError({"max_redeems": "Redeems must be more than 1."})
|
|
|
|
# Ensure discount_amount is non-negative
|
|
if self.discount_amount is not None and self.discount_amount < 1:
|
|
raise ValidationError(
|
|
{"discount_amount": "Discount amount must be more than 1."}
|
|
)
|
|
|
|
# Ensure discount_percentage is non-negative
|
|
if self.discount_percentage is not None and self.discount_percentage < 1:
|
|
raise ValidationError(
|
|
{"discount_percentage": "Discount percentage must be more than 1."}
|
|
)
|
|
|
|
if self.discount_amount and self.discount_percentage:
|
|
raise ValidationError(
|
|
"You can only set either a discount amount or a discount percentage, not both."
|
|
)
|
|
|
|
if not self.discount_amount and not self.discount_percentage:
|
|
raise ValidationError(
|
|
"You must set either a discount amount or a discount percentage."
|
|
)
|
|
|
|
if self.valid_from and self.valid_to and self.valid_from >= self.valid_to:
|
|
raise ValidationError(
|
|
"The valid_from date must be earlier than the valid_to date."
|
|
)
|
|
|
|
def save(self, *args, **kwargs):
|
|
from goodtimes.services import StripeService
|
|
if not self.delete:
|
|
self.clean() # Call clean before saving to ensure validation
|
|
|
|
if not self.pk and not self.coupon_id:
|
|
amount_off = int(self.discount_amount * Decimal(100)) if self.discount_amount else None
|
|
percent_off = float(self.discount_percentage) if self.discount_percentage else None
|
|
|
|
result = StripeService.create_coupon(
|
|
amount_off=amount_off,
|
|
percent_off=percent_off,
|
|
duration="once",
|
|
name=self.title,
|
|
redeem_by=int(self.valid_to.timestamp()),
|
|
max_redemptions=self.max_redeems,
|
|
currency='gbp',
|
|
metadata={"local_id": self.id}
|
|
)
|
|
|
|
if not result["success"]:
|
|
raise ValueError(f"Failed to create Stripe coupon: {result['message']}")
|
|
|
|
self.coupon_code = result['data'].id
|
|
self.coupon_id = result["data"].id
|
|
|
|
super().save(*args, **kwargs)
|
|
|
|
# If max_redeems is 0, it means that we are allowing unlimited redeems
|
|
|
|
# def is_valid(self):
|
|
# now = timezone.now()
|
|
# return (
|
|
# self.active
|
|
# and not self.deleted
|
|
# and self.valid_from <= now <= self.valid_to
|
|
# and (self.max_redeems == 0 or self.no_of_redeems < self.max_redeems)
|
|
# )
|