100 lines
4.0 KiB
Python
100 lines
4.0 KiB
Python
from sqlalchemy import Column, String, Numeric, DateTime, ForeignKey, Integer, Enum, Text, BigInteger
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.orm import declarative_base, relationship
|
|
from datetime import datetime
|
|
from enum import Enum as PyEnum
|
|
import uuid
|
|
|
|
Base = declarative_base()
|
|
|
|
class SubscriptionStatus(PyEnum):
|
|
ACTIVE = "active"
|
|
EXPIRED = "expired"
|
|
CANCELLED = "cancelled"
|
|
|
|
class TransactionStatus(PyEnum):
|
|
PENDING = "pending"
|
|
SUCCESS = "success"
|
|
FAILED = "failed"
|
|
|
|
class TransactionType(PyEnum):
|
|
DEPOSIT = "deposit"
|
|
WITHDRAWAL = "withdrawal"
|
|
PAYMENT = "payment"
|
|
|
|
# Пользователи
|
|
class User(Base):
|
|
__tablename__ = 'users'
|
|
|
|
telegram_id = Column(BigInteger, primary_key=True)
|
|
username = Column(String(255))
|
|
balance = Column(Numeric(10, 2), default=0.0)
|
|
ref_code = Column(String(7), unique=True) # Реферальный код пользователя
|
|
invited_by = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=True)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
# Relationships
|
|
subscriptions = relationship("Subscription", back_populates="user")
|
|
transactions = relationship("Transaction", back_populates="user")
|
|
sent_referrals = relationship("Referral",
|
|
foreign_keys="Referral.inviter_id",
|
|
back_populates="inviter")
|
|
received_referrals = relationship("Referral",
|
|
foreign_keys="Referral.invited_id",
|
|
back_populates="invited")
|
|
|
|
# Реферальные связи
|
|
class Referral(Base):
|
|
__tablename__ = 'referrals'
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
inviter_id = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=False)
|
|
invited_id = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=False)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
inviter = relationship("User", foreign_keys=[inviter_id], back_populates="sent_referrals")
|
|
invited = relationship("User", foreign_keys=[invited_id], back_populates="received_referrals")
|
|
|
|
# Тарифные планы
|
|
class Plan(Base):
|
|
__tablename__ = 'plans'
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(String(100), nullable=False)
|
|
price = Column(Numeric(10, 2), nullable=False)
|
|
duration_days = Column(Integer, nullable=False)
|
|
description = Column(Text)
|
|
|
|
subscriptions = relationship("Subscription", back_populates="plan")
|
|
|
|
# Подписки
|
|
class Subscription(Base):
|
|
__tablename__ = 'subscriptions'
|
|
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
user_id = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=False)
|
|
plan_id = Column(Integer, ForeignKey('plans.id'), nullable=False)
|
|
vpn_server_id = Column(String) # ID сервера в Marzban
|
|
status = Column(Enum(SubscriptionStatus), default=SubscriptionStatus.ACTIVE)
|
|
start_date = Column(DateTime, nullable=False)
|
|
end_date = Column(DateTime, nullable=False)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
user = relationship("User", back_populates="subscriptions")
|
|
plan = relationship("Plan", back_populates="subscriptions")
|
|
|
|
# Транзакции
|
|
class Transaction(Base):
|
|
__tablename__ = 'transactions'
|
|
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
|
user_id = Column(BigInteger, ForeignKey('users.telegram_id'), nullable=False)
|
|
amount = Column(Numeric(10, 2), nullable=False)
|
|
status = Column(Enum(TransactionStatus), default=TransactionStatus.PENDING)
|
|
type = Column(Enum(TransactionType), nullable=False)
|
|
payment_provider = Column(String(100))
|
|
payment_id = Column(String, unique=True) # ID платежа в внешней системе
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
user = relationship("User", back_populates="transactions") |