From 0b2574c7c2d4993ed3e684007771e853d22c3916 Mon Sep 17 00:00:00 2001 From: Esa Kataja Date: Sat, 9 Nov 2024 16:23:19 +0200 Subject: [PATCH] Add Message model --- src/models/__init__.py | 15 +++++++++++---- src/models/general.py | 20 ++++++++++++++++++++ 2 files changed, 31 insertions(+), 4 deletions(-) create mode 100644 src/models/general.py diff --git a/src/models/__init__.py b/src/models/__init__.py index 659c8af..30d1986 100644 --- a/src/models/__init__.py +++ b/src/models/__init__.py @@ -1,4 +1,11 @@ -from .user import User -from .movie import Movie -from .score import Score -from .card import Card +# from .movie import Movie +# from .score import Score +# from .card import Card +from .general import Message + +__all__ = [ + # "Movie", + # "Score", + # "Card", + "Message", +] diff --git a/src/models/general.py b/src/models/general.py new file mode 100644 index 0000000..a58ad13 --- /dev/null +++ b/src/models/general.py @@ -0,0 +1,20 @@ +from pydantic import BaseModel, Field +from enum import Enum +from datetime import datetime + + +class MessageType(str, Enum): + """Message type enum""" + + success = "success" + info = "info" + warning = "warning" + error = "error" + + +class Message(BaseModel): + """Message model""" + + message: str + message_type: MessageType = Field(default=MessageType.success) + timestamp: datetime = Field(default_factory=datetime.utcnow)