loic/clusterman
1
0
Fork 0
mirror of https://gitlab.com/manzerbredes/clusterman.git synced 2025-04-06 03:56:27 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Minor changes

Browse source
This commit is contained in:
Loic Guegan 2023-10-26 17:50:34 +02:00
parent 2250082bb3
commit 2d46fcf955
2 changed files with 20 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
clusterman/config.py
View file

@ -1,5 +1,6 @@
from pathlib import Path from pathlib import Path
import os, json import os, json, sys
from jsonschema import validate
class Config: class Config:
@ -16,6 +17,18 @@ class Config:
"plugins": { "ls": "ls -al" }, "plugins": { "ls": "ls -al" },
"timeout": 0.5 "timeout": 0.5
} }
SCHEMA_CONFIG = {
"type": "object",
"properties": {
"timeout": {"type": "number"},
"plugins": {"type": "object"},
"cluster": {"type": "object", "properties":{
"ip4_from": {"type": "string"},
"ip4_to": {"type": "string"},
"ip4_ignore": {"type": "array", "items":{"type": "string"}}
}}
}
}
def __init__(self): def __init__(self):
Path(self.CONF_DIR).mkdir(parents=True, exist_ok=True) Path(self.CONF_DIR).mkdir(parents=True, exist_ok=True)
@ -27,6 +40,11 @@ class Config:
if os.path.exists(self.CONF_FILE): if os.path.exists(self.CONF_FILE):
with open(self.CONF_FILE) as f: with open(self.CONF_FILE) as f:
self.config=json.load(f) self.config=json.load(f)
try:
validate(instance=self.config, schema=self.SCHEMA_CONFIG)
except:
print("Invalid configuration file")
sys.exit(1)
else: else:
self.save() self.save()

2
pyproject.toml
View file

@ -1,7 +1,7 @@
[project] [project]
name = "clusterman" name = "clusterman"
dynamic = ["version"] dynamic = ["version"]
dependencies = ["psutil"] dependencies = ["psutil", "jsonschema"]
description = "Simple cluster management utility" description = "Simple cluster management utility"
readme = "README.md" readme = "README.md"