import logging import re import os import yaml import sys from errors import ConfigError logger = logging.getLogger() class Config(object): def __init__(self, filepath): """ Args: filepath (str): Path to config file """ if not os.path.isfile(filepath): raise ConfigError(f"Config file '{filepath}' does not exist") # Load in the config file at the given filepath with open(filepath) as file_stream: config = yaml.full_load(file_stream.read()) # Logging setup formatter = logging.Formatter('%(asctime)s | %(name)s [%(levelname)s] %(message)s') log_dict = config.get("logging", {}) log_level = log_dict.get("level", "INFO") logger.setLevel(log_level) file_logging = log_dict.get("file_logging", {}) file_logging_enabled = file_logging.get("enabled", False) file_logging_filepath = file_logging.get("filepath", "bot.log") if file_logging_enabled: handler = logging.FileHandler(file_logging_filepath) handler.setFormatter(formatter) logger.addHandler(handler) console_logging = log_dict.get("console_logging", {}) console_logging_enabled = console_logging.get("enabled", True) if console_logging_enabled: handler = logging.StreamHandler(sys.stdout) handler.setFormatter(formatter) logger.addHandler(handler) # Database setup database_dict = config.get("database", {}) self.database_filepath = database_dict.get("filepath") # Matrix bot account setup matrix = config.get("matrix", {}) self.user_id = matrix.get("user_id") if not self.user_id: raise ConfigError("matrix.user_id is a required field") elif not re.match("@.*:.*", self.user_id): raise ConfigError("matrix.user_id must be in the form @name:domain") self.access_token = matrix.get("access_token") if not self.access_token: raise ConfigError("matrix.access_token is a required field") self.device_id = matrix.get("device_id") if not self.device_id: logger.warning( "matrix.device_id is not provided, which means that end-to-end " encryption won't work correctly" ) self.homeserver_url = matrix.get("homeserver_url") if not self.homeserver_url: raise ConfigError("matrix.homeserver_url is a required field") self.command_prefix = config.get("command_prefix", "!c") + " "