File size: 1,907 Bytes
9e798a1 f9fbe3f 9e798a1 f9fbe3f 9e798a1 f9fbe3f dfa6f47 f9fbe3f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
from tortoise import fields
from tortoise.models import Model
from datetime import datetime, timedelta
class Plan(Model):
id = fields.UUIDField(pk=True)
name = fields.CharField(
max_length=100, unique=True, description="Name of the subscription plan"
)
amount = fields.DecimalField(
max_digits=10, decimal_places=2, description="Cost of the plan"
)
duration = fields.IntField(description="Duration of the subscription in hours")
download_speed = fields.FloatField(description="Download speed in Mbps")
upload_speed = fields.FloatField(description="Upload speed in Mbps")
expire_date = fields.DatetimeField(
default=None, null=True, description="Expiration date of the plan"
)
is_promo = fields.BooleanField(
default=False, description="Indicates if the plan is a promotional plan"
)
promo_duration_days = fields.IntField(
default=None, null=True, description="Number of days the promotion is valid"
)
is_valid = fields.BooleanField(
default=True, description="Indicates if the plan is valid"
)
class Meta:
table = "plans"
async def set_expiration(self):
"""Set the expiration date based on the promo duration if applicable."""
if self.is_promo and self.promo_duration_days:
self.expire_date = datetime.now() + timedelta(days=self.promo_duration_days)
elif self.duration:
self.expire_date = datetime.now() + timedelta(
days=self.duration // 24
) # Convert hours to days
await self.save()
async def is_promo_valid(self) -> bool:
"""Check if the promotional plan is still valid."""
if self.is_promo and self.expire_date:
return datetime.now() < self.expire_date.replace(tzinfo=None)
self.is_valid = False
await self.save()
return False
|