From 85a37e4a7863b302c70b1fc7b0af27c37d75a7d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mateusz=20Gruszczy=C5=84ski?=
 <mateusz.gruszczynski@firma.interia.pl>
Date: Fri, 28 Feb 2025 16:28:13 +0100
Subject: [PATCH] ! RFACTOR TEMPLATE2

---
 app.py              | 13 +++++++++++++
 templates/base.html | 24 +++++++++++-------------
 2 files changed, 24 insertions(+), 13 deletions(-)

diff --git a/app.py b/app.py
index 2e143d6..96c2385 100644
--- a/app.py
+++ b/app.py
@@ -427,6 +427,19 @@ def format_version(value):
 
     return result
 
+@app.template_global()
+def bootstrap_alert_category(cat):
+    mapping = {
+        'error': 'danger',
+        'fail': 'danger',
+        'warning': 'warning',
+        'warn': 'warning',
+        'ok': 'success',
+        'success': 'success',
+        'info': 'info'
+    }
+    return mapping.get(cat.lower(), 'info')
+
 def fetch_changelogs(force=False):
     changelog_url = "https://mikrotik.com/download/changelogs"
     current_date = datetime.utcnow()
diff --git a/templates/base.html b/templates/base.html
index 8284240..37e5b06 100644
--- a/templates/base.html
+++ b/templates/base.html
@@ -219,19 +219,17 @@
     <!-- Kontener do wyświetlania komunikatów flash -->
     <div class="container mt-3">
       {% with messages = get_flashed_messages(with_categories=True) %}
-        {% if messages %}
-          {% for category, message in messages %}
-            <!-- Jeśli brak kategorii, używamy 'info' -->
-            {% if category == 'message' %}
-              {% set category = 'info' %}
-            {% endif %}
-            <div class="alert alert-{{ category }} alert-dismissible fade show" role="alert">
-              {{ message }}
-              <button type="button" class="btn-close" data-bs-dismiss="alert" aria-label="Close"></button>
-            </div>
-          {% endfor %}
-        {% endif %}
-      {% endwith %}
+      {% if messages %}
+        {% for category, message in messages %}
+          <!-- Zmapuj do stylu Bootstrapa -->
+          {% set bs_cat = bootstrap_alert_category(category) %}
+          <div class="alert alert-{{ bs_cat }} alert-dismissible fade show" role="alert">
+            {{ message }}
+            <button type="button" class="btn-close" data-bs-dismiss="alert" aria-label="Close"></button>
+          </div>
+        {% endfor %}
+      {% endif %}
+    {% endwith %}    
     </div>
 
     <main class="container my-4 flex-fill">