python-mylib/mylib/config.py

1174 lines
41 KiB
Python
Raw Normal View History

2021-11-03 17:38:31 +01:00
# -*- coding: utf-8 -*-
2023-01-06 18:16:46 +01:00
# pylint: disable=too-many-lines
2021-11-03 17:38:31 +01:00
""" Configuration & options parser """
import argparse
from configparser import ConfigParser
from getpass import getpass
from logging.config import fileConfig
import logging
import os
import re
2021-11-03 17:38:31 +01:00
import stat
import sys
import textwrap
import traceback
2021-11-03 17:38:31 +01:00
import argcomplete
import keyring
from systemd.journal import JournalHandler
log = logging.getLogger(__name__)
# Constants
DEFAULT_ENCODING = 'utf-8'
DEFAULT_CONFIG_DIRPATH = os.path.expanduser('./')
DEFAULT_CONSOLE_LOG_FORMAT = '%(asctime)s - %(module)s:%(lineno)d - %(levelname)s - %(message)s'
2021-11-03 17:38:31 +01:00
class BaseOption: # pylint: disable=too-many-instance-attributes
2021-11-03 17:38:31 +01:00
""" Base configuration option class """
def __init__(self, config, section, name, default=None, comment=None,
arg=None, short_arg=None, arg_help=None, no_arg=False):
2021-11-03 17:38:31 +01:00
self.config = config
self.section = section
self.name = name
self.default = default
self.comment = comment
self.no_arg = no_arg
2021-11-03 17:38:31 +01:00
self.arg = arg
self.short_arg = short_arg
self.arg_help = arg_help if arg_help else comment
@property
def _isset_in_options(self):
""" Check if option is defined in registered arguments parser options """
return (
self.config.options
and not self.no_arg
2021-11-03 17:38:31 +01:00
and self._from_options != self.default
)
@property
def _from_options(self):
""" Get option from arguments parser options """
value = (
getattr(self.config.options, self.parser_dest)
if self.config.options and not self.no_arg else None
2021-11-03 17:38:31 +01:00
)
log.debug(
'_from_options(%s, %s) = %s',
self.section.name, self.name, value
)
return value
@property
def _isset_in_config_file(self):
""" Check if option is defined in the loaded configuration file """
return (
self.config.config_parser
and self.config.config_parser.has_option(self.section.name, self.name)
)
@property
def _from_config(self):
""" Get option value from ConfigParser """
return self.config.config_parser.get(self.section.name, self.name)
@property
def _default_in_config(self):
""" Get option default value considering current value from configuration """
return (
self._from_config if self._isset_in_config_file
else self.default
)
2021-11-03 17:38:31 +01:00
def isset(self):
""" Check if option is defined in the loaded configuration file """
return self._isset_in_config_file or self._isset_in_options
def get(self):
""" Get option value from options, config or default """
if self._isset_in_options:
return self._from_options
if self._isset_in_config_file:
return self._from_config
return self.default
def set(self, value):
""" Set option value to config file """
assert self.config.config_parser or (self.config.options and not self.no_arg), (
"Can't set option value: configuration file not configured, not options (or no argument)")
2021-11-03 17:38:31 +01:00
if value == '':
value = None
if self.config.config_parser:
if value == self.default or value is None:
# Remove option from config (is section exists)
if self.config.config_parser.has_section(self.section.name):
self.config.config_parser.remove_option(
self.section.name, self.name)
else:
# Store option to config
if not self.config.config_parser.has_section(self.section.name):
self.config.config_parser.add_section(self.section.name)
self.config.config_parser.set(
self.section.name, self.name, self.to_config(value)
)
if self.config.options and not self.no_arg:
setattr(self.config.options, self.parser_dest, value)
2021-11-03 17:38:31 +01:00
@property
def parser_action(self):
""" Get action as accept by argparse.ArgumentParser """
return 'store'
@property
def parser_type(self):
""" Get type as handle by argparse.ArgumentParser """
return str
@property
def parser_dest(self):
""" Get option name in arguments parser options """
2023-01-06 18:16:46 +01:00
return f'{self.section.name}_{self.name}'
2021-11-03 17:38:31 +01:00
@property
def parser_help(self):
""" Get option help message in arguments parser options """
if self.arg_help and self.default is not None:
2023-01-06 18:16:46 +01:00
# pylint: disable=consider-using-f-string
return '{0} (Default: {1})'.format(
2023-01-06 18:16:46 +01:00
self.arg_help,
re.sub(r'%([^%])', r'%%\1', str(self._default_in_config))
)
2021-11-03 17:38:31 +01:00
if self.arg_help:
return self.arg_help
return None
@property
def parser_argument_name(self):
""" Get option argument name in parser options """
return (
self.arg if self.arg else
2023-01-06 18:16:46 +01:00
f'--{self.section.name}-{self.name}'.lower().replace('_', '-')
2021-11-03 17:38:31 +01:00
)
def add_option_to_parser(self, section_opts):
""" Add option to arguments parser """
if self.no_arg:
return
2021-11-03 17:38:31 +01:00
args = [self.parser_argument_name]
if self.short_arg:
args.append(self.short_arg)
kwargs = dict(
action=self.parser_action,
dest=self.parser_dest,
help=self.parser_help,
default=self.default,
)
if self.parser_type: # pylint: disable=using-constant-test
kwargs['type'] = self.parser_type
log.debug(
'add_option_to_parser(%s, %s): argument name(s)=%s / kwargs=%s',
self.section.name, self.name, ', '.join(args), kwargs
)
section_opts.add_argument(*args, **kwargs)
def to_config(self, value=None):
""" Format value as stored in configuration file """
value = value if value is not None else self.get()
return '' if value is None else str(value)
def export_to_config(self):
""" Export option to configuration file """
lines = []
if self.comment:
lines.append(f'# {self.comment}')
2021-11-03 17:38:31 +01:00
value = self.to_config()
default_value = (
'' if self.default is None else
self.to_config(self.default)
)
log.debug(
'export_to_config(%s, %s): value=%s / default=%s',
self.section.name, self.name, value, default_value)
if default_value:
if isinstance(default_value, str) and '\n' in default_value:
lines.append('# Default:')
lines.extend([f'# {line}' for line in default_value.split('\n')])
else:
lines.append(
f'# Default: {default_value}'
)
2021-11-03 17:38:31 +01:00
if value and value != default_value:
if isinstance(value, str) and '\n' in value:
value_lines = value.split('\n')
lines.append(f'# Default: {value_lines[0]}')
lines.extend([f' {line.replace("#", "%(hash)s")}' for line in value_lines[1:]])
else:
lines.append(
f'{self.name} = {value}'
)
2021-11-03 17:38:31 +01:00
else:
lines.append(f'# {self.name} =')
2021-11-03 17:38:31 +01:00
lines.append('')
return '\n'.join(lines)
@staticmethod
def _get_user_input(prompt):
"""
Get user console input
Note: do not use directly input() to allow to mock it in tests
"""
return input(prompt)
def _ask_value(self, prompt=None, **kwargs):
""" Ask to user to enter value of this option and return it """
if self.comment:
print(f'# {self.comment}')
default_value = kwargs.get('default_value', self.get())
if not prompt:
prompt = f'{self.name}: '
if default_value is not None:
if isinstance(default_value, str) and '\n' in default_value:
prompt += "[\n %s\n] " % "\n ".join(
self.to_config(default_value).split('\n')
)
else:
prompt += f'[{self.to_config(default_value)}] '
value = self._get_user_input(prompt)
return default_value if value == '' else value
def ask_value(self, set_it=True):
"""
Ask to user to enter value of this option and set or
return it regarding set parameter
"""
value = self._ask_value()
if set_it:
return self.set(value)
return value
2021-11-03 17:38:31 +01:00
class StringOption(BaseOption):
""" String configuration option class """
class BooleanOption(BaseOption):
""" Boolean configuration option class """
@property
def _from_config(self):
""" Get option value from ConfigParser """
return self.config.config_parser.getboolean(self.section.name, self.name)
def to_config(self, value=None):
""" Format value as stored in configuration file """
return super().to_config(value).lower()
@property
def _isset_in_options(self):
""" Check if option is defined in registered arguments parser options """
return (
self.config.options
and not self.no_arg
and getattr(self.config.options, self.parser_dest)
)
@property
def _from_options(self):
""" Get option from arguments parser options """
return (
not self._default_in_config if self._isset_in_options
else self._default_in_config
)
2021-11-03 17:38:31 +01:00
@property
def parser_action(self):
return "store_true"
2021-11-03 17:38:31 +01:00
@property
def parser_type(self):
return None
@property
def parser_argument_name(self):
""" Get option argument name in parser options """
2023-01-06 18:16:46 +01:00
# pylint: disable=consider-using-f-string
return (
self.arg if self.arg else
'--{0}-{1}-{2}'.format(
self.section.name,
'enable' if not self._default_in_config else 'disable',
self.name
).lower().replace('_', '-')
)
def _ask_value(self, prompt=None, **kwargs):
""" Ask to user to enter value of this option and return it """
default_value = self.get()
2023-01-06 18:16:46 +01:00
prompt = f'{self.name}: '
if default_value:
prompt += '[Y/n] '
else:
prompt += '[y/N] '
while True:
value = super()._ask_value(prompt, **kwargs)
if value in ['', None, default_value]:
return default_value
if value.lower() == 'y':
return True
if value.lower() == 'n':
return False
print('Invalid answer. Possible values: Y or N (case insensitive)')
2021-11-03 17:38:31 +01:00
class FloatOption(BaseOption):
""" Float configuration option class """
@property
def _from_config(self):
""" Get option value from ConfigParser """
return self.config.config_parser.getfloat(self.section.name, self.name)
@property
def parser_type(self):
return float
def _ask_value(self, prompt=None, **kwargs):
""" Ask to user to enter value of this option and return it """
default_value = self.get()
while True:
value = super()._ask_value(prompt, **kwargs)
if value in ['', None, default_value]:
return default_value
try:
return float(value)
except ValueError:
print('Invalid answer. Must a numeric value, for instance "12" or "12.5"')
2021-11-03 17:38:31 +01:00
class IntegerOption(BaseOption):
""" Integer configuration option class """
@property
def _from_config(self):
""" Get option value from ConfigParser """
return self.config.config_parser.getint(self.section.name, self.name)
def to_config(self, value=None):
""" Format value as stored in configuration file """
value = value if value is not None else self.get()
return str(int(value)) if value is not None else ''
@property
def parser_type(self):
return int
def _ask_value(self, prompt=None, **kwargs):
""" Ask to user to enter value of this option and return it """
default_value = kwargs.pop('default_value', self.get())
while True:
value = super()._ask_value(prompt, default_value=default_value, **kwargs)
if value in ['', None, default_value]:
return default_value
try:
return int(value)
except ValueError:
print('Invalid answer. Must a integer value')
2021-11-03 17:38:31 +01:00
class PasswordOption(StringOption):
""" Password configuration option class """
def __init__(self, *arg, username_option=None, keyring_value=None, **kwargs):
super().__init__(*arg, **kwargs)
self.username_option = username_option
self.keyring_value = keyring_value if keyring_value is not None else 'keyring'
@property
def _keyring_service_name(self):
""" Return keyring service name """
return '.'.join([
self.config.shortname,
self.section.name, self.name
])
@property
def _keyring_username(self):
""" Return keyring username """
return self.section.get(self.username_option) if self.username_option else self.name
def get(self):
""" Get option value """
value = super().get()
if value != self.keyring_value:
return value
service_name = self._keyring_service_name
username = self._keyring_username
log.debug(
'Retreive password %s for username=%s from keyring',
service_name, username
)
value = keyring.get_password(service_name, username)
if value is None:
2023-01-06 18:16:46 +01:00
# pylint: disable=consider-using-f-string
2021-11-03 17:38:31 +01:00
value = getpass(
2023-01-06 18:16:46 +01:00
'Please enter {0}{1}: '.format(
f'{self.section.name} {self.name}',
f' for {username}' if username != self.name else ''
2021-11-03 17:38:31 +01:00
)
)
keyring.set_password(service_name, username, value)
return value
def to_config(self, value=None):
""" Format value as stored in configuration file """
if super().get() == self.keyring_value:
return self.keyring_value
return super().to_config(value)
def set(self, value, use_keyring=None): # pylint: disable=arguments-differ
2021-11-03 17:38:31 +01:00
""" Set option value to config file """
if (use_keyring is None and super().get() == self.keyring_value) or use_keyring:
2021-11-03 17:38:31 +01:00
keyring.set_password(
self._keyring_service_name, self._keyring_username,
value)
value = self.keyring_value
super().set(value)
def _ask_value(self, prompt=None, **kwargs):
""" Ask to user to enter value of this option and return it """
if self.comment:
print('# ' + self.comment)
default_value = kwargs.pop('default_value', self.get())
if not prompt:
2023-01-06 18:16:46 +01:00
prompt = f'{self.name}: '
if default_value is not None:
# Hide value only if it differed from default value
if default_value == self.default:
2023-01-06 18:16:46 +01:00
prompt += f'[{default_value}] '
else:
prompt += '[secret defined, leave to empty to keep it as unchange] '
value = getpass(prompt)
return default_value if value == '' else value
def ask_value(self, set_it=True):
"""
Ask to user to enter value of this option and set or
return it regarding set parameter
"""
value = self._ask_value()
if set_it:
use_keyring = None
default_use_keyring = (super().get() == self.keyring_value)
while use_keyring is None:
prompt = (
2023-01-06 18:16:46 +01:00
'Do you want to use XDG keyring ? '
f"[{'Y/n' if default_use_keyring else 'y/N'}] "
)
result = input(prompt).lower()
if result == '':
use_keyring = default_use_keyring
elif result == 'y':
use_keyring = True
elif result == 'n':
use_keyring = False
else:
print('Invalid answer. Possible values: Y or N (case insensitive)')
return self.set(value, use_keyring=use_keyring)
return value
2021-11-03 17:38:31 +01:00
class ConfigSection:
""" Configuration section class """
def __init__(self, config, name, comment=None, order=None):
self.config = config
self.name = name
2023-01-06 18:16:46 +01:00
self.options = {}
2021-11-03 17:38:31 +01:00
self.comment = comment
self.order = order if isinstance(order, int) else 10
def add_option(self, _type, name, **kwargs):
"""
Add option
:param _type: Option type, derivated from BaseOption
:param name: Option name
:param **kwargs: Dict of raw option for type class
"""
2023-01-06 18:16:46 +01:00
assert not self.defined(name), f'Duplicated option {name}'
2021-11-03 17:38:31 +01:00
self.options[name] = _type(self.config, self, name, **kwargs)
return self.options[name]
def defined(self, option):
""" Check if option is defined """
return option in self.options
def isset(self, option):
""" Check if option is set """
return self.defined(option) and self.options[option].isset()
def get(self, option):
""" Get option value """
2023-01-06 18:16:46 +01:00
assert self.defined(option), f'Option {option} unknown'
2021-11-03 17:38:31 +01:00
return self.options[option].get()
def set(self, option, value):
""" Set option value """
2023-01-06 18:16:46 +01:00
assert self.defined(option), f'Option {option} unknown'
2021-11-03 17:38:31 +01:00
return self.options[option].set(value)
def add_options_to_parser(self, parser):
""" Add section to argparse.ArgumentParser """
assert isinstance(parser, argparse.ArgumentParser)
section_opts = parser.add_argument_group(
self.comment if self.comment
else self.name.capitalize()
)
2023-01-06 18:16:46 +01:00
for option in self.options: # pylint: disable=consider-using-dict-items
2021-11-03 17:38:31 +01:00
self.options[option].add_option_to_parser(section_opts)
def export_to_config(self):
""" Export section and their options to configuration file """
lines = []
if self.comment:
2023-01-06 18:16:46 +01:00
lines.append(f'# {self.comment}')
lines.append(f'[{self.name}]')
for option in self.options: # pylint: disable=consider-using-dict-items
2021-11-03 17:38:31 +01:00
lines.append(self.options[option].export_to_config())
return '\n'.join(lines)
def ask_values(self, set_it=True):
"""
Ask user to enter value for each configuration option of the section
:param set_it: If True (default), option value will be updated with user input
:return: If set_it is True, return True if valid value for each configuration
option have been retrieved and set. If False, return a dict of configuration
options and their value.
:rtype: bool of dict
"""
if self.comment:
2023-01-06 18:16:46 +01:00
print(f'# {self.comment}')
print(f'[{self.name}]''\n')
result = {}
error = False
for name, option in self.options.items():
option_result = option.ask_value(set_it=set_it)
if set_it:
result[name] = option_result
elif not option_result:
error = True
print()
print()
if set_it:
return not error
return result
2021-11-03 17:38:31 +01:00
class RawWrappedTextHelpFormatter(argparse.RawDescriptionHelpFormatter):
2021-11-23 13:08:44 +01:00
"""
Custom TextHelpFormatter for argparse.ArgumentParser that allowing line to keep line return
"""
def _split_lines(self, text, width):
result = []
for line in textwrap.dedent(text).splitlines():
# Keep empty line
if line == "":
result.append(line)
continue
# Split ident prefix and line text
m = re.match('^( *)(.*)$', line)
ident = m.group(1)
line_text = m.group(2)
# Wrap each lines and add in result with ident prefix
2022-01-20 19:14:16 +01:00
for subline in textwrap.wrap(line_text, width - len(ident)):
result.append(ident + subline)
return result
2021-11-23 13:08:44 +01:00
class Config: # pylint: disable=too-many-instance-attributes
2021-11-03 17:38:31 +01:00
""" Configuration helper """
def __init__(self, appname, shortname=None, version=None, encoding=None,
config_file_env_variable=None, default_config_dirpath=None):
2021-11-03 17:38:31 +01:00
self.appname = appname
self.shortname = shortname
self.version = version if version else '0.0'
self.encoding = encoding if encoding else 'utf-8'
self.config_parser = None
self.options_parser = None
self.options = None
self.sections = {}
self._loaded_callbacks = []
self._loaded_callbacks_executed = []
2021-11-03 17:38:31 +01:00
self._filepath = None
self.config_file_env_variable = config_file_env_variable
self.default_config_dirpath = default_config_dirpath
2021-11-03 17:38:31 +01:00
def add_section(self, name, loaded_callback=None, **kwargs):
"""
Add section
: param name: The section name
: param loaded_callback: An optional callback method that will be executed after configuration is loaded
The specified callback method will receive Config object as parameter.
: param ** kwargs: Raw parameters dict pass to ConfigSection __init__() method
"""
2023-01-06 18:16:46 +01:00
assert name not in self.sections, f'Duplicated section {name}'
2021-11-03 17:38:31 +01:00
self.sections[name] = ConfigSection(self, name, **kwargs)
if loaded_callback:
self._loaded_callbacks.append(loaded_callback)
# If configuration is already loaded, execute callback immediatly
if self.config_parser or self.options:
self._loaded()
2021-11-03 17:38:31 +01:00
return self.sections[name]
def defined(self, section, option):
""" Check option is defined in specified section """
return section in self.sections and self.sections[section].defined(option)
def isset(self, section, option):
""" Check option is set in specified section """
return section in self.sections and self.sections[section].isset(option)
def get(self, section, option):
""" Get option value """
assert self.config_parser or self.options, 'Unconfigured options parser'
assert self.defined(
2023-01-06 18:16:46 +01:00
section, option), f'Unknown option {section}.{option}'
2021-11-03 17:38:31 +01:00
value = self.sections[section].get(option)
log.debug('get(%s, %s): %s (%s)', section, option, value, type(value))
return value
2022-05-27 15:15:34 +02:00
def get_option(self, option, default=None):
""" Get an argument parser option value """
if self.options and hasattr(self.options, option):
return getattr(self.options, option)
return default
def __getitem__(self, key):
assert self.config_parser or self.options, 'Unconfigured options parser'
assert key in self.sections, f'Unknown section {key}'
return ConfigSectionAsDictWrapper(self.sections[key])
2021-11-03 17:38:31 +01:00
def set(self, section, option, value):
""" Set option value """
assert self.config_parser, 'Unconfigured options parser'
assert self.defined(
2023-01-06 18:16:46 +01:00
section, option), f'Unknown option {section}.{option}'
2021-11-03 17:38:31 +01:00
self.sections[section].set(option, value)
def load_file(self, filepath, execute_callback=True):
2021-11-03 17:38:31 +01:00
""" Read configuration file """
self.config_parser = ConfigParser(defaults={'hash': '#'})
2021-11-03 17:38:31 +01:00
self._filepath = filepath
# Checking access of configuration file
if not os.path.isfile(filepath):
return True
if not os.access(filepath, os.R_OK):
return False
try:
self.config_parser.read(filepath, encoding=self.encoding)
except Exception: # pylint: disable=broad-except
self.config_parser = None
log.exception('Failed to read configuration file %s', filepath)
return False
# Logging initialization
if self.config_parser.has_section('loggers'):
fileConfig(filepath)
else:
# Otherwise, use systemd journal handler
handler = JournalHandler(SYSLOG_IDENTIFIER=self.shortname)
handler.setFormatter(
logging.Formatter(
'%(levelname)s | %(name)s | %(message)s'
)
)
logging.getLogger().addHandler(handler)
self._filepath = filepath
if execute_callback:
self._loaded()
2021-11-03 17:38:31 +01:00
return True
def _loaded(self):
""" Execute loaded callbacks """
error = False
for callback in self._loaded_callbacks:
if callback in self._loaded_callbacks_executed:
continue
if not callback(self):
error = True
self._loaded_callbacks_executed.append(callback)
return not error
2021-11-03 17:38:31 +01:00
def save(self, filepath=None):
""" Save configuration file """
filepath = filepath if filepath else self._filepath
assert filepath, 'Configuration filepath is not set or provided'
# Checking access of target file/directory
2021-11-03 17:38:31 +01:00
dirpath = os.path.dirname(filepath)
if os.path.isfile(filepath):
if not os.access(filepath, os.W_OK):
2022-01-20 19:14:16 +01:00
log.error('Configuration file "%s" is not writable', filepath)
return False
elif not os.path.isdir(dirpath) or not os.access(dirpath, os.R_OK | os.W_OK | os.X_OK):
2021-11-03 17:38:31 +01:00
log.error(
'Configuration directory "%s" does not exist (or not writable)', dirpath)
return False
2023-01-06 18:16:46 +01:00
lines = [
'#\n'
f'# {self.appname} configuration'
'\n#\n'
]
2021-11-03 17:38:31 +01:00
for section_name in self._ordered_section_names:
lines.append('')
lines.append(self.sections[section_name].export_to_config())
try:
with open(filepath, 'wb') as fd:
fd.write(
'\n'.join(lines).encode(self.encoding)
)
# Privacy!
os.chmod(filepath, stat.S_IRUSR | stat.S_IWUSR)
except Exception: # pylint: disable=broad-except
log.exception(
'Failed to write generated configuration file %s', filepath)
return False
self.load_file(filepath)
2021-11-03 17:38:31 +01:00
return True
@property
def _ordered_section_names(self):
""" Get ordered list of section names """
return sorted(self.sections.keys(), key=lambda section: self.sections[section].order)
def get_arguments_parser(self, reconfigure=False, **kwargs):
2021-11-03 17:38:31 +01:00
""" Get arguments parser """
if self.options_parser:
return self.options_parser
self.options_parser = argparse.ArgumentParser(
description=kwargs.pop('description', self.appname),
formatter_class=RawWrappedTextHelpFormatter,
**kwargs)
2021-11-03 17:38:31 +01:00
2023-01-06 18:16:46 +01:00
config_file_help = (
f'Configuration file to use (default: {self.config_filepath})'
)
if self.config_file_env_variable:
2023-01-06 18:16:46 +01:00
config_file_help += (
'\n\nYou also could set '
f'{self.config_file_env_variable} environment variable to '
'specify your configuration file path.'
)
2021-11-03 17:38:31 +01:00
self.options_parser.add_argument(
'-c',
'--config',
default=self.config_filepath,
help=config_file_help
2021-11-03 17:38:31 +01:00
)
self.options_parser.add_argument(
'--save',
action='store_true',
dest='save',
help='Save current configuration to file',
)
if reconfigure:
self.options_parser.add_argument(
'--reconfigure',
action='store_true',
dest='mylib_config_reconfigure',
help='Reconfigure and update configuration file',
)
2021-11-03 17:38:31 +01:00
self.options_parser.add_argument(
'-d',
'--debug',
action='store_true',
help='Show debug messages'
)
self.options_parser.add_argument(
'-v',
'--verbose',
action='store_true',
help='Show verbose messages'
)
section = self.add_section('console', comment='Console logging')
section.add_option(
BooleanOption, 'enabled', default=False,
arg='--console', short_arg='-C',
comment='Enable/disable console log')
2023-01-09 11:25:58 +01:00
section.add_option(
BooleanOption, 'force_stderr', default=False,
arg='--console-stderr',
comment='Force console log on stderr')
section.add_option(
StringOption, 'log_format', default=DEFAULT_CONSOLE_LOG_FORMAT,
arg='--console-log-format', comment='Console log format')
2021-11-18 19:55:44 +01:00
2021-11-03 17:38:31 +01:00
self.add_options_to_parser(self.options_parser)
return self.options_parser
def parse_arguments_options(self, argv=None, parser=None, create=True, ask_values=True,
exit_after_created=True, execute_callback=True):
2021-11-03 17:38:31 +01:00
"""
Parse arguments options
:param argv: Optional arguments list to parse (default: sys.argv[1:])
:param parser: Optional argparse.ArgumentParser use
(default: generated by self.get_arguments_parser())
:param create: If True, configuration file will be created if it does not exits
(default: True)
:param ask_values: If True, ask user to enter valor of each configuration options
(default: True)
2021-11-03 17:38:31 +01:00
:param exit_after_created: If True, script will end after configuration file
creation (default: True)
:param execute_callback: Sections's loaded callbacks will be executed only if True
(default: True)
2021-11-03 17:38:31 +01:00
"""
parser = parser if parser else self.get_arguments_parser()
argcomplete.autocomplete(parser)
options = parser.parse_args(argv if argv is not None else sys.argv[1:])
self.load_options(options, execute_callback=False)
2021-11-03 17:38:31 +01:00
if options.config:
options.config = os.path.abspath(options.config)
already_saved = False
if not os.path.isfile(options.config) and (create or options.save):
log.warning(
"Configuration file is missing, generate it (%s)",
options.config
)
if ask_values:
self.ask_values(set_it=True)
2021-11-03 17:38:31 +01:00
self.save(options.config)
if exit_after_created:
sys.exit(0)
already_saved = True
# Load configuration file
if os.path.isfile(options.config) and not self.load_file(options.config, execute_callback=False):
2021-11-03 17:38:31 +01:00
parser.error(
2023-01-06 18:16:46 +01:00
f'Failed to load configuration from file {options.config}'
2021-11-03 17:38:31 +01:00
)
if options.save and not already_saved:
self.save()
sys.exit(0)
if options.debug:
logging.getLogger().setLevel(logging.DEBUG)
elif options.verbose:
logging.getLogger().setLevel(logging.INFO)
if self.get('console', 'enabled'):
2023-01-09 11:25:58 +01:00
stdout_console_handler = logging.StreamHandler(
sys.stderr if self.get('console', 'force_stderr')
else sys.stdout)
stdout_console_handler.addFilter(StdoutInfoFilter())
stdout_console_handler.setLevel(logging.DEBUG)
stderr_console_handler = logging.StreamHandler(sys.stderr)
stderr_console_handler.setLevel(logging.WARNING)
if self.get('console', 'log_format'):
console_formater = logging.Formatter(self.get('console', 'log_format'))
stdout_console_handler.setFormatter(console_formater)
stderr_console_handler.setFormatter(console_formater)
logging.getLogger().addHandler(stdout_console_handler)
logging.getLogger().addHandler(stderr_console_handler)
2021-11-18 19:55:44 +01:00
if execute_callback:
self._loaded()
if self.get_option('mylib_config_reconfigure', default=False):
if self.ask_values(set_it=True) and self.save():
sys.exit(0)
sys.exit(1)
2021-11-03 17:38:31 +01:00
return options
def load_options(self, options, execute_callback=True):
2021-11-03 17:38:31 +01:00
""" Register arguments parser options """
assert isinstance(options, argparse.Namespace)
self.options = options
log.debug('Argument options: %s', options)
if execute_callback:
self._loaded()
2021-11-03 17:38:31 +01:00
def add_options_to_parser(self, parser):
""" Add sections and their options to parser """
for section in self._ordered_section_names:
self.sections[section].add_options_to_parser(parser)
def ask_values(self, set_it=True, execute_callback=False):
"""
Ask user to enter value for each configuration option
:param set_it: If True (default), option value will be updated with user input
:param execute_callback: Sections's loaded callbacks will be finally executed
(only if set_it is True, default: False)
:return: If set_it is True, return True if valid value for each configuration
option have been retrieved and set. If False, return a dict of configuration
section and their options value.
:rtype: bool of dict
"""
# On set it mode, ensure configuration file parser is initialized
if set_it and not self.config_parser:
self.config_parser = ConfigParser()
2023-01-06 18:16:46 +01:00
result = {}
error = False
for name, section in self.sections.items():
section_result = section.ask_values(set_it=set_it)
if not set_it:
result[name] = section_result
elif not section_result:
error = True
if set_it:
if error:
return False
if execute_callback:
self._loaded()
return True
return result
def configure(self, argv=None, description=False):
"""
Entry point of a script you could use to created your configuration file
Note: make sure to load to define all your configuration section and options
before running it.
"""
parser = self.get_arguments_parser(
description=description if description else "Generate configuration file"
)
parser.add_argument(
'-i', '--interactive',
action='store_true', dest='interactive',
help="Enable configuration interactive mode"
)
parser.add_argument(
'-O', '--overwrite',
action='store_true', dest='overwrite',
help="Overwrite configuration file if exists"
)
parser.add_argument(
'-V', '--validate',
action='store_true', dest='validate',
help=(
"Validate configuration: initialize application to test if provided parameters works.\n\n"
"Note: Validation will occured after configuration file creation or update. On error, "
"re-run with -O/--overwrite parameter to fix it."
)
)
options = self.parse_arguments_options(
argv, create=False, execute_callback=False)
if os.path.exists(options.config) and not options.overwrite:
2023-01-06 18:16:46 +01:00
print(f'Configuration file {options.config} already exists')
sys.exit(1)
if options.interactive:
self.ask_values(set_it=True)
if self.save(options.config):
2023-01-06 18:16:46 +01:00
print(f'Configuration file {options.config} created.')
if options.validate:
print('Validate your configuration...')
try:
if self._loaded():
print('Your configuration seem valid.')
else:
print('Error(s) occurred validating your configuration. See logs for details.')
sys.exit(1)
2023-01-06 18:16:46 +01:00
except Exception: # pylint: disable=broad-except
print(
2023-01-06 18:16:46 +01:00
'Exception occurred validating your configuration:\n'
f'{traceback.format_exc()}'
'\n\nSee logs for details.'
)
sys.exit(2)
else:
2023-01-06 18:16:46 +01:00
print(f'Error occured creating configuration file {options.config}')
sys.exit(1)
sys.exit(0)
2021-11-03 17:38:31 +01:00
@property
def config_dir(self):
""" Retrieve configuration directory path """
if self._filepath:
return os.path.dirname(self._filepath)
if self.default_config_dirpath:
return self.default_config_dirpath
return DEFAULT_CONFIG_DIRPATH
2021-11-03 17:38:31 +01:00
@property
def config_filepath(self):
""" Retrieve configuration file path """
if self._filepath:
return self._filepath
if self.config_file_env_variable and os.environ.get(self.config_file_env_variable):
return os.environ.get(self.config_file_env_variable)
2021-11-03 17:38:31 +01:00
return os.path.join(
self.config_dir,
2023-01-06 18:16:46 +01:00
f'{self.shortname}.ini' if self.shortname
else 'config.ini'
2021-11-03 17:38:31 +01:00
)
class ConfigurableObject:
"""
Base class of configurable object
This class provide a way to configure an object using :
- mylib.config.Config object
- argparse.Namespace object
- kwargs passed to __init__ method
"""
# Configuration object name (used for default options prefix and config section)
# Note: required if options_prefix or/and config_section parameters not provided
# to __init__ method.
_config_name = None
# Configuration comment (used for config section)
_config_comment = None
# Default options value
# Important: all supported options MUST HAVE a default value defined
_defaults = {}
# Store options passed throuht __init__ method
_kwargs = {}
_options = {}
_options_prefix = None
_config = None
_config_section = None
def __init__(self, options=None, options_prefix=None, config=None, config_section=None,
**kwargs):
for key, value in kwargs.items():
2023-01-06 18:16:46 +01:00
assert key in self._defaults, f'Unknown {key} option'
self._kwargs[key] = value
if options:
self._options = options
if options_prefix is not None:
self._options_prefix = options_prefix
elif self._config_name:
self._options_prefix = self._config_name + '_'
else:
2023-01-06 18:16:46 +01:00
raise Exception(f'No configuration name defined for {__name__}')
if config:
self._config = config
if config_section:
self._config_section = config_section
elif self._config_name:
self._config_section = self._config_name
else:
2023-01-06 18:16:46 +01:00
raise Exception(f'No configuration name defined for {__name__}')
def _get_option(self, option, default=None, required=False):
""" Retreive option value """
if self._kwargs and option in self._kwargs:
return self._kwargs[option]
if self._options and hasattr(self._options, self._options_prefix + option):
return getattr(self._options, self._options_prefix + option)
if self._config and self._config.defined(self._config_section, option):
return self._config.get(self._config_section, option)
2023-01-06 18:16:46 +01:00
assert not required, f'Options {option} not defined'
return default if default is not None else self._defaults.get(option)
def configure(self, comment=None, ** kwargs):
""" Configure options on registered mylib.Config object """
assert self._config, "mylib.Config object not registered. Must be passed to __init__ as config keyword argument."
return self._config.add_section(
self._config_section,
comment=comment if comment else self._config_comment,
loaded_callback=self.initialize, **kwargs)
def initialize(self, loaded_config=None):
""" Configuration initialized hook """
if loaded_config:
self.config = loaded_config # pylint: disable=attribute-defined-outside-init
class ConfigSectionAsDictWrapper:
"""
Wrapper for ConfigSection that offer __getitems__ and __setitem__ methods
to allow access to section options as with a dictionary.
"""
__section = None
def __init__(self, section):
self.__section = section
def __getitem__(self, key):
return self.__section.get(key)
def __setitem__(self, key, value):
self.__section.set(key, value)
def __delitem__(self, key):
raise Exception('Deleting a configuration option is not supported')
# pylint: disable=too-few-public-methods
class StdoutInfoFilter(logging.Filter):
""" Logging filter to keep messages only >= logging.INFO """
def filter(self, record):
return record.levelno in (logging.DEBUG, logging.INFO)