Merge branch 'add_monitoring_prometheus' of git.selfprivacy.org:SelfPrivacy/selfprivacy-rest-api into add_monitoring_prometheus

This commit is contained in:
dettlaff 2024-07-24 21:04:42 +04:00
commit df4a20ebf7
3 changed files with 17 additions and 16 deletions

View file

@ -15,12 +15,12 @@ from selfprivacy_api.migrations.check_for_system_rebuild_jobs import (
CheckForSystemRebuildJobs, CheckForSystemRebuildJobs,
) )
from selfprivacy_api.migrations.add_roundcube import AddRoundcube from selfprivacy_api.migrations.add_roundcube import AddRoundcube
from selfprivacy_api.migrations.add_prometheus import AddPrometheus from selfprivacy_api.migrations.add_monitoring import AddMonitoring
migrations = [ migrations = [
WriteTokenToRedis(), WriteTokenToRedis(),
CheckForSystemRebuildJobs(), CheckForSystemRebuildJobs(),
AddPrometheus(), AddMonitoring(),
AddRoundcube(), AddRoundcube(),
] ]

View file

@ -2,35 +2,36 @@ from selfprivacy_api.migrations.migration import Migration
from selfprivacy_api.services.flake_service_manager import FlakeServiceManager from selfprivacy_api.services.flake_service_manager import FlakeServiceManager
from selfprivacy_api.utils import ReadUserData, WriteUserData from selfprivacy_api.utils import ReadUserData, WriteUserData
from selfprivacy_api.utils.block_devices import BlockDevices
class AddPrometheus(Migration): class AddMonitoring(Migration):
"""Adds the Prometheus if it is not present.""" """Adds monitoring service if it is not present."""
def get_migration_name(self) -> str: def get_migration_name(self) -> str:
return "add_prometheus" return "add_monitoring"
def get_migration_description(self) -> str: def get_migration_description(self) -> str:
return "Adds the Prometheus if it is not present." return "Adds the Monitoring if it is not present."
def is_migration_needed(self) -> bool: def is_migration_needed(self) -> bool:
with FlakeServiceManager() as manager: with FlakeServiceManager() as manager:
if "prometheus" not in manager.services: if "monitoring" not in manager.services:
return True return True
with ReadUserData() as data: with ReadUserData() as data:
if "prometheus" not in data["modules"]: if "monitoring" not in data["modules"]:
return True return True
return False return False
def migrate(self) -> None: def migrate(self) -> None:
with FlakeServiceManager() as manager: with FlakeServiceManager() as manager:
if "prometheus" not in manager.services: if "monitoring" not in manager.services:
manager.services[ manager.services[
"prometheus" "monitoring"
] = "git+https://git.selfprivacy.org/SelfPrivacy/selfprivacy-nixos-config.git?ref=flakes&dir=sp-modules/prometheus" ] = "git+https://git.selfprivacy.org/SelfPrivacy/selfprivacy-nixos-config.git?ref=flakes&dir=sp-modules/monitoring"
with WriteUserData() as data: with WriteUserData() as data:
if "prometheus" not in data["modules"]: if "monitoring" not in data["modules"]:
data["modules"]["prometheus"] = { data["modules"]["monitoring"] = {
"enable": False, "enable": True,
"subdomain": "prometheus", "location": BlockDevices().get_root_block_device().name
} }

View file

@ -39,7 +39,7 @@ class PrometheusQueries:
json = response.json() json = response.json()
print(json) print(json)
return PrometheusQueryResult( return PrometheusQueryResult(
result_type=json["result_type"], result=json["result"] result_type=json["resultType"], result=json["result"]
) )
except Exception as error: except Exception as error:
raise Exception("Prometheus request failed! " + str(error)) raise Exception("Prometheus request failed! " + str(error))