refactor: adds category to flash messages
This commit is contained in:
@@ -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
|
||||
5
app/utils/flash_message_category.py
Normal file
5
app/utils/flash_message_category.py
Normal file
@@ -0,0 +1,5 @@
|
||||
from enum import Enum
|
||||
|
||||
class FlashMessageCategory(Enum):
|
||||
PERSONAL_INFO = "personal_info"
|
||||
PASSWORD = "password"
|
||||
Reference in New Issue
Block a user