chore: bug fixes

Signed-off-by: deepak-linux <[email protected]>

Author: codeperfectplus

src/models/user_profile.py

@@ -18,7 +18,7 @@ class UserProfile(db.Model, UserMixin):
     __tablename__ = 'users'
 
     id = db.Column(db.Integer, primary_key=True)
-    username = db.Column(db.String(50), unique=True, nullable=False)
+    username = db.Column(db.String(50), index=True, unique=True, nullable=False)
     email = db.Column(db.String(100), unique=True, nullable=False)
     password = db.Column(db.String(100), nullable=False)
     user_level = db.Column(db.String(10), nullable=False, default='user')

src/routes/prometheus.py

@@ -5,6 +5,8 @@
 from collections import OrderedDict
 from werkzeug.security import check_password_hash
 
+from functools import lru_cache
+from flask import g  # 'g' is a request-specific object
 from src.config import app, db
 from src.models import ExternalMonitornig, UserProfile
 from src.utils import ROOT_DIR
@@ -18,14 +20,21 @@
     update_prometheus_container,
     update_prometheus_config)
 
-
 # Define the Prometheus Blueprint
 prometheus_bp = Blueprint('prometheus', __name__)
 
+# Cache user queries with LRU cache (memory-based, not ideal for distributed apps)
+@lru_cache(maxsize=128)
+def get_user_by_username(username):
+    print('Querying the database...')
+    return UserProfile.query.filter_by(username=username).first()
+
+# Verify user login information
 def verify_user(username, password):
-    user = UserProfile.query.filter_by(username=username).first()
+    user = get_user_by_username(username)
     if user and check_password_hash(user.password, password):
         return True
+    return False
 
 # Define a route to serve Prometheus metrics
 @app.route('/metrics')

src/scripts/prometheus.sh

@@ -89,8 +89,8 @@ run_output=$(docker run -d \
   -p "$PROMETHEUS_PORT:$PROMETHEUS_PORT" \
   --restart always \
   -v "$PROMETHEUS_CONFIG_FILE:/etc/prometheus/prometheus.yml" \
-  -v "PROMETHEUS_DATA_DIR:/prometheus" \
-  "$PROMETHEUS_IMAGE" 2>&1)  # Capture both stdout and stderr
+  -v "$PROMETHEUS_DATA_DIR:/prometheus" \
+  "$PROMETHEUS_IMAGE" 2>&1)
 
 # Check if Prometheus started successfully
 if [ $? -eq 0 ]; then

src/scripts/update_prometheus.sh

@@ -51,7 +51,7 @@ else
       -p "$PROMETHEUS_PORT:$PROMETHEUS_PORT" \
       --restart always \
       -v "$PROMETHEUS_CONFIG:/etc/prometheus/prometheus.yml" \
-      -v "PROMETHEUS_DATA_DIR:/prometheus" \
+      -v "$PROMETHEUS_DATA_DIR:/prometheus" \
       "$PROMETHEUS_IMAGE" &> /dev/null
 
     log "Prometheus container started successfully."