-
Notifications
You must be signed in to change notification settings - Fork 0
/
configurator.py
40 lines (36 loc) · 1.15 KB
/
configurator.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
37
38
39
40
import getpass
from configparser import ConfigParser, ExtendedInterpolation
from pathlib import Path
from typing import Dict, Iterable, Optional
class Config:
def __init__(
self,
config_dir: Optional[Path] = None,
config_files: Optional[Iterable[Path]] = None,
interpolate_vars: Optional[Dict[str, str]] = None,
):
self.config_dir = (
Path(__file__).parent
if config_dir is None
else Path(config_dir)
)
config_files = (
["base.ini", "local.ini"]
if config_files is None
else [Path(file) for file in config_files]
)
interpolate_vars = (
{"username": getpass.getuser()}
if interpolate_vars is None
else interpolate_vars
)
self.parser = ConfigParser(
defaults=interpolate_vars,
interpolation=ExtendedInterpolation(),
converters={"path": Path},
)
self.config_files = self.parser.read(
self.config_dir / file for file in config_files
)
def __getitem__(self, item):
return self.parser[item]