341 lines
12 KiB
Python
341 lines
12 KiB
Python
from django.conf import settings
|
|
from django.core.exceptions import ValidationError
|
|
from django.db import models
|
|
from django.db.models import Q
|
|
|
|
from apps.users.models import ApiKey
|
|
|
|
|
|
def normalize_resolution(value: str | None) -> str:
|
|
return "" if value is None else str(value).strip().upper()
|
|
|
|
|
|
class CallRecord(models.Model):
|
|
class OperationType(models.TextChoices):
|
|
TITLE = "title", "生成标题"
|
|
IMAGE = "image", "生成图片"
|
|
VISION = "vision", "图片理解"
|
|
|
|
class Status(models.TextChoices):
|
|
PENDING = "pending", "待处理"
|
|
SUCCESS = "success", "成功"
|
|
FAILED = "failed", "失败"
|
|
|
|
user = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
verbose_name="用户",
|
|
on_delete=models.PROTECT,
|
|
related_name="call_records",
|
|
)
|
|
api_key = models.ForeignKey(
|
|
ApiKey,
|
|
verbose_name="API 密钥",
|
|
null=True,
|
|
blank=True,
|
|
on_delete=models.SET_NULL,
|
|
related_name="call_records",
|
|
)
|
|
operation_type = models.CharField("操作类型", max_length=32, choices=OperationType.choices)
|
|
alias = models.CharField("能力别名", max_length=64, blank=True)
|
|
model_used = models.CharField("实际模型", max_length=128, blank=True)
|
|
resolution = models.CharField("分辨率", max_length=32, blank=True)
|
|
prompt = models.TextField("提示词", blank=True)
|
|
points_cost = models.BigIntegerField("扣费点数", default=0)
|
|
status = models.CharField(
|
|
"状态",
|
|
max_length=20,
|
|
choices=Status.choices,
|
|
default=Status.PENDING,
|
|
)
|
|
upstream_latency_ms = models.PositiveIntegerField("上游耗时毫秒", null=True, blank=True)
|
|
error_message = models.TextField("错误信息", blank=True)
|
|
result_ref = models.TextField("结果引用", blank=True)
|
|
result_summary = models.TextField("结果摘要", blank=True)
|
|
created_at = models.DateTimeField("创建时间", auto_now_add=True)
|
|
updated_at = models.DateTimeField("更新时间", auto_now=True)
|
|
|
|
class Meta:
|
|
db_table = "call_record"
|
|
verbose_name = "调用记录"
|
|
verbose_name_plural = "调用记录"
|
|
ordering = ("-created_at", "-id")
|
|
constraints = [
|
|
models.CheckConstraint(
|
|
condition=Q(points_cost__gte=0),
|
|
name="call_record_points_cost_non_negative",
|
|
),
|
|
]
|
|
indexes = [
|
|
models.Index(fields=("user", "created_at")),
|
|
models.Index(fields=("api_key", "created_at")),
|
|
models.Index(fields=("operation_type", "status")),
|
|
models.Index(fields=("alias",)),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.operation_type}:{self.alias or '<default>'} {self.status}"
|
|
|
|
|
|
class PricingRule(models.Model):
|
|
operation_type = models.CharField("操作类型", max_length=32, choices=CallRecord.OperationType.choices)
|
|
alias = models.CharField("能力别名", max_length=64)
|
|
resolution = models.CharField(
|
|
"分辨率",
|
|
max_length=32,
|
|
blank=True,
|
|
default="",
|
|
help_text="留空表示该别名所有分辨率通用。",
|
|
)
|
|
points_cost = models.BigIntegerField("点数单价")
|
|
is_active = models.BooleanField("启用", default=True)
|
|
created_at = models.DateTimeField("创建时间", auto_now_add=True)
|
|
updated_at = models.DateTimeField("更新时间", auto_now=True)
|
|
|
|
class Meta:
|
|
db_table = "pricing_rule"
|
|
verbose_name = "计费规则"
|
|
verbose_name_plural = "计费规则"
|
|
ordering = ("operation_type", "alias", "resolution")
|
|
constraints = [
|
|
models.UniqueConstraint(
|
|
fields=("operation_type", "alias", "resolution"),
|
|
name="unique_pricing_rule_per_alias_resolution",
|
|
),
|
|
models.CheckConstraint(
|
|
condition=Q(points_cost__gt=0),
|
|
name="pricing_rule_points_cost_positive",
|
|
),
|
|
]
|
|
indexes = [
|
|
models.Index(fields=("operation_type", "alias", "is_active")),
|
|
models.Index(fields=("operation_type", "alias", "resolution", "is_active")),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
resolution = self.resolution or "*"
|
|
return f"{self.operation_type}:{self.alias}:{resolution} = {self.points_cost}"
|
|
|
|
def clean(self) -> None:
|
|
super().clean()
|
|
self.alias = str(self.alias or "").strip()
|
|
self.resolution = normalize_resolution(self.resolution)
|
|
if not self.alias:
|
|
raise ValidationError({"alias": "Alias is required."})
|
|
|
|
def save(self, *args, **kwargs) -> None:
|
|
self.full_clean()
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class ExchangeRate(models.Model):
|
|
currency = models.CharField("币种", max_length=10, default="CNY")
|
|
points_per_unit = models.DecimalField("每单位金额点数", max_digits=12, decimal_places=4)
|
|
is_active = models.BooleanField("启用", default=True)
|
|
effective_from = models.DateTimeField("生效时间")
|
|
note = models.CharField("备注", max_length=255, blank=True)
|
|
created_at = models.DateTimeField("创建时间", auto_now_add=True)
|
|
updated_at = models.DateTimeField("更新时间", auto_now=True)
|
|
|
|
class Meta:
|
|
db_table = "exchange_rate"
|
|
verbose_name = "汇率"
|
|
verbose_name_plural = "汇率"
|
|
ordering = ("-effective_from", "-id")
|
|
constraints = [
|
|
models.CheckConstraint(
|
|
condition=Q(points_per_unit__gt=0),
|
|
name="exchange_rate_points_per_unit_positive",
|
|
),
|
|
]
|
|
indexes = [
|
|
models.Index(fields=("currency", "is_active", "effective_from")),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.currency} 1 = {self.points_per_unit} points"
|
|
|
|
def clean(self) -> None:
|
|
super().clean()
|
|
self.currency = str(self.currency or "").strip().upper()
|
|
if not self.currency:
|
|
raise ValidationError({"currency": "Currency is required."})
|
|
|
|
def save(self, *args, **kwargs) -> None:
|
|
self.full_clean()
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class RechargeOrder(models.Model):
|
|
class PayMethod(models.TextChoices):
|
|
WEIXIN = "weixin", "微信"
|
|
ALIPAY = "alipay", "支付宝"
|
|
|
|
class Status(models.TextChoices):
|
|
PENDING = "pending", "待支付"
|
|
PAID = "paid", "已支付"
|
|
FAILED = "failed", "失败"
|
|
EXPIRED = "expired", "已过期"
|
|
|
|
order_no = models.CharField("订单号", max_length=64, unique=True)
|
|
user = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
verbose_name="用户",
|
|
on_delete=models.PROTECT,
|
|
related_name="recharge_orders",
|
|
)
|
|
amount_money = models.DecimalField("金额", max_digits=12, decimal_places=2)
|
|
currency = models.CharField("币种", max_length=10, default="CNY")
|
|
pay_method = models.CharField("支付方式", max_length=20, choices=PayMethod.choices)
|
|
exchange_rate = models.DecimalField("锁定汇率", max_digits=12, decimal_places=4)
|
|
points_granted = models.BigIntegerField("到账点数")
|
|
status = models.CharField(
|
|
"状态",
|
|
max_length=20,
|
|
choices=Status.choices,
|
|
default=Status.PENDING,
|
|
)
|
|
code_url = models.TextField("二维码票据", blank=True)
|
|
expires_at = models.DateTimeField("过期时间", null=True, blank=True)
|
|
payment_txn_no = models.CharField("支付交易号", max_length=128, blank=True)
|
|
paid_at = models.DateTimeField("支付时间", null=True, blank=True)
|
|
created_at = models.DateTimeField("创建时间", auto_now_add=True)
|
|
updated_at = models.DateTimeField("更新时间", auto_now=True)
|
|
|
|
class Meta:
|
|
db_table = "recharge_order"
|
|
verbose_name = "充值订单"
|
|
verbose_name_plural = "充值订单"
|
|
ordering = ("-created_at", "-id")
|
|
constraints = [
|
|
models.CheckConstraint(
|
|
condition=Q(amount_money__gt=0),
|
|
name="recharge_order_amount_money_positive",
|
|
),
|
|
models.CheckConstraint(
|
|
condition=Q(exchange_rate__gt=0),
|
|
name="recharge_order_exchange_rate_positive",
|
|
),
|
|
models.CheckConstraint(
|
|
condition=Q(points_granted__gt=0),
|
|
name="recharge_order_points_granted_positive",
|
|
),
|
|
]
|
|
indexes = [
|
|
models.Index(fields=("user", "created_at")),
|
|
models.Index(fields=("status", "created_at")),
|
|
models.Index(fields=("pay_method", "status")),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.order_no} {self.status}"
|
|
|
|
def clean(self) -> None:
|
|
super().clean()
|
|
self.order_no = str(self.order_no or "").strip()
|
|
self.currency = str(self.currency or "").strip().upper()
|
|
if not self.order_no:
|
|
raise ValidationError({"order_no": "Order number is required."})
|
|
if not self.currency:
|
|
raise ValidationError({"currency": "Currency is required."})
|
|
|
|
def save(self, *args, **kwargs) -> None:
|
|
self.full_clean()
|
|
super().save(*args, **kwargs)
|
|
|
|
|
|
class PointsLedger(models.Model):
|
|
class ChangeType(models.TextChoices):
|
|
RECHARGE = "recharge", "充值"
|
|
CONSUME = "consume", "消费"
|
|
ADJUST = "adjust", "调整"
|
|
REFUND = "refund", "退款"
|
|
SIGNUP_BONUS = "signup_bonus", "注册赠点"
|
|
|
|
user = models.ForeignKey(
|
|
settings.AUTH_USER_MODEL,
|
|
verbose_name="用户",
|
|
on_delete=models.PROTECT,
|
|
related_name="points_ledger_entries",
|
|
)
|
|
change_type = models.CharField("变动类型", max_length=20, choices=ChangeType.choices)
|
|
points_delta = models.BigIntegerField("点数变动")
|
|
balance_after = models.BigIntegerField("变动后余额")
|
|
ref_order_id = models.PositiveBigIntegerField("关联订单 ID", null=True, blank=True)
|
|
ref_call = models.ForeignKey(
|
|
CallRecord,
|
|
verbose_name="关联调用",
|
|
null=True,
|
|
blank=True,
|
|
on_delete=models.PROTECT,
|
|
related_name="ledger_entries",
|
|
)
|
|
reason = models.TextField("原因", blank=True)
|
|
created_at = models.DateTimeField("创建时间", auto_now_add=True)
|
|
|
|
class Meta:
|
|
db_table = "points_ledger"
|
|
verbose_name = "点数流水"
|
|
verbose_name_plural = "点数流水"
|
|
ordering = ("-created_at", "-id")
|
|
constraints = [
|
|
models.CheckConstraint(
|
|
condition=Q(balance_after__gte=0),
|
|
name="points_ledger_balance_after_non_negative",
|
|
),
|
|
models.CheckConstraint(
|
|
condition=~Q(points_delta=0),
|
|
name="points_ledger_points_delta_non_zero",
|
|
),
|
|
models.UniqueConstraint(
|
|
fields=("ref_call", "change_type"),
|
|
name="unique_ledger_change_type_per_call",
|
|
),
|
|
models.UniqueConstraint(
|
|
fields=("ref_order_id", "change_type"),
|
|
name="unique_ledger_change_type_per_order",
|
|
),
|
|
]
|
|
indexes = [
|
|
models.Index(fields=("user", "created_at")),
|
|
models.Index(fields=("change_type", "created_at")),
|
|
models.Index(fields=("ref_order_id",)),
|
|
models.Index(fields=("ref_call",)),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.user} {self.change_type} {self.points_delta}"
|
|
|
|
def clean(self) -> None:
|
|
super().clean()
|
|
if self.change_type == self.ChangeType.ADJUST and not (self.reason or "").strip():
|
|
raise ValidationError({"reason": "Adjust ledger entries require a reason."})
|
|
|
|
|
|
class SignupBonusGrant(models.Model):
|
|
user = models.OneToOneField(
|
|
settings.AUTH_USER_MODEL,
|
|
verbose_name="用户",
|
|
on_delete=models.PROTECT,
|
|
related_name="signup_bonus_grant",
|
|
)
|
|
points_granted = models.BigIntegerField("赠送点数", default=100)
|
|
created_at = models.DateTimeField("创建时间", auto_now_add=True)
|
|
|
|
class Meta:
|
|
db_table = "signup_bonus_grant"
|
|
verbose_name = "注册赠点记录"
|
|
verbose_name_plural = "注册赠点记录"
|
|
ordering = ("-created_at", "-id")
|
|
constraints = [
|
|
models.CheckConstraint(
|
|
condition=Q(points_granted__gt=0),
|
|
name="signup_bonus_grant_points_positive",
|
|
),
|
|
]
|
|
indexes = [
|
|
models.Index(fields=("user", "created_at")),
|
|
]
|
|
|
|
def __str__(self) -> str:
|
|
return f"{self.user} signup bonus: {self.points_granted}"
|