You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import configparser
|
|
import logging
|
|
import os
|
|
from typing import *
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
CONFIG_PATH = os.path.join('data', 'config.ini')
|
|
|
|
_config: Optional['AppConfig'] = None
|
|
|
|
|
|
def init():
|
|
if reload():
|
|
return
|
|
logger.warning('Using default config')
|
|
global _config
|
|
_config = AppConfig()
|
|
|
|
|
|
def reload():
|
|
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.enable_translate = True
|
|
self.allow_translate_rooms = {}
|
|
self.tornado_xheaders = False
|
|
self.loader_url = ''
|
|
|
|
def load(self, path):
|
|
try:
|
|
config = configparser.ConfigParser()
|
|
config.read(path)
|
|
|
|
app_section = config['app']
|
|
self.database_url = app_section['database_url']
|
|
self.enable_translate = app_section.getboolean('enable_translate')
|
|
|
|
allow_translate_rooms = app_section['allow_translate_rooms']
|
|
if allow_translate_rooms == '':
|
|
self.allow_translate_rooms = {}
|
|
else:
|
|
allow_translate_rooms = allow_translate_rooms.split(',')
|
|
self.allow_translate_rooms = set(map(lambda id_: int(id_.strip()), allow_translate_rooms))
|
|
|
|
self.tornado_xheaders = app_section.getboolean('tornado_xheaders')
|
|
self.loader_url = app_section['loader_url']
|
|
|
|
except (KeyError, ValueError):
|
|
logger.exception('Failed to load config:')
|
|
return False
|
|
return True
|