📦 NEW: DEV-871 Add Hetzner Plugin to Grafana monitor all Server Dynamic
This commit is contained in:
86
hetzner_discovery.py
Normal file
86
hetzner_discovery.py
Normal file
@@ -0,0 +1,86 @@
|
||||
import requests
|
||||
import json
|
||||
import os
|
||||
from dotenv import load_dotenv
|
||||
|
||||
# Load environment variables from .env file
|
||||
load_dotenv()
|
||||
|
||||
# 🔑 Load Hetzner API Token
|
||||
HETZNER_API_TOKEN = os.getenv("HETZNER_API_TOKEN")
|
||||
|
||||
# Check if token is loaded
|
||||
if not HETZNER_API_TOKEN:
|
||||
raise ValueError("❌ HETZNER_API_TOKEN is missing! Make sure it's set in the .env file.")
|
||||
|
||||
# 📂 Path to the Prometheus service discovery file
|
||||
PROMETHEUS_TARGETS_FILE = "/opt/phx/main/config/hetzner_targets.json"
|
||||
ERROR_LOG_FILE = "/opt/phx/main/config/hetzner_error_servers.json"
|
||||
|
||||
# 📌 Hetzner API URL
|
||||
HETZNER_API_URL = os.getenv("HETZNER_API_URL")
|
||||
|
||||
# Check if API URL is loaded
|
||||
if not HETZNER_API_URL:
|
||||
raise ValueError("❌ HETZNER_API_URL is missing! Make sure it's set in the .env file.")
|
||||
|
||||
|
||||
# 📡 Fetch Hetzner server list
|
||||
def get_hetzner_servers():
|
||||
headers = {"Authorization": f"Bearer {HETZNER_API_TOKEN}"}
|
||||
response = requests.get(HETZNER_API_URL, headers=headers)
|
||||
|
||||
if response.status_code != 200:
|
||||
print(f"❌ Error fetching servers: {response.text}")
|
||||
return []
|
||||
|
||||
return response.json().get("servers", [])
|
||||
|
||||
|
||||
# 🏗️ Generate Prometheus JSON file
|
||||
def generate_prometheus_sd_config():
|
||||
servers = get_hetzner_servers()
|
||||
targets = []
|
||||
error_servers = []
|
||||
|
||||
for server in servers:
|
||||
if "public_net" in server and "ipv4" in server["public_net"]:
|
||||
ipv4 = server["public_net"]["ipv4"]["ip"]
|
||||
server_name = server["name"]
|
||||
datacenter = server["datacenter"]["location"]["name"]
|
||||
|
||||
targets.append({
|
||||
"targets": [f"{ipv4}:9100"],
|
||||
"labels": {
|
||||
"instance": server_name,
|
||||
"datacenter": datacenter
|
||||
}
|
||||
})
|
||||
else:
|
||||
# Log the server that couldn't be added
|
||||
error_servers.append({
|
||||
"server_id": server["id"],
|
||||
"name": server["name"],
|
||||
"status": server["status"],
|
||||
"datacenter": server["datacenter"]["location"]["name"],
|
||||
"reason": "Missing public_net or IPv4"
|
||||
})
|
||||
|
||||
# Save JSON to Prometheus target file
|
||||
with open(PROMETHEUS_TARGETS_FILE, "w") as f:
|
||||
json.dump(targets, f, indent=4)
|
||||
|
||||
print(f"✅ Updated Prometheus targets in {PROMETHEUS_TARGETS_FILE}")
|
||||
|
||||
# Save error logs if any servers were skipped
|
||||
if error_servers:
|
||||
with open(ERROR_LOG_FILE, "w") as f:
|
||||
json.dump(error_servers, f, indent=4)
|
||||
print(f"⚠️ Some servers could not be added. Check {ERROR_LOG_FILE} for details.")
|
||||
else:
|
||||
print("✅ All servers were added successfully, no errors detected.")
|
||||
|
||||
|
||||
# 🔄 Run the script
|
||||
if __name__ == "__main__":
|
||||
generate_prometheus_sd_config()
|
||||
Reference in New Issue
Block a user