From 59a44f4ee2c9f68c27a4e8408cbf0150eadc91d3 Mon Sep 17 00:00:00 2001
From: Gabriel De Los Rios <gabriel.delosrios@tutamail.com>
Date: Sat, 30 Nov 2024 00:56:40 -0300
Subject: [PATCH] refactor: adds category to flash messages

---
 app/__init__.py                     | 5 ++++-
 app/utils/flash_message.py          | 8 ++++++--
 app/utils/flash_message_category.py | 5 +++++
 3 files changed, 15 insertions(+), 3 deletions(-)
 create mode 100644 app/utils/flash_message_category.py

diff --git a/app/__init__.py b/app/__init__.py
index d3988d4..864d7e4 100644
--- a/app/__init__.py
+++ b/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)
diff --git a/app/utils/flash_message.py b/app/utils/flash_message.py
index d8ac145..779865e 100644
--- a/app/utils/flash_message.py
+++ b/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
\ No newline at end of file
+    self.alertType = alertType
+    self.category = category
\ No newline at end of file
diff --git a/app/utils/flash_message_category.py b/app/utils/flash_message_category.py
new file mode 100644
index 0000000..1af0474
--- /dev/null
+++ b/app/utils/flash_message_category.py
@@ -0,0 +1,5 @@
+from enum import Enum
+
+class FlashMessageCategory(Enum):
+    PERSONAL_INFO = "personal_info"
+    PASSWORD = "password"
\ No newline at end of file