chore: premetheus container will update after changing yml file

Author: codeperfectplus

src/routes/helper/prometheus_helper.py

@@ -1,8 +1,10 @@
 import os
 import yaml
+import subprocess
 from src.utils import ROOT_DIR
 
 prometheus_yml_path = os.path.join(ROOT_DIR, 'prometheus_config/prometheus.yml')
+update_prometheus_path = os.path.join(ROOT_DIR, 'src/scripts/update_prometheus.sh')
 
 def is_valid_file(file_path: str) -> bool:
     """Checks if a file is valid and has key-value pairs separated by a colon."""
@@ -38,3 +40,22 @@ def show_targets():
             'scrape_interval': scrape_interval
         })
     return targets_info
+
+
+def update_prometheus_container():
+    """Update the Prometheus container."""
+    # Define the path to your shell script
+    try:
+        # Use subprocess.run to execute the shell script
+        result = subprocess.run(['bash', update_prometheus_path], check=True, text=True, capture_output=True)
+
+        # Print the output of the script
+        print("Output:")
+        print(result.stdout)
+        
+        # Print any errors (if any)
+        if result.stderr:
+            print("Errors:")
+            print(result.stderr)
+    except subprocess.CalledProcessError as e:
+        print(f"An error occurred: {e}")

src/routes/prometheus.py

@@ -6,7 +6,14 @@
 from src.config import app, db
 from src.models import ExternalMonitornig
 from src.utils import ROOT_DIR
-from src.routes.helper.prometheus_helper import load_yaml, save_yaml, is_valid_file, show_targets, prometheus_yml_path
+from src.routes.helper.prometheus_helper import (
+    load_yaml, 
+    save_yaml, 
+    is_valid_file, 
+    show_targets, 
+    prometheus_yml_path,
+    update_prometheus_container)
+
 
 # Define the Prometheus Blueprint
 prometheus_bp = Blueprint('prometheus', __name__)
@@ -57,6 +64,7 @@ def targets():
     targets_info = show_targets()
     return render_template('other/targets.html', targets_info=targets_info)
 
+
 @app.route('/targets/add_target', methods=['POST'])
 def add_target():
     job_name = request.form.get('job_name')
@@ -75,7 +83,8 @@ def add_target():
             'scrape_interval': scrape_interval  # Set the specific interval
         }
         config['scrape_configs'].append(new_job)
-
+        
+    update_prometheus_container()
     save_yaml(config, prometheus_yml_path)
     flash('Target added successfully!', 'success')
     return redirect(url_for('targets'))
@@ -92,7 +101,8 @@ def remove_target():
             if target_to_remove in targets:
                 targets.remove(target_to_remove)
                 flash(f'Target {target_to_remove} removed successfully!', 'success')
-
+                
+                update_prometheus_container()
                 # Check if this was the last target, then remove the job
                 if not targets:  # If the list is now empty
                     config['scrape_configs'].remove(scrape_config)
@@ -116,6 +126,7 @@ def change_interval():
         if scrape_config['job_name'] == job_name:
             scrape_config['scrape_interval'] = new_interval
             flash('Scrape interval updated successfully!', 'success')
+            update_prometheus_container()
             break
 
     save_yaml(config, prometheus_yml_path)

src/scripts/update_prometheus.sh

@@ -1,8 +1,12 @@
 #!/bin/bash
 
 # Configuration
+CURRENT_SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
+CURRENT_DIR="$CURRENT_SCRIPT_DIR"
+ROOT_DIR="$(dirname "$(dirname "$CURRENT_DIR")")"
+PROMETHEUS_CONFIG_DIR="$ROOT_DIR/prometheus_config"
 CONTAINER_NAME="prometheus"
-PROMETHEUS_CONFIG="$(pwd)/prometheus.yml"
+PROMETHEUS_CONFIG="$PROMETHEUS_CONFIG_DIR/prometheus.yml"
 PROMETHEUS_IMAGE="prom/prometheus"  # Add your image name if needed
 NETWORK_NAME="flask-prometheus-net"  # Specify your network name
 PROMETHEUS_PORT="9090"  # Specify your port

src/static/css/targets.css

@@ -1,16 +1,12 @@
-* {
-    box-sizing: border-box;
-    margin: 0;
-    padding: 0;
-}
 
-body {
+
+/* body {
     font-family: 'Arial', sans-serif;
     background-color: #f4f4f9;
     color: #333;
     line-height: 1.6;
     padding: 20px;
-}
+} */
 
 h1, h2 {
     color: #4a4a4a;

src/templates/dashboard/developer.html

@@ -1,5 +1,5 @@
 {% extends 'base/base.html' %}
-{% block title %}Server Dashboard{% endblock %}
+{% block title %}{{ system_info['system_username'] }}@{{ system_info['nodename'] }} - SystemGuard{% endblock %}
 {% block extra_head %}
 <link rel="stylesheet" href="{{ url_for('static', filename='css/refresher.css') }}">
 {% endblock %}