File size: 3,466 Bytes
9e798a1 73fedea 9e798a1 427638d 73fedea 427638d 73fedea 427638d 73fedea 427638d 73fedea 427638d |
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 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
from tortoise import fields
from tortoise.models import Model
import datetime
from decimal import Decimal
from App.Users.Model import User
from App.Plans.Model import Plan
from App.Subscriptions.Model import Subscription
from .Schema import PaymentMethod
class Payment(Model):
id = fields.UUIDField(pk=True)
user = fields.ForeignKeyField("models.User", related_name="payments", null=True)
plan = fields.ForeignKeyField(
"models.Plan",
related_name="payments",
null=True,
description="Plan associated with the payment",
)
amount = fields.DecimalField(
max_digits=10, decimal_places=2, description="Payment amount"
)
status = fields.CharField(
max_length=50,
default="pending",
description="Payment status (e.g., pending, completed, failed, balance-assigned)",
)
payment_method = fields.CharField(
max_length=50, choices=PaymentMethod.CHOICES, description="Payment method"
)
transaction_id = fields.CharField(
max_length=100,
unique=True,
null=True,
description="Unique transaction ID for payment (for methods like Lipa Number)",
)
created_time = fields.DatetimeField(auto_now_add=True)
updated_time = fields.DatetimeField(auto_now=True)
class Meta:
table = "payments"
async def create_subscription_if_cash(self):
if self.payment_method == PaymentMethod.CASH and self.user and self.plan:
if self.amount >= self.plan.amount:
expiration_time = datetime.datetime.now() + datetime.timedelta(
hours=self.plan.duration
)
# Create the subscription
await Subscription.create(
user=self.user,
duration=self.plan.duration,
download_mb=self.plan.download_speed * 1024,
upload_mb=self.plan.upload_speed * 1024,
created_time=datetime.datetime.now(),
expiration_time=expiration_time,
active=True,
)
self.status = "subscription-created"
else:
self.status = "insufficient-funds"
await self.save()
async def create_subscription_or_balance(self):
if self.user and self.plan:
if self.amount >= self.plan.amount:
expiration_time = datetime.datetime.now() + datetime.timedelta(
hours=self.plan.duration
)
user = await self.user
# Create the subscription
await Subscription.create(
user=user,
duration=self.plan.duration,
download_mb=self.plan.download_speed * 1024,
upload_mb=self.plan.upload_speed * 1024,
created_time=datetime.datetime.now(),
expiration_time=expiration_time,
active=True,
)
self.status = "subscription-created"
else:
self.status = "insufficient-funds"
await self.save()
elif not self.plan and self.user:
# Await the related user object
user = await self.user
user.balance += self.amount
await user.save()
self.status = "balance-assigned"
await self.save()
|