tzk/config.py

22 lines
619 B
Python
Raw Normal View History

2021-08-25 20:06:42 +00:00
import importlib
import os
from pathlib import Path
import sys
config_path = Path.cwd()
for child in sorted(config_path.iterdir()):
if child.is_file() and child.name.endswith('.py'):
mod_name = child.name.rsplit('.', 1)[0]
if mod_name == 'tzk_config':
sys.path.insert(0, str(config_path))
importlib.import_module(mod_name)
del sys.path[0]
break
else:
print(
f"Your TZK config file could not be found. "
f"Please ensure there is a file called tzk_config.py "
f"in the current directory.", file=sys.stderr)
sys.exit(1)