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 import uuid from db.base import Base class User(Base): __tablename__ = "user" id = Column(BigInteger, primary_key=True) name = Column(String) username = Column(String) 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) owner_id = Column(BigInteger, ForeignKey("user.id")) users = relationship("QueueUser", backref="queue") 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"))