from aiogram.types import MessageEntity
from pydantic import BaseModel
from sqlalchemy import JSON
from sqlalchemy.orm import Mapped, mapped_column

from models import BaseTable


class Message(BaseModel):
    text: str
    entities: list[MessageEntity] = []


class Announcement(BaseTable):
    __tablename__ = "announcement"

    id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
    message: Mapped[str] = mapped_column(JSON())