-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathconfig.py
36 lines (28 loc) · 993 Bytes
/
config.py
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
import pickle
from pydantic import BaseModel, validator
class Config(BaseModel):
illegal_vars = ["illegal_vars", "_glob_vars", "class_name"]
_glob_vars = tuple()
class_name: str = None
class Config:
validate_all = True
extra = "forbid"
@validator("class_name", pre=True, always=True)
def set_class_name(cls, v):
assert v is None, "Config property 'class_name' is not settable."
return cls.__name__
def set_globs(self, globs=None):
if globs is None:
globs = {}
else:
for key, value in globs.items():
if hasattr(self, key):
setattr(self, key, value)
for k in self._glob_vars:
globs[k] = getattr(self, k)
for _, value in self:
if isinstance(value, Config):
value.set_globs(globs=globs)
def to_pickle(self, file_name):
with open(file_name, "wb") as f:
pickle.dump(self, f)