|
|
|
|
|
import configparser |
|
import logging |
|
import os |
|
from typing import * |
|
|
|
logger = logging.getLogger(__name__) |
|
|
|
CONFIG_PATH_LIST = [ |
|
os.path.join('config', 'config.ini'), |
|
os.path.join('config', 'config.example.ini') |
|
] |
|
|
|
_config: Optional['AppConfig'] = None |
|
|
|
|
|
def init(): |
|
if reload(): |
|
return |
|
logger.warning('Using default config') |
|
global _config |
|
_config = AppConfig() |
|
|
|
|
|
def reload(): |
|
config_path = '' |
|
for path in CONFIG_PATH_LIST: |
|
if os.path.exists(path): |
|
config_path = path |
|
break |
|
if config_path == '': |
|
return False |
|
|
|
config = AppConfig() |
|
if not config.load(config_path): |
|
return False |
|
global _config |
|
_config = config |
|
return True |
|
|
|
|
|
def get_config(): |
|
return _config |
|
|
|
|
|
class AppConfig: |
|
def __init__(self): |
|
self.database_url = 'sqlite:///data/database.db' |
|
self.tornado_xheaders = False |
|
|
|
def load(self, path): |
|
try: |
|
config = configparser.ConfigParser() |
|
config.read(path, 'utf-8-sig') |
|
|
|
self._load_app_config(config) |
|
except Exception as e: |
|
logger.exception(f'Failed to load config: {e}') |
|
return False |
|
return True |
|
|
|
def _load_app_config(self, config): |
|
app_section = config['app'] |
|
self.database_url = app_section['database_url'] |
|
self.tornado_xheaders = app_section.getboolean('tornado_xheaders') |
|
|
|
|
|
def _str_to_list(value, item_type: Type = str, container_type: Type = list): |
|
value = value.strip() |
|
if value == '': |
|
return container_type |
|
items = value.split(',') |
|
items = map(lambda item: item.strip(), items) |
|
if item_type is not str: |
|
items = map(lambda item: item_type(item), items) |
|
return container_type(items) |
|
|
