95 lines
3.7 KiB
Python
95 lines
3.7 KiB
Python
from sqlalchemy import Column, String, Numeric, DateTime, Boolean, ForeignKey, Integer, Enum, Text
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
from sqlalchemy.ext.asyncio import AsyncSession, create_async_engine
|
|
from sqlalchemy.orm import declarative_base, relationship, sessionmaker
|
|
from datetime import datetime
|
|
from enum import Enum as PyEnum
|
|
import uuid
|
|
|
|
Base = declarative_base()
|
|
|
|
def generate_uuid():
|
|
return str(uuid.uuid4())
|
|
|
|
|
|
class TicketStatus(PyEnum):
|
|
OPEN = "open"
|
|
PENDING = "pending"
|
|
CLOSED = "closed"
|
|
|
|
|
|
"""Пользователи"""
|
|
class User(Base):
|
|
__tablename__ = 'users'
|
|
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=generate_uuid)
|
|
telegram_id = Column(String, unique=True, nullable=False) # telegram_id как уникальный идентификатор
|
|
username = Column(String)
|
|
balance = Column(Numeric(10, 2), default=0.0)
|
|
referrer_id = Column(String, ForeignKey('users.telegram_id'), nullable=True) # Ссылка на telegram_id
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
referrals = relationship("User", backref="referrer", remote_side=[telegram_id]) # Ссылка на telegram_id
|
|
subscriptions = relationship("Subscription", back_populates="user")
|
|
transactions = relationship("Transaction", back_populates="user")
|
|
admins = relationship("Administrators", back_populates="user")
|
|
|
|
|
|
"""Подписки"""
|
|
class Subscription(Base):
|
|
__tablename__ = 'subscriptions'
|
|
|
|
id = Column(String, primary_key=True, default=generate_uuid)
|
|
user_id = Column(UUID(as_uuid=True), ForeignKey('users.id'))
|
|
vpn_server_id = Column(String)
|
|
plan = Column(String)
|
|
expiry_date = Column(DateTime)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
|
|
user = relationship("User", back_populates="subscriptions")
|
|
|
|
"""Транзакции"""
|
|
class Transaction(Base):
|
|
__tablename__ = 'transactions'
|
|
|
|
id = Column(String, primary_key=True, default=generate_uuid)
|
|
user_id = Column(UUID(as_uuid=True), ForeignKey('users.id'))
|
|
amount = Column(Numeric(10, 2))
|
|
transaction_type = Column(String)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
|
|
user = relationship("User", back_populates="transactions")
|
|
|
|
"""Тикет"""
|
|
class SupportTicket(Base):
|
|
__tablename__ = "support_tickets"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
user_id = Column(UUID(as_uuid=True), ForeignKey("users.id"), nullable=False)
|
|
subject = Column(String, nullable=False)
|
|
message = Column(String, nullable=False)
|
|
status = Column(Enum(TicketStatus), default=TicketStatus.OPEN, nullable=False)
|
|
created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=False)
|
|
|
|
"""Сообщения из тикетов"""
|
|
class TicketMessage(Base):
|
|
__tablename__ = "ticket_messages"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
ticket_id = Column(Integer, ForeignKey("support_tickets.id"), nullable=False) # ID тикета
|
|
sender = Column(String, nullable=False) # "user" или "support"
|
|
message = Column(Text, nullable=False) # Текст сообщения
|
|
created_at = Column(DateTime, default=datetime.utcnow, nullable=False) #
|
|
|
|
"""Администраторы"""
|
|
class Administrators(Base):
|
|
__tablename__ = 'admins'
|
|
|
|
id = Column(String, primary_key=True, default=generate_uuid)
|
|
user_id = Column(UUID(as_uuid=True), ForeignKey('users.id'))
|
|
|
|
user = relationship("User", back_populates="admins")
|