refactor: adds category to flash messages

app/__init__.py

@@ -1,6 +1,7 @@
 from flask import Flask
 from flask_session import Session
 
+from app.utils.flash_message_category import FlashMessageCategory
 from config import Config
 from app.extensions import db
 from app.utils import DBUtils, ROUTES
@@ -16,7 +17,9 @@ def create_app(config_class=Config):
     def inject_routes():
         """Inject ROUTES dictionary into all templates."""
         return {'ROUTES': ROUTES}
-    
+    @app.context_processor
+    def inject_enums():
+        return dict(FlashMessageCategory=FlashMessageCategory)
     #models_setup()
     DBUtils.drop_db(app)
     DBUtils.setup_db(app)

app/utils/flash_message.py

@@ -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.category = category

app/utils/flash_message_category.py

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