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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
|
import os
import json
from typing import List, Dict, Any
from appdirs import user_config_dir # type: ignore
from functools import wraps
CONFIG_FILE = "config.json"
class Config:
def __init__(self, conf_name=None, name=None):
self.conf_name = conf_name
self.name = name
self.settings = {
"GLOBAL": {},
"GROUP": {},
"PROJECT": {},
}
self.group_id = None
self.project_id = None
if self.conf_name and self.name:
self.setup(self.conf_name, self.name)
def setup(self, conf_name, name):
instance_path = instance_config_dir(name)
if conf_name.endswith(".json"):
conf_name = conf_name[:-5]
if conf_name == "config":
raise ValueError("Config cannot be named config")
self.conf_name = conf_name
self.name = name
try:
settings = json.load(
open(os.path.join(instance_path, f"{conf_name}.json")))
# allow the config on disk to overwrite existing config keys
# but keep new ones around
self.settings["GLOBAL"].update(settings.get("GLOBAL", {}))
self.settings["GROUP"].update(settings.get("GROUP", {}))
self.settings["PROJECT"].update(settings.get("PROJECT", {}))
except (IOError, ValueError):
pass
def __getitem__(self, key):
try:
return self.settings["PROJECT"][str(self.project_id)][key]
except KeyError:
try:
return self.settings["GROUP"][str(self.group_id)][key]
except KeyError:
return self.settings["GLOBAL"][key]
def save(self):
instance_path = instance_config_dir(self.name)
conf_path = os.path.join(instance_path, f"{self.conf_name}.json")
with open(conf_path, "w") as f:
json.dump(self.settings, f)
def config_decorator(self, **dkwargs):
def decorator(func):
@wraps(func)
async def wrapper(*args, **kwargs):
full_kwargs = kwargs.copy()
full_kwargs.update(zip(func.__code__.co_varnames, args))
try:
event = full_kwargs["event"]
self.project_id = event.project_id
except KeyError:
pass
try:
return await func(*args, **kwargs)
finally:
self.group_id = None
self.project_id = None
self.save()
return wrapper
return decorator
def set_global_data(self, **kwargs):
kwargs.update(self.settings["GLOBAL"])
self.settings["GLOBAL"] = kwargs
def set_group_data(self, group_id, **kwargs):
kwargs.update(self.settings["GROUP"][group_id])
self.settings["GROUP"][group_id] = kwargs
def set_project_data(self, project_id, **kwargs):
kwargs.update(self.settings["PROJECT"][project_id])
self.settings["PROJECT"][project_id] = kwargs
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)
data = json.loads(json.dumps(data))
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 {}
|
