From bc7f486aa7b543a934f4cf23dc80a95f44afcb64 Mon Sep 17 00:00:00 2001 From: Tolmachev Igor Date: Mon, 23 Mar 2026 00:07:08 +0300 Subject: Add initial database migration --- models/announcement.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) (limited to 'models/announcement.py') diff --git a/models/announcement.py b/models/announcement.py index 5f752d7..9a6eef2 100644 --- a/models/announcement.py +++ b/models/announcement.py @@ -1,18 +1,19 @@ -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] = [] +from models import BaseTable, RichText class Announcement(BaseTable): __tablename__ = "announcement" id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True) - message: Mapped[str] = mapped_column(JSON()) + __message: Mapped[str] = mapped_column("message", JSON()) + + @property + def message(self) -> RichText: + return RichText.model_validate_json(self.__message) + + @message.setter + def message_set(self, value: RichText) -> None: + self.__message = value.model_dump_json() -- cgit v1.3