feat(naive-python): Add grammar via inflect, improve overall handling
This commit is contained in:
		| @@ -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,13 +161,41 @@ 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: | ||||||
|             for key in config_obj.options(this_section, no_defaults=True): |             for key in config_obj.options(this_section, no_defaults=True): | ||||||
| @@ -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}]' ...") | ||||||
|         # ... |         # ... | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user