31 lines
1.4 KiB
Python
31 lines
1.4 KiB
Python
from __future__ import annotations
|
|
from datetime import datetime
|
|
from sqlalchemy import ForeignKey, String, BigInteger, Boolean, UniqueConstraint, func, DateTime
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
from app.db.session import Base
|
|
from app.models.user import User
|
|
from app.models.bot import Bot
|
|
|
|
class Channel(Base):
|
|
__tablename__ = "channels"
|
|
__table_args__ = (UniqueConstraint("owner_id", "chat_id", name="uq_owner_chat"),)
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
|
|
owner_id: Mapped[int] = mapped_column(ForeignKey("users.id", ondelete="CASCADE"))
|
|
bot_id: Mapped[int] = mapped_column(ForeignKey("bots.id", ondelete="CASCADE"))
|
|
chat_id: Mapped[int] = mapped_column(index=True)
|
|
title: Mapped[str | None] = mapped_column(String(128))
|
|
username: Mapped[str | None] = mapped_column(String(64))
|
|
created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), server_default=func.now())
|
|
|
|
owner = relationship("User")
|
|
bot = relationship("Bot", back_populates="channels")
|
|
|
|
class BotChannel(Base):
|
|
__tablename__ = "bot_channels"
|
|
|
|
id: Mapped[int] = mapped_column(primary_key=True)
|
|
bot_id: Mapped[int] = mapped_column(ForeignKey("bots.id", ondelete="CASCADE"))
|
|
channel_id: Mapped[int] = mapped_column(ForeignKey("channels.id", ondelete="CASCADE"))
|
|
can_post: Mapped[bool] = mapped_column(Boolean, default=True)
|