99 lines
4.7 KiB
Python
99 lines
4.7 KiB
Python
"""激励域：referrals / asset_wallets / asset_transactions / withdrawals / video_earnings"""
|
|
from datetime import datetime
|
|
from decimal import Decimal
|
|
|
|
from sqlalchemy import DateTime, Index, Numeric, SmallInteger, String, func
|
|
from sqlalchemy.dialects.mysql import BIGINT
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.models.base import Base
|
|
|
|
|
|
class Referral(Base):
|
|
__tablename__ = "referrals"
|
|
|
|
id: Mapped[int] = mapped_column(BIGINT(unsigned=True), primary_key=True, autoincrement=True)
|
|
inviter_id: Mapped[int] = mapped_column(BIGINT(unsigned=True), comment="推荐人")
|
|
invitee_id: Mapped[int] = mapped_column(BIGINT(unsigned=True), comment="被推荐人")
|
|
invite_code: Mapped[str] = mapped_column(String(10))
|
|
status: Mapped[int] = mapped_column(SmallInteger, default=0, comment="0待激活 1有效 2无效 3风控冻结")
|
|
risk_flag: Mapped[str] = mapped_column(String(50), default="")
|
|
register_ip: Mapped[str] = mapped_column(String(45), default="")
|
|
activated_at: Mapped[datetime | None] = mapped_column(DateTime)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
|
|
|
|
__table_args__ = (
|
|
Index("uk_invitee", "invitee_id", unique=True),
|
|
Index("idx_inviter", "inviter_id", "created_at"),
|
|
Index("idx_status", "status", "created_at"),
|
|
)
|
|
|
|
|
|
class AssetWallet(Base):
|
|
__tablename__ = "asset_wallets"
|
|
|
|
user_id: Mapped[int] = mapped_column(BIGINT(unsigned=True), primary_key=True)
|
|
assets: Mapped[Decimal] = mapped_column(Numeric(12, 1), default=0, comment="当前资产(允许为负)")
|
|
total_earned: Mapped[Decimal] = mapped_column(Numeric(14, 1), default=0)
|
|
total_withdrawn: Mapped[Decimal] = mapped_column(Numeric(14, 1), default=0)
|
|
version: Mapped[int] = mapped_column(default=0)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now())
|
|
|
|
|
|
class AssetTransaction(Base):
|
|
__tablename__ = "asset_transactions"
|
|
|
|
id: Mapped[int] = mapped_column(BIGINT(unsigned=True), primary_key=True, autoincrement=True)
|
|
user_id: Mapped[int] = mapped_column(BIGINT(unsigned=True))
|
|
change_amount: Mapped[Decimal] = mapped_column(Numeric(12, 1))
|
|
balance_after: Mapped[Decimal] = mapped_column(Numeric(12, 1))
|
|
biz_type: Mapped[int] = mapped_column(
|
|
SmallInteger, comment="1充值返利 2完整观看 3点赞 4提现 5提现退回 6返利扣回 7后台调整"
|
|
)
|
|
biz_id: Mapped[str] = mapped_column(String(64), default="")
|
|
remark: Mapped[str] = mapped_column(String(200), default="")
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
|
|
|
|
__table_args__ = (
|
|
Index("idx_user_created", "user_id", "created_at"),
|
|
Index("idx_biz", "biz_type", "biz_id"),
|
|
)
|
|
|
|
|
|
class Withdrawal(Base):
|
|
__tablename__ = "withdrawals"
|
|
|
|
id: Mapped[int] = mapped_column(BIGINT(unsigned=True), primary_key=True, autoincrement=True)
|
|
withdraw_no: Mapped[str] = mapped_column(String(32), unique=True)
|
|
user_id: Mapped[int] = mapped_column(BIGINT(unsigned=True))
|
|
assets: Mapped[Decimal] = mapped_column(Numeric(12, 1))
|
|
amount_fen: Mapped[int] = mapped_column(comment="折合人民币(分)")
|
|
channel: Mapped[int] = mapped_column(SmallInteger, comment="1支付宝 2微信 3USDT")
|
|
account: Mapped[str] = mapped_column(String(200))
|
|
status: Mapped[int] = mapped_column(SmallInteger, default=0, comment="0待审核 1已打款 2已拒绝")
|
|
reason: Mapped[str] = mapped_column(String(200), default="")
|
|
admin_id: Mapped[int | None] = mapped_column(BIGINT(unsigned=True))
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now())
|
|
processed_at: Mapped[datetime | None] = mapped_column(DateTime)
|
|
|
|
__table_args__ = (
|
|
Index("idx_user", "user_id", "created_at"),
|
|
Index("idx_status", "status", "created_at"),
|
|
)
|
|
|
|
|
|
class VideoEarning(Base):
|
|
"""视频收益结算刻度（完整观看/点赞按增量结算，余数保留）"""
|
|
|
|
__tablename__ = "video_earnings"
|
|
|
|
video_id: Mapped[int] = mapped_column(BIGINT(unsigned=True), primary_key=True)
|
|
author_id: Mapped[int] = mapped_column(BIGINT(unsigned=True))
|
|
complete_views: Mapped[int] = mapped_column(default=0, comment="累计完整观看次数")
|
|
views_settled: Mapped[int] = mapped_column(default=0, comment="已结算的完整观看刻度")
|
|
likes_accrued: Mapped[int] = mapped_column(default=0, comment="累计点赞(取消会回退)")
|
|
likes_settled: Mapped[int] = mapped_column(default=0, comment="已结算的点赞刻度")
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, server_default=func.now(), onupdate=func.now())
|
|
|
|
__table_args__ = (Index("idx_author", "author_id"),)
|