Files
queuebot/bot/app/db/models.py
2023-06-09 18:21:36 +03:00

42 lines
1.2 KiB
Python

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, BigInteger, ForeignKey
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import relationship, backref
import uuid
from db.base import Base
class User(Base):
__tablename__ = "user"
id = Column(BigInteger, primary_key=True)
name = Column(String(64))
username = Column(String(32))
owns_queues = relationship("Queue", backref="owner")
takes_part_in_queues = relationship("QueueUser", backref="user")
class Queue(Base):
__tablename__ = "queue"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
name = Column(String(40), default="Новая очередь")
description = Column(String(120), default=None)
owner_id = Column(BigInteger, ForeignKey("user.id"))
users = relationship(
"QueueUser", backref="queue"
) # TODO: Delete all QueueUser user if Queue deletes
class QueueUser(Base):
__tablename__ = "queueuser"
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
user_id = Column(BigInteger, ForeignKey("user.id"))
queue_id = Column(UUID(as_uuid=True), ForeignKey("queue.id"))
position = Column(Integer)