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)