|
from tortoise import fields, models
|
|
from decimal import Decimal
|
|
|
|
class SubscriptionPlan(models.Model):
|
|
id = fields.IntField(pk=True)
|
|
name = fields.CharField(max_length=100, unique=True)
|
|
display_name = fields.CharField(max_length=255)
|
|
description = fields.TextField(null=True)
|
|
storage_gb = fields.IntField()
|
|
bandwidth_gb = fields.IntField()
|
|
price_monthly = fields.DecimalField(max_digits=10, decimal_places=2)
|
|
price_yearly = fields.DecimalField(max_digits=10, decimal_places=2, null=True)
|
|
stripe_price_id = fields.CharField(max_length=255, null=True)
|
|
is_active = fields.BooleanField(default=True)
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
updated_at = fields.DatetimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
table = "subscription_plans"
|
|
|
|
class UserSubscription(models.Model):
|
|
id = fields.IntField(pk=True)
|
|
user = fields.ForeignKeyField("models.User", related_name="subscription")
|
|
plan = fields.ForeignKeyField("billing.SubscriptionPlan", related_name="subscriptions", null=True)
|
|
billing_type = fields.CharField(max_length=20, default="pay_as_you_go")
|
|
stripe_customer_id = fields.CharField(max_length=255, unique=True, null=True)
|
|
stripe_subscription_id = fields.CharField(max_length=255, unique=True, null=True)
|
|
status = fields.CharField(max_length=50, default="active")
|
|
current_period_start = fields.DatetimeField(null=True)
|
|
current_period_end = fields.DatetimeField(null=True)
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
updated_at = fields.DatetimeField(auto_now=True)
|
|
canceled_at = fields.DatetimeField(null=True)
|
|
|
|
class Meta:
|
|
table = "user_subscriptions"
|
|
|
|
class UsageRecord(models.Model):
|
|
id = fields.BigIntField(pk=True)
|
|
user = fields.ForeignKeyField("models.User", related_name="usage_records")
|
|
record_type = fields.CharField(max_length=50, index=True)
|
|
amount_bytes = fields.BigIntField()
|
|
resource_type = fields.CharField(max_length=50, null=True)
|
|
resource_id = fields.IntField(null=True)
|
|
timestamp = fields.DatetimeField(auto_now_add=True, index=True)
|
|
idempotency_key = fields.CharField(max_length=255, unique=True, null=True)
|
|
metadata = fields.JSONField(null=True)
|
|
|
|
class Meta:
|
|
table = "usage_records"
|
|
indexes = [("user_id", "record_type", "timestamp")]
|
|
|
|
class UsageAggregate(models.Model):
|
|
id = fields.IntField(pk=True)
|
|
user = fields.ForeignKeyField("models.User", related_name="usage_aggregates")
|
|
date = fields.DateField()
|
|
storage_bytes_avg = fields.BigIntField(default=0)
|
|
storage_bytes_peak = fields.BigIntField(default=0)
|
|
bandwidth_up_bytes = fields.BigIntField(default=0)
|
|
bandwidth_down_bytes = fields.BigIntField(default=0)
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
table = "usage_aggregates"
|
|
unique_together = (("user", "date"),)
|
|
|
|
class Invoice(models.Model):
|
|
id = fields.IntField(pk=True)
|
|
user = fields.ForeignKeyField("models.User", related_name="invoices")
|
|
invoice_number = fields.CharField(max_length=50, unique=True)
|
|
stripe_invoice_id = fields.CharField(max_length=255, unique=True, null=True)
|
|
period_start = fields.DateField(db_index=True)
|
|
period_end = fields.DateField()
|
|
subtotal = fields.DecimalField(max_digits=10, decimal_places=4)
|
|
tax = fields.DecimalField(max_digits=10, decimal_places=4, default=0)
|
|
total = fields.DecimalField(max_digits=10, decimal_places=4)
|
|
currency = fields.CharField(max_length=3, default="USD")
|
|
status = fields.CharField(max_length=50, default="draft", index=True)
|
|
due_date = fields.DateField(null=True)
|
|
paid_at = fields.DatetimeField(null=True)
|
|
created_at = fields.DatetimeField(auto_now_add=True, index=True)
|
|
updated_at = fields.DatetimeField(auto_now=True)
|
|
metadata = fields.JSONField(null=True)
|
|
|
|
class Meta:
|
|
table = "invoices"
|
|
indexes = [("user_id", "status", "created_at")]
|
|
|
|
class InvoiceLineItem(models.Model):
|
|
id = fields.IntField(pk=True)
|
|
invoice = fields.ForeignKeyField("billing.Invoice", related_name="line_items")
|
|
description = fields.TextField()
|
|
quantity = fields.DecimalField(max_digits=15, decimal_places=6)
|
|
unit_price = fields.DecimalField(max_digits=10, decimal_places=6)
|
|
amount = fields.DecimalField(max_digits=10, decimal_places=4)
|
|
item_type = fields.CharField(max_length=50, null=True)
|
|
metadata = fields.JSONField(null=True)
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
table = "invoice_line_items"
|
|
|
|
class PricingConfig(models.Model):
|
|
id = fields.IntField(pk=True)
|
|
config_key = fields.CharField(max_length=100, unique=True)
|
|
config_value = fields.DecimalField(max_digits=10, decimal_places=6)
|
|
description = fields.TextField(null=True)
|
|
unit = fields.CharField(max_length=50, null=True)
|
|
updated_by = fields.ForeignKeyField("models.User", related_name="pricing_updates", null=True)
|
|
updated_at = fields.DatetimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
table = "pricing_config"
|
|
|
|
class PaymentMethod(models.Model):
|
|
id = fields.IntField(pk=True)
|
|
user = fields.ForeignKeyField("models.User", related_name="payment_methods")
|
|
stripe_payment_method_id = fields.CharField(max_length=255)
|
|
type = fields.CharField(max_length=50)
|
|
is_default = fields.BooleanField(default=False)
|
|
last4 = fields.CharField(max_length=4, null=True)
|
|
brand = fields.CharField(max_length=50, null=True)
|
|
exp_month = fields.IntField(null=True)
|
|
exp_year = fields.IntField(null=True)
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
updated_at = fields.DatetimeField(auto_now=True)
|
|
|
|
class Meta:
|
|
table = "payment_methods"
|
|
|
|
class BillingEvent(models.Model):
|
|
id = fields.BigIntField(pk=True)
|
|
user = fields.ForeignKeyField("models.User", related_name="billing_events", null=True)
|
|
event_type = fields.CharField(max_length=100)
|
|
stripe_event_id = fields.CharField(max_length=255, unique=True, null=True)
|
|
data = fields.JSONField(null=True)
|
|
processed = fields.BooleanField(default=False)
|
|
created_at = fields.DatetimeField(auto_now_add=True)
|
|
|
|
class Meta:
|
|
table = "billing_events"
|