Add initial database migration

1 files changed, 10 insertions, 9 deletions

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()