blob: f372008e64a803c47605a9bcac0c85568d0b14b2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
import os
import json
from typing import List, Dict, Any
from appdirs import user_config_dir # type: ignore
CONFIG_FILE = "config.json"
def list_instances() -> List[str]:
instances = []
with os.scandir(config_dir()) as it:
for entry in it:
if entry.is_dir() and os.path.isfile(f"{entry.path}/{CONFIG_FILE}"):
instances.append(entry.name)
return instances
def config_dir() -> str:
path: str = user_config_dir("labbot")
os.makedirs(path, exist_ok=True)
return path
def instance_config_dir(name) -> str:
path = os.path.join(config_dir(), name)
return path
def write_instance_config(name: str, data: dict) -> None:
instance_path = instance_config_dir(name)
os.makedirs(instance_path, exist_ok=True)
conf_path = os.path.join(instance_path, CONFIG_FILE)
with open(conf_path, "w") as file:
json.dump(data, file)
if data != read_instance_config(name):
raise ValueError("Config could not be saved properly")
def read_instance_config(name: str) -> Dict[str, Any]:
conf_path = os.path.join(instance_config_dir(name), CONFIG_FILE)
try:
with open(conf_path, "r") as file:
return json.load(file)
except FileNotFoundError:
return {}
|
