1 files changed, 11 insertions, 2 deletions

diff --git a/models/invoce.py b/models/invoce.py
index 54a445e..66ff019 100644
--- a/models/invoce.py
+++ b/models/invoce.py
@@ -1,13 +1,22 @@
 from datetime import datetime
 
+from sqlalchemy import JSON
 from sqlalchemy.orm import Mapped, mapped_column
 
-from models import BaseTable
+from models import BaseTable, RichText
 
 
 class Invoice(BaseTable):
     __tablename__ = "invoice"
 
     id: Mapped[int] = mapped_column(primary_key=True, autoincrement=True)
-    amount: Mapped[float]
+    __message: Mapped[str] = mapped_column("message", JSON())
     datetime: Mapped[datetime]
+
+    @property
+    def message(self) -> RichText:
+        return RichText.model_validate_json(self.__message)
+
+    @message.setter
+    def message(self, value: RichText) -> None:
+        self.__message = value.model_dump_json()