option added to restart prometheus

Author: codeperfectplus

src/routes/helper/common_helper.py

@@ -4,6 +4,10 @@
 from functools import wraps
 from flask import flash, redirect, url_for, render_template, request, session
 import subprocess
+import os
+from src.utils import ROOT_DIR
+
+
 
 def get_email_addresses(user_level=None, receive_email_alerts=True, fetch_all_users=False):
     """ Retrieve email addresses of users based on filters."""
@@ -113,3 +117,4 @@ def decorated_function(*args, **kwargs):
             return f(*args, **kwargs)
         return decorated_function
     return decorator
+

src/routes/helper/prometheus_helper.py

@@ -41,6 +41,25 @@ def show_targets():
         })
     return targets_info
 
+def update_prometheus_config():
+    """
+    update the first target whenever the function called
+    as the network change the ip address will also change 
+    so 1st ip address in the list is the ip adddress of the machine
+    with the job localhost
+    """
+    print("Updating Prometheus config...")
+    ipv4_address = subprocess.run(['hostname', '-I'], capture_output=True, text=True, check=True).stdout.split()[0]
+    prometheus_config_path = os.path.join(ROOT_DIR, 'prometheus_config', 'prometheus.yml')
+    config = load_yaml(prometheus_config_path)
+    # fetch the localhost job
+    localhost_job = next((job for job in config['scrape_configs'] if job['job_name'] == 'localhost'), None)
+    if localhost_job:
+        localhost_job['static_configs'][0]['targets'][0] = f'{ipv4_address}:5050'
+        save_yaml(config, prometheus_config_path)
+        return True
+
+    return False
 
 def update_prometheus_container():
     """Update the Prometheus container."""

src/routes/prometheus.py

@@ -12,7 +12,9 @@
     is_valid_file, 
     show_targets, 
     prometheus_yml_path,
-    update_prometheus_container)
+    update_prometheus_container,
+    update_prometheus_config)
+
 
 
 # Define the Prometheus Blueprint
@@ -61,9 +63,17 @@ def delete_file_path(id):
 
 @app.route('/targets')
 def targets():
+    update_prometheus_config()
+    # update_prometheus_container()
     targets_info = show_targets()
     return render_template('other/targets.html', targets_info=targets_info)
 
+@app.route('/targets/restart_prometheus')
+def restart_prometheus():
+    update_prometheus_config
+    update_prometheus_container()
+    flash('Prometheus container restarted successfully!', 'success')
+    return redirect(url_for('dashboard_network'))
 
 @app.route('/targets/add_target', methods=['POST'])
 def add_target():
@@ -72,6 +82,11 @@ def add_target():
     scrape_interval = request.form.get('scrape_interval', '15s') + 's'  # New scrape interval
     config = load_yaml(prometheus_yml_path)
 
+    # new target should be <ip>:<port> check if it is in the correct format
+    if ':' not in new_target:
+        flash('Invalid target format. It should be in the format <ip>:<port>.', 'danger')
+        return redirect(url_for('dashboard_network'))
+    
     for scrape_config in config['scrape_configs']:
         if scrape_config['job_name'] == job_name:
             scrape_config['static_configs'][0]['targets'].append(new_target)
@@ -127,7 +142,7 @@ def change_interval():
             scrape_config['scrape_interval'] = new_interval
             flash('Scrape interval updated successfully!', 'success')
             break
-
+    
     save_yaml(config, prometheus_yml_path)
     update_prometheus_container()
     return redirect(url_for('targets'))

src/scripts/prometheus.sh

@@ -43,7 +43,7 @@ mkdir -p "$PROMETHEUS_DATA_DIR"
 log "Generating prometheus.yml configuration file."
 cat > "$PROMETHEUS_CONFIG_FILE" <<EOL
 scrape_configs:
-  - job_name: $JOB_NAME
+  - job_name: localhost
     scrape_interval: $SCRAPING_INTERVAL
     static_configs:
     - targets:

src/templates/network/dashboard_network.html

@@ -7,6 +7,7 @@
 {% block content %}
 <div class="container">
     <h1>SystemGuard Active Targets</h1>
+    {% include 'ext/message.html' %}
     <table>
         <thead>
             <tr>
@@ -28,6 +29,14 @@ <h1>SystemGuard Active Targets</h1>
     </table>
 </div>
 
+
+<div class="text-center">
+    <h2>Restart Prometheus Docker Service</h2>
+    <form action="{{ url_for('restart_prometheus') }}">
+        <input type="submit" value="Restart Prometheus">
+    </form>
+</div>
+
 <div class="text-center">
     <h2>Add New Target</h2>
     <form action="{{ url_for('add_target') }}" method="post">

src/templates/other/targets.html

@@ -7,6 +7,7 @@
 
 {% block content %}
     <h1>SystemGuard Targets</h1>
+    {% include 'ext/message.html' %}
     <table>
         <tr>
             <th>Job Name</th>