161 строка
8.5 KiB
Python
161 строка
8.5 KiB
Python
from datetime import datetime
|
|
from sqlalchemy import (
|
|
String, Integer, Float, Boolean, DateTime,
|
|
ForeignKey, Text, BigInteger, Enum,
|
|
)
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
from database import Base
|
|
import enum
|
|
|
|
|
|
class PlanEnum(str, enum.Enum):
|
|
trial_week = "trial_week"
|
|
trial = "trial"
|
|
basic = "basic"
|
|
pro = "pro"
|
|
enterprise = "enterprise"
|
|
postpaid_custom = "postpaid_custom"
|
|
|
|
|
|
class OrderStatus(str, enum.Enum):
|
|
pending = "pending"
|
|
confirming = "confirming"
|
|
completed = "completed"
|
|
rejected = "rejected"
|
|
cancelled = "cancelled"
|
|
|
|
|
|
class Tenant(Base):
|
|
__tablename__ = "tenants"
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
name: Mapped[str] = mapped_column(String(128))
|
|
email: Mapped[str] = mapped_column(String(256), unique=True)
|
|
password_hash: Mapped[str] = mapped_column(String(256))
|
|
telegram_id: Mapped[int|None] = mapped_column(BigInteger, nullable=True, index=True)
|
|
plan: Mapped[PlanEnum] = mapped_column(Enum(PlanEnum), default=PlanEnum.trial_week)
|
|
trial_ends_at: Mapped[datetime|None] = mapped_column(DateTime, nullable=True)
|
|
postpaid_commission_percent: Mapped[float] = mapped_column(Float, default=5.0)
|
|
postpaid_due_day: Mapped[int] = mapped_column(Integer, default=5)
|
|
postpaid_note: Mapped[str] = mapped_column(String(512), default="")
|
|
postpaid_enabled_at: Mapped[datetime|None] = mapped_column(DateTime, nullable=True)
|
|
balance: Mapped[float] = mapped_column(Float, default=0.0)
|
|
alert_threshold: Mapped[float] = mapped_column(Float, default=200.0)
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
is_blocked: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
|
|
|
shops: Mapped[list["Shop"]] = relationship(back_populates="tenant")
|
|
transactions: Mapped[list["BalanceTransaction"]] = relationship(back_populates="tenant")
|
|
|
|
|
|
class Shop(Base):
|
|
__tablename__ = "shops"
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id"), index=True)
|
|
name: Mapped[str] = mapped_column(String(128))
|
|
ctrl_bot_token: Mapped[str] = mapped_column(String(128))
|
|
ctrl_bot_username:Mapped[str] = mapped_column(String(64), default="")
|
|
welcome_msg: Mapped[str] = mapped_column(Text, default="Добро пожаловать!")
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
|
|
|
tenant: Mapped["Tenant"] = relationship(back_populates="shops")
|
|
tokens: Mapped[list["ShopToken"]] = relationship(back_populates="shop")
|
|
products: Mapped[list["Product"]] = relationship(back_populates="shop")
|
|
cards: Mapped[list["PaymentCard"]] = relationship(back_populates="shop")
|
|
orders: Mapped[list["Order"]] = relationship(back_populates="shop")
|
|
team: Mapped[list["ShopMember"]] = relationship(back_populates="shop")
|
|
|
|
|
|
class ShopToken(Base):
|
|
__tablename__ = "shop_tokens"
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
shop_id: Mapped[int] = mapped_column(ForeignKey("shops.id"), index=True)
|
|
token: Mapped[str] = mapped_column(String(128))
|
|
username: Mapped[str] = mapped_column(String(64), default="")
|
|
note: Mapped[str] = mapped_column(String(64), default="Основной")
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=False)
|
|
added_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
|
|
|
shop: Mapped["Shop"] = relationship(back_populates="tokens")
|
|
|
|
|
|
class ShopMember(Base):
|
|
__tablename__ = "shop_members"
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
shop_id: Mapped[int] = mapped_column(ForeignKey("shops.id"), index=True)
|
|
telegram_id: Mapped[int] = mapped_column(BigInteger, index=True)
|
|
username: Mapped[str] = mapped_column(String(64), default="")
|
|
name: Mapped[str] = mapped_column(String(128))
|
|
role: Mapped[str] = mapped_column(String(32), default="moderator")
|
|
added_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
|
|
|
shop: Mapped["Shop"] = relationship(back_populates="team")
|
|
|
|
|
|
class Product(Base):
|
|
__tablename__ = "products"
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
shop_id: Mapped[int] = mapped_column(ForeignKey("shops.id"), index=True)
|
|
name: Mapped[str] = mapped_column(String(256))
|
|
category: Mapped[str] = mapped_column(String(64), default="Общее")
|
|
description: Mapped[str] = mapped_column(Text, default="")
|
|
content: Mapped[str] = mapped_column(Text, default="")
|
|
photo_url: Mapped[str|None] = mapped_column(String(512), nullable=True)
|
|
price: Mapped[float] = mapped_column(Float)
|
|
stock: Mapped[int] = mapped_column(Integer, default=999)
|
|
sold: Mapped[int] = mapped_column(Integer, default=0)
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
|
|
|
|
shop: Mapped["Shop"] = relationship(back_populates="products")
|
|
|
|
|
|
class PaymentCard(Base):
|
|
__tablename__ = "payment_cards"
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
shop_id: Mapped[int] = mapped_column(ForeignKey("shops.id"), index=True)
|
|
bank: Mapped[str] = mapped_column(String(64))
|
|
number: Mapped[str] = mapped_column(String(32))
|
|
holder: Mapped[str] = mapped_column(String(128))
|
|
phone: Mapped[str] = mapped_column(String(32), default="")
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True)
|
|
orders_count: Mapped[int] = mapped_column(Integer, default=0)
|
|
received_total: Mapped[float] = mapped_column(Float, default=0.0)
|
|
|
|
shop: Mapped["Shop"] = relationship(back_populates="cards")
|
|
|
|
|
|
class Order(Base):
|
|
__tablename__ = "orders"
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
shop_id: Mapped[int] = mapped_column(ForeignKey("shops.id"), index=True)
|
|
product_id: Mapped[int] = mapped_column(ForeignKey("products.id"))
|
|
card_id: Mapped[int|None] = mapped_column(ForeignKey("payment_cards.id"), nullable=True)
|
|
buyer_telegram_id: Mapped[int] = mapped_column(BigInteger, index=True)
|
|
buyer_username: Mapped[str] = mapped_column(String(64), default="")
|
|
amount: Mapped[float] = mapped_column(Float)
|
|
commission: Mapped[float] = mapped_column(Float, default=0.0)
|
|
status: Mapped[OrderStatus] = mapped_column(Enum(OrderStatus), default=OrderStatus.pending, index=True)
|
|
proof_file_id: Mapped[str|None] = mapped_column(String(256), nullable=True)
|
|
confirmed_by: Mapped[int|None] = mapped_column(BigInteger, nullable=True)
|
|
product_content: Mapped[str] = mapped_column(Text, default="")
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, index=True)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
shop: Mapped["Shop"] = relationship(back_populates="orders")
|
|
|
|
|
|
class BalanceTransaction(Base):
|
|
__tablename__ = "balance_transactions"
|
|
id: Mapped[int] = mapped_column(Integer, primary_key=True)
|
|
tenant_id: Mapped[int] = mapped_column(ForeignKey("tenants.id"), index=True)
|
|
type: Mapped[str] = mapped_column(String(32)) # deposit / commission / refund
|
|
amount: Mapped[float] = mapped_column(Float)
|
|
balance_after: Mapped[float] = mapped_column(Float)
|
|
note: Mapped[str] = mapped_column(String(512), default="")
|
|
order_id: Mapped[int|None] = mapped_column(Integer, nullable=True)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, index=True)
|
|
|
|
tenant: Mapped["Tenant"] = relationship(back_populates="transactions")
|
© 2026 StackWorks