blivechat/config.py

# -*- 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