romaaaka/queuebot: Код бота для создания виртуальных очередей

src/database/models/user.py

30 строк1 KB

Roman

fix cascade deleting

25 апр 2025, 16:19

25 апр 2025, 16:192f52002

from sqlalchemy import Column, BigInteger, Boolean, String, Sequence
from sqlalchemy.orm import relationship

from database.models.basemodel import BaseModel


class User(BaseModel):
    __tablename__ = 'user'

    # Columns
    id = Column(BigInteger, primary_key=True)
    is_blocked = Column(Boolean, default=False)
    is_admin = Column(Boolean, default=False)
    first_name = Column(String, nullable=True)
    last_name = Column(String, nullable=True)
    username = Column(String, nullable=True)

    # relationships
    chats = relationship('Chat', back_populates='owner', cascade="all, delete-orphan", passive_deletes=True)
    queues = relationship('Queue', back_populates='owner', cascade="all, delete-orphan", passive_deletes=True)
    permissions = relationship('QueuePermission', back_populates='user', cascade="all, delete-orphan",
                               passive_deletes=True)

    def __str__(self):
        if self.first_name or self.last_name:
            return f'{self.first_name or ""} {self.last_name or ""}'.strip()
        elif self.username:
            return f'@{self.username}'
        else:
            return f'[{self.id}]'