refactor: adds category to flash messages

This commit is contained in:
2024-11-30 00:56:40 -03:00
parent 512cec6a3c
commit 59a44f4ee2
3 changed files with 15 additions and 3 deletions

View File

@@ -1,4 +1,8 @@
from app.utils.flash_message_category import FlashMessageCategory
class FlashMessage:
def __init__(self, message, alertType):
def __init__(self, message, alertType, category: FlashMessageCategory = None):
self.message = message
self.alertType = alertType
self.alertType = alertType
self.category = category

View File

@@ -0,0 +1,5 @@
from enum import Enum
class FlashMessageCategory(Enum):
PERSONAL_INFO = "personal_info"
PASSWORD = "password"