"""
SQLAlchemy models
"""

from .base import Base, TimestampedMixin
from .user import User
from .streamer import (
    StreamerInfo,
    McnAgency,
    PlatformRecharge,
)
from .contract import Contract
from .order import Order
from .settlement import Settlement
from .expense import Expense
from .tax_declaration import TaxDeclaration
from .bank_transaction import BankTransaction
from .invoice import Invoice
from .risk_detection import (
    DetectionRule,
    DetectionTask,
    RuleExecution,
    TaskExecution,
    DetectionResult,
    RiskLevel,
    RuleStatus,
    TaskType,
    TaskStatus,
)

# 导入所有模型以确保它们被注册到Base.metadata
__all__ = [
    "Base",
    "TimestampedMixin",
    "User",
    "StreamerInfo",
    "McnAgency",
    "PlatformRecharge",
    "Contract",
    "Order",
    "Settlement",
    "Expense",
    "TaxDeclaration",
    "BankTransaction",
    "Invoice",
    "DetectionRule",
    "DetectionTask",
    "RuleExecution",
    "TaskExecution",
    "DetectionResult",
    "RiskLevel",
    "RuleStatus",
    "TaskType",
    "TaskStatus",
]