Add Message model
This commit is contained in:
+11
-4
@@ -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",
|
||||
]
|
||||
|
||||
@@ -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)
|
||||
Reference in New Issue
Block a user