88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
# Configuration Helper
|
|
#
|
|
# Copyright (C) 2020 Eric Callahan <arksine.code@gmail.com>
|
|
#
|
|
# This file may be distributed under the terms of the GNU GPLv3 license
|
|
import configparser
|
|
import os
|
|
import logging
|
|
|
|
class ConfigError(Exception):
|
|
pass
|
|
|
|
class Sentinel:
|
|
pass
|
|
|
|
class ConfigHelper:
|
|
error = ConfigError
|
|
def __init__(self, server, config, section):
|
|
self.server = server
|
|
self.config = config
|
|
self.section = section
|
|
self.sections = config.sections
|
|
self.has_section = config.has_section
|
|
|
|
def get_server(self):
|
|
return self.server
|
|
|
|
def __getitem__(self, key):
|
|
return self.getsection(key)
|
|
|
|
def __contains__(self, key):
|
|
return key in self.config
|
|
|
|
def getsection(self, section):
|
|
if section not in self.config:
|
|
raise ConfigError(f"No section [{section}] in config")
|
|
return ConfigHelper(self.server, self.config, section)
|
|
|
|
def _get_option(self, func, option, default):
|
|
try:
|
|
val = func(option, default)
|
|
except Exception:
|
|
raise ConfigError(
|
|
f"Error parsing option ({option}) from "
|
|
f"section [{self.section}]")
|
|
if val == Sentinel:
|
|
raise ConfigError(
|
|
f"No option found ({option}) in section [{self.section}]")
|
|
return val
|
|
|
|
def get(self, option, default=Sentinel):
|
|
return self._get_option(
|
|
self.config[self.section].get, option, default)
|
|
|
|
def getint(self, option, default=Sentinel):
|
|
return self._get_option(
|
|
self.config[self.section].getint, option, default)
|
|
|
|
def getboolean(self, option, default=Sentinel):
|
|
return self._get_option(
|
|
self.config[self.section].getboolean, option, default)
|
|
|
|
def getfloat(self, option, default=Sentinel):
|
|
return self._get_item(
|
|
self.config[self.section].getfloat, option, default)
|
|
|
|
def get_configuration(server, cmd_line_args):
|
|
cfg_file_path = os.path.normpath(os.path.expanduser(
|
|
cmd_line_args.configfile))
|
|
if not os.path.isfile(cfg_file_path):
|
|
raise ConfigError(f"Configuration File Not Found: '{cfg_file_path}''")
|
|
config = configparser.ConfigParser(interpolation=None)
|
|
try:
|
|
config.read(cfg_file_path)
|
|
except Exception:
|
|
raise ConfigError(f"Error Reading Config: '{cfg_file_path}'") from None
|
|
|
|
try:
|
|
server_cfg = config['server']
|
|
except KeyError:
|
|
raise ConfigError("No section [server] in config")
|
|
|
|
if server_cfg.get('enable_debug_logging', True):
|
|
logging.getLogger().setLevel(logging.DEBUG)
|
|
|
|
config['cmd_args'] = {'logfile': cmd_line_args.logfile}
|
|
return ConfigHelper(server, config, 'server')
|