feat(naive-python): Add grammar via inflect, improve overall handling
This commit is contained in:
parent
fb835f4f40
commit
bfbc829f91
@ -1 +1,2 @@
|
|||||||
rich
|
rich
|
||||||
|
inflect
|
||||||
|
@ -6,6 +6,8 @@
|
|||||||
#
|
#
|
||||||
commonmark==0.9.1
|
commonmark==0.9.1
|
||||||
# via rich
|
# via rich
|
||||||
|
inflect==5.6.0
|
||||||
|
# via -r requirements.in
|
||||||
pygments==2.12.0
|
pygments==2.12.0
|
||||||
# via rich
|
# via rich
|
||||||
rich==12.4.4
|
rich==12.4.4
|
||||||
|
@ -1,13 +1,20 @@
|
|||||||
|
# Path and env manipulation
|
||||||
import os
|
import os
|
||||||
|
# Use a config file
|
||||||
import configparser
|
import configparser
|
||||||
|
# Exit with various exit codes
|
||||||
import sys
|
import sys
|
||||||
|
# Manipulate style and content of logs
|
||||||
import logging
|
import logging
|
||||||
from rich.logging import RichHandler
|
from rich.logging import RichHandler
|
||||||
|
# Correctly generate plurals, singular nouns etc.
|
||||||
|
import inflect
|
||||||
|
|
||||||
|
|
||||||
# Exit codes
|
# Exit codes
|
||||||
# 1: Config file invalid, it has no sections
|
# 1: Config file invalid, it has no sections
|
||||||
# 2: Config file invalid, sections must define at least CONST.CFG_MANDATORY
|
# 2: Config file invalid, sections must define at least CONST.CFG_MANDATORY
|
||||||
|
# 7 : An option that must have a non-null value is either unset or null
|
||||||
|
|
||||||
|
|
||||||
class CONST(object):
|
class CONST(object):
|
||||||
@ -20,42 +27,55 @@ class CONST(object):
|
|||||||
# Values you don't have to set, these are their internal defaults. You may optionally add a key 'is_global' equal
|
# Values you don't have to set, these are their internal defaults. You may optionally add a key 'is_global' equal
|
||||||
# to either True or False. By default if left off it'll be assumed False. Script will treat values where
|
# to either True or False. By default if left off it'll be assumed False. Script will treat values where
|
||||||
# 'is_global' equals True as not being overridable in a '[section]'. It's a setting that only makes sense in a
|
# 'is_global' equals True as not being overridable in a '[section]'. It's a setting that only makes sense in a
|
||||||
# global context for the entire script.
|
# global context for the entire script. An option where 'empty_ok' equals True can safely be unset or set to
|
||||||
|
# an empty string. An example config.ini file may give a sane config example value here, removing that value
|
||||||
|
# still results in a valid file.
|
||||||
CFG_KNOWN_DEFAULTS = [
|
CFG_KNOWN_DEFAULTS = [
|
||||||
{"key": "self_name", "value": "rich-and-config"},
|
{"key": "self_name", "value": "rich-and-config", "empty_ok": False},
|
||||||
{"key": "tmp_base_dir", "value": os.path.join(CFG_THIS_FILE_DIRNAME, "data/tmp/%(self_name)s")},
|
{"key": "tmp_base_dir", "value": os.path.join(CFG_THIS_FILE_DIRNAME, "data/tmp/%(self_name)s"),
|
||||||
{"key": "state_base_dir", "value": os.path.join(CFG_THIS_FILE_DIRNAME, "data/var/lib/%(self_name)s")},
|
"empty_ok": False},
|
||||||
{"key": "state_files_dir", "value": "%(state_base_dir)s/state", "is_global": False},
|
{"key": "state_base_dir", "value": os.path.join(CFG_THIS_FILE_DIRNAME, "data/var/lib/%(self_name)s"),
|
||||||
{"key": "state_file_retention", "value": "50", "is_global": False},
|
"empty_ok": False},
|
||||||
{"key": "state_file_name_prefix", "value": "state-", "is_global": False},
|
{"key": "state_files_dir", "value": "%(state_base_dir)s/state", "is_global": False, "empty_ok": False},
|
||||||
{"key": "state_file_name_suffix", "value": ".log", "is_global": False},
|
{"key": "state_file_retention", "value": "50", "is_global": False, "empty_ok": True},
|
||||||
{"key": "rich_and_config_some_option", "value": "http://localhost:8000/api/query", "is_global": True},
|
{"key": "state_file_name_prefix", "value": "state-", "is_global": False, "empty_ok": True},
|
||||||
{"key": "another_option", "value": "first", "is_global": True}
|
{"key": "state_file_name_suffix", "value": ".log", "is_global": False, "empty_ok": True},
|
||||||
|
{"key": "rich_and_config_some_option", "value": "http://localhost:8000/api/query", "is_global": True,
|
||||||
|
"empty_ok": False},
|
||||||
|
{"key": "another_option", "value": "first", "is_global": True, "empty_ok": True}
|
||||||
]
|
]
|
||||||
# In all sections other than 'default' the following settings are known and accepted. We silently ignore other
|
# In all sections other than 'default' the following settings are known and accepted. We ignore other settings.
|
||||||
# settings. We use 'is_mandatory' to determine if we have to raise errors on missing settings.
|
# Per CFG_KNOWN_DEFAULTS above most '[DEFAULT]' options are accepted by virtue of being defaults and overridable.
|
||||||
|
# The only exception are options where "is_global" equals True, they can't be overridden in '[sections]'; any
|
||||||
|
# attempt at doing it anyway will be ignored. The main purpose of this list is to name settings that do not have
|
||||||
|
# a default value but can - if set - influence how a '[section]' behaves. Repeating a '[DEFAULT]' here does not
|
||||||
|
# make sense. We use 'is_mandatory' to determine if we have to raise errors on missing settings. Here
|
||||||
|
# 'is_mandatory' means the setting must be given in a '[section]'. It may be empty.
|
||||||
CFG_KNOWN_SECTION = [
|
CFG_KNOWN_SECTION = [
|
||||||
{"key": "min_duration", "is_mandatory": False},
|
# {"key": "an_option", "is_mandatory": True},
|
||||||
{"key": "max_duration", "is_mandatory": False},
|
# {"key": "another_one", "is_mandatory": False}
|
||||||
{"key": "title_not_regex", "is_mandatory": False},
|
|
||||||
{"key": "query", "is_mandatory": True},
|
|
||||||
{"key": "dl_dir", "is_mandatory": True}
|
|
||||||
]
|
]
|
||||||
CFG_MANDATORY = [section_cfg["key"] for section_cfg in CFG_KNOWN_SECTION if section_cfg["is_mandatory"]]
|
CFG_MANDATORY = [section_cfg["key"] for section_cfg in CFG_KNOWN_SECTION if section_cfg["is_mandatory"]]
|
||||||
|
|
||||||
|
|
||||||
|
is_systemd = any([systemd_env_var in os.environ for systemd_env_var in ["SYSTEMD_EXEC_PID", "INVOCATION_ID"]])
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
# Default for all modules is NOTSET so log everything
|
# Default for all modules is NOTSET so log everything
|
||||||
level="NOTSET",
|
level="NOTSET",
|
||||||
format=CONST.LOG_FORMAT,
|
format=CONST.LOG_FORMAT,
|
||||||
datefmt="[%X]",
|
datefmt="[%X]",
|
||||||
handlers=[RichHandler(
|
handlers=[RichHandler(
|
||||||
|
show_time=False if is_systemd else True,
|
||||||
|
show_path=False if is_systemd else True,
|
||||||
|
show_level=False if is_systemd else True,
|
||||||
rich_tracebacks=True
|
rich_tracebacks=True
|
||||||
)]
|
)]
|
||||||
)
|
)
|
||||||
log = logging.getLogger("rich")
|
log = logging.getLogger("rich")
|
||||||
# Our own code logs with this level
|
# Our own code logs with this level
|
||||||
log.setLevel(logging.DEBUG)
|
log.setLevel(os.environ.get("LOGLEVEL") if "LOGLEVEL" in [k for k, v in os.environ.items()] else logging.INFO)
|
||||||
|
|
||||||
|
p = inflect.engine()
|
||||||
|
|
||||||
|
|
||||||
# Use this version of class ConfigParser to log.debug contents of our config file. When parsing sections other than
|
# Use this version of class ConfigParser to log.debug contents of our config file. When parsing sections other than
|
||||||
@ -80,7 +100,9 @@ class ConfigParser(
|
|||||||
ini_defaults = []
|
ini_defaults = []
|
||||||
internal_defaults = {default["key"]: default["value"] for default in CONST.CFG_KNOWN_DEFAULTS}
|
internal_defaults = {default["key"]: default["value"] for default in CONST.CFG_KNOWN_DEFAULTS}
|
||||||
internal_globals = [default["key"] for default in CONST.CFG_KNOWN_DEFAULTS if default["is_global"]]
|
internal_globals = [default["key"] for default in CONST.CFG_KNOWN_DEFAULTS if default["is_global"]]
|
||||||
config = ConfigParser(defaults=internal_defaults)
|
internal_empty_ok = [default["key"] for default in CONST.CFG_KNOWN_DEFAULTS if default["empty_ok"]]
|
||||||
|
config = ConfigParser(defaults=internal_defaults,
|
||||||
|
converters={'list': lambda x: [i.strip() for i in x.split(',') if len(x) > 0]})
|
||||||
config.read(CONST.CFG_DEFAULT_ABS_PATH)
|
config.read(CONST.CFG_DEFAULT_ABS_PATH)
|
||||||
|
|
||||||
|
|
||||||
@ -139,12 +161,40 @@ def is_same_as_default(
|
|||||||
return config_kv_pair in ini_defaults
|
return config_kv_pair in ini_defaults
|
||||||
|
|
||||||
|
|
||||||
|
def we_have_unset_options(
|
||||||
|
config_obj: configparser.ConfigParser(),
|
||||||
|
section_name: str) -> list:
|
||||||
|
|
||||||
|
options_must_be_non_empty = []
|
||||||
|
|
||||||
|
for option in config_obj.options(section_name):
|
||||||
|
if not config_obj.get(section_name, option):
|
||||||
|
if option not in internal_empty_ok:
|
||||||
|
log.warning(f"In section '[{section_name}]' option '{option}' is empty, it mustn't be.")
|
||||||
|
options_must_be_non_empty.append(option)
|
||||||
|
|
||||||
|
return options_must_be_non_empty
|
||||||
|
|
||||||
|
|
||||||
def validate_config_sections(
|
def validate_config_sections(
|
||||||
config_obj: configparser.ConfigParser()) -> None:
|
config_obj: configparser.ConfigParser()) -> None:
|
||||||
for this_section in config_obj.sections():
|
for this_section in config_obj.sections():
|
||||||
log.debug(print_section_header(this_section))
|
log.debug(print_section_header(this_section))
|
||||||
|
|
||||||
|
unset_options = we_have_unset_options(config_obj, this_section)
|
||||||
|
if unset_options:
|
||||||
|
log.error(f"""{p.plural("Option", len(unset_options))} {unset_options} """
|
||||||
|
f"""{p.plural("is", len(unset_options))} unset. """
|
||||||
|
f"""{p.singular_noun("They", len(unset_options))} """
|
||||||
|
f"must have a non-null value. "
|
||||||
|
f"""{p.plural("Default", len(unset_options))} {p.plural("is", len(unset_options))}:""")
|
||||||
|
for unset_option in unset_options:
|
||||||
|
log.error(f"{unset_option} = {internal_defaults[unset_option]}")
|
||||||
|
log.error(f"Exiting 7 ...")
|
||||||
|
sys.exit(7)
|
||||||
|
|
||||||
if not set(CONST.CFG_MANDATORY).issubset(config_obj.options(this_section, no_defaults=True)):
|
if not set(CONST.CFG_MANDATORY).issubset(config_obj.options(this_section, no_defaults=True)):
|
||||||
log.debug(f"Config section '[{this_section}]' does not have all mandatory options "
|
log.warning(f"Config section '[{this_section}]' does not have all mandatory options "
|
||||||
f"{CONST.CFG_MANDATORY} set, skipping section ...")
|
f"{CONST.CFG_MANDATORY} set, skipping section ...")
|
||||||
config_obj.remove_section(this_section)
|
config_obj.remove_section(this_section)
|
||||||
else:
|
else:
|
||||||
@ -172,16 +222,16 @@ def an_important_function(
|
|||||||
return ["I", "am", "a", "list"]
|
return ["I", "am", "a", "list"]
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == "__main__":
|
||||||
validate_default_section(config)
|
validate_default_section(config)
|
||||||
if config_has_valid_section(config):
|
if config_has_valid_section(config):
|
||||||
validate_config_sections(config)
|
validate_config_sections(config)
|
||||||
else:
|
else:
|
||||||
log.debug(f"No valid config section found. A valid config section has at least the mandatory options "
|
log.error(f"No valid config section found. A valid config section has at least the mandatory options "
|
||||||
f"{CONST.CFG_MANDATORY} set. Exiting 2 ...")
|
f"{CONST.CFG_MANDATORY} set. Exiting 2 ...")
|
||||||
sys.exit(2)
|
sys.exit(2)
|
||||||
|
|
||||||
log.debug(f"Iterating over config sections ...")
|
log.debug(f"Iterating over config sections ...")
|
||||||
for section in config.sections():
|
for section in config.sections():
|
||||||
log.debug(f"Processing section '[{section}]' ...")
|
log.info(f"Processing section '[{section}]' ...")
|
||||||
# ...
|
# ...
|
||||||
|
Loading…
x
Reference in New Issue
Block a user