mirror of
https://github.com/x1xhlol/system-prompts-and-models-of-ai-tools.git
synced 2026-06-18 15:29:36 +00:00
67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
import enum
|
|
from datetime import datetime, timezone
|
|
from sqlalchemy import Column, String, Float, Integer, Text, DateTime, Enum, ForeignKey
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import relationship
|
|
from app.models.base import TenantModel, BaseModel
|
|
|
|
|
|
class CommissionStatus(str, enum.Enum):
|
|
DRAFT = "draft"
|
|
PENDING = "pending"
|
|
APPROVED = "approved"
|
|
HELD = "held"
|
|
PAID = "paid"
|
|
REJECTED = "rejected"
|
|
DISPUTED = "disputed"
|
|
CLAWBACK = "clawback"
|
|
|
|
|
|
class PayoutStatus(str, enum.Enum):
|
|
PENDING = "pending"
|
|
PROCESSING = "processing"
|
|
PAID = "paid"
|
|
FAILED = "failed"
|
|
|
|
|
|
class Commission(TenantModel):
|
|
__tablename__ = "commissions"
|
|
|
|
affiliate_id = Column(UUID(as_uuid=True), ForeignKey("affiliate_marketers.id"), nullable=False, index=True)
|
|
deal_id = Column(UUID(as_uuid=True), ForeignKey("deals.id"), nullable=False, index=True)
|
|
payout_id = Column(UUID(as_uuid=True), ForeignKey("payouts.id"), nullable=True)
|
|
amount = Column(Float, nullable=False)
|
|
rate = Column(Float, nullable=False)
|
|
plan_type = Column(String(50), nullable=True)
|
|
status = Column(Enum(CommissionStatus), default=CommissionStatus.DRAFT, nullable=False)
|
|
approved_by = Column(UUID(as_uuid=True), ForeignKey("users.id"), nullable=True)
|
|
approved_at = Column(DateTime(timezone=True), nullable=True)
|
|
held_reason = Column(Text, nullable=True)
|
|
paid_at = Column(DateTime(timezone=True), nullable=True)
|
|
payment_reference = Column(String(100), nullable=True)
|
|
dispute_id = Column(UUID(as_uuid=True), ForeignKey("disputes.id"), nullable=True)
|
|
notes = Column(Text, nullable=True)
|
|
|
|
affiliate = relationship("AffiliateMarketer")
|
|
deal = relationship("Deal")
|
|
payout = relationship("Payout", back_populates="commissions")
|
|
approved_user = relationship("User", foreign_keys=[approved_by])
|
|
dispute = relationship("Dispute", foreign_keys=[dispute_id])
|
|
|
|
|
|
class Payout(BaseModel):
|
|
__tablename__ = "payouts"
|
|
|
|
affiliate_id = Column(UUID(as_uuid=True), ForeignKey("affiliate_marketers.id"), nullable=False, index=True)
|
|
total_amount = Column(Float, nullable=False)
|
|
commissions_count = Column(Integer, default=0)
|
|
status = Column(Enum(PayoutStatus), default=PayoutStatus.PENDING, nullable=False)
|
|
bank_name = Column(String(100), nullable=True)
|
|
bank_account = Column(String(50), nullable=True)
|
|
paid_at = Column(DateTime(timezone=True), nullable=True)
|
|
payment_reference = Column(String(100), nullable=True)
|
|
notes = Column(Text, nullable=True)
|
|
|
|
affiliate = relationship("AffiliateMarketer")
|
|
commissions = relationship("Commission", back_populates="payout")
|