ansible-doctor/ansibledoctor/Config.py

298 lines
9.6 KiB
Python
Raw Normal View History

2019-10-08 09:39:27 +00:00
#!/usr/bin/env python3
"""Global settings definition."""
2019-10-07 06:52:00 +00:00
import os
import sys
2019-10-07 06:52:00 +00:00
import anyconfig
2019-10-08 22:59:40 +00:00
import environs
2019-10-08 09:30:31 +00:00
import jsonschema.exceptions
2019-10-09 21:21:45 +00:00
import logging
2019-10-08 09:39:27 +00:00
import yaml
from appdirs import AppDirs
from jsonschema._utils import format_as_index
from pkg_resources import resource_filename
2019-10-07 06:52:00 +00:00
2019-10-08 09:30:31 +00:00
import ansibledoctor.Exception
2019-10-08 09:39:27 +00:00
from ansibledoctor.Utils import Singleton
2019-10-07 06:52:00 +00:00
config_dir = AppDirs("ansible-doctor").user_config_dir
default_config_file = os.path.join(config_dir, "config.yml")
class Config():
"""
Create an object with all necessary settings.
Settings are loade from multiple locations in defined order (last wins):
- default settings defined by `self._get_defaults()`
- yaml config file, defaults to OS specific user config dir (https://pypi.org/project/appdirs/)
- provides cli parameters
"""
2019-10-08 22:56:39 +00:00
SETTINGS = {
"config_file": {
"default": "",
"env": "CONFIG_FILE",
"type": environs.Env().str
},
"role_dir": {
2019-10-08 22:56:39 +00:00
"default": "",
"env": "ROLE_DIR",
2019-10-08 22:56:39 +00:00
"type": environs.Env().str
},
"dry_run": {
"default": False,
"env": "DRY_RUN",
"file": True,
"type": environs.Env().bool
},
"logging.level": {
"default": "WARNING",
"env": "LOG_LEVEL",
"file": True,
"type": environs.Env().str
},
"logging.json": {
"default": False,
"env": "LOG_JSON",
"file": True,
"type": environs.Env().bool
},
"output_dir": {
"default": os.getcwd(),
"env": "OUTPUT_DIR",
"file": True,
"type": environs.Env().str
},
"template_dir": {
"default": os.path.join(os.path.dirname(os.path.realpath(__file__)), "templates"),
"env": "TEMPLATE_DIR",
"file": True,
"type": environs.Env().str
},
"template": {
"default": "readme",
"env": "TEMPLATE",
"file": True,
"type": environs.Env().str
},
"force_overwrite": {
"default": False,
"env": "FORCE_OVERWRITE",
"file": True,
"type": environs.Env().bool
},
"custom_header": {
"default": "",
"env": "CUSTOM_HEADER",
"file": True,
"type": environs.Env().str
},
"exclude_files": {
"default": [],
"env": "EXCLUDE_FILES",
"file": True,
"type": environs.Env().list
},
}
ANNOTATIONS = {
"meta": {
"name": "meta",
"automatic": True
},
"todo": {
"name": "todo",
"automatic": True,
},
"var": {
"name": "var",
"automatic": True,
},
"example": {
"name": "example",
"regex": r"(\#\ *\@example\ *\: *.*)"
},
"tag": {
"name": "tag",
"automatic": True,
},
}
def __init__(self, args={}):
"""
Initialize a new settings class.
2019-10-07 06:52:00 +00:00
:param args: An optional dict of options, arguments and commands from the CLI.
:param config_file: An optional path to a yaml config file.
:returns: None
"""
2019-10-08 22:56:39 +00:00
self._args = args
self._schema = None
self.config_file = default_config_file
self.role_dir = os.getcwd()
2019-10-08 22:56:39 +00:00
self.config = None
self._set_config()
self.is_role = self._set_is_role() or False
2019-10-08 22:56:39 +00:00
def _get_args(self, args):
cleaned = dict(filter(lambda item: item[1] is not None, args.items()))
2019-10-08 22:56:39 +00:00
normalized = {}
for key, value in cleaned.items():
normalized = self._add_dict_branch(normalized, key.split("."), value)
# Override correct log level from argparse
levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
2019-10-08 22:56:39 +00:00
log_level = levels.index(self.SETTINGS["logging.level"]["default"])
if normalized.get("logging"):
for adjustment in normalized["logging"]["level"]:
log_level = min(len(levels) - 1, max(log_level + adjustment, 0))
2019-10-08 22:56:39 +00:00
normalized["logging"]["level"] = levels[log_level]
2019-10-08 22:56:39 +00:00
return normalized
def _get_defaults(self):
2019-10-08 22:56:39 +00:00
normalized = {}
for key, item in self.SETTINGS.items():
normalized = self._add_dict_branch(normalized, key.split("."), item["default"])
self.schema = anyconfig.gen_schema(normalized)
return normalized
def _get_envs(self):
normalized = {}
for key, item in self.SETTINGS.items():
if item.get("env"):
prefix = "ANSIBLE_DOCTOR_"
envname = prefix + item["env"]
try:
value = item["type"](envname)
normalized = self._add_dict_branch(normalized, key.split("."), value)
except environs.EnvError as e:
if '"{}" not set'.format(envname) in str(e):
pass
else:
2019-10-08 22:59:40 +00:00
raise ansibledoctor.Exception.ConfigError("Unable to read environment variable", str(e))
2019-10-08 22:56:39 +00:00
return normalized
def _set_config(self):
args = self._get_args(self._args)
envs = self._get_envs()
defaults = self._get_defaults()
2019-10-08 22:56:39 +00:00
# preset config file path
if envs.get("config_file"):
self.config_file = self._normalize_path(envs.get("config_file"))
if envs.get("role_dir"):
self.role_dir = self._normalize_path(envs.get("role_dir"))
2019-10-08 22:56:39 +00:00
if args.get("config_file"):
self.config_file = self._normalize_path(args.get("config_file"))
if args.get("role_dir"):
self.role_dir = self._normalize_path(args.get("role_dir"))
2019-10-08 22:56:39 +00:00
source_files = []
source_files.append(self.config_file)
source_files.append(os.path.join(os.getcwd(), ".ansibledoctor"))
source_files.append(os.path.join(os.getcwd(), ".ansibledoctor.yml"))
source_files.append(os.path.join(os.getcwd(), ".ansibledoctor.yaml"))
for config in source_files:
if config and os.path.exists(config):
with open(config, "r", encoding="utf8") as stream:
s = stream.read()
2019-10-08 09:30:31 +00:00
try:
2019-10-08 22:56:39 +00:00
file_dict = yaml.safe_load(s)
2019-10-08 09:30:31 +00:00
except yaml.parser.ParserError as e:
message = "{}\n{}".format(e.problem, str(e.problem_mark))
raise ansibledoctor.Exception.ConfigError("Unable to read file", message)
2019-10-08 22:56:39 +00:00
if self._validate(file_dict):
anyconfig.merge(defaults, file_dict, ac_merge=anyconfig.MS_DICTS)
defaults["logging"]["level"] = defaults["logging"]["level"].upper()
2019-10-08 22:56:39 +00:00
if self._validate(envs):
anyconfig.merge(defaults, envs, ac_merge=anyconfig.MS_DICTS)
2019-10-08 22:56:39 +00:00
if self._validate(args):
anyconfig.merge(defaults, args, ac_merge=anyconfig.MS_DICTS)
fix_files = ["output_dir", "template_dir", "custom_header"]
for file in fix_files:
if defaults[file] and defaults[file] != "":
defaults[file] = self._normalize_path(defaults[file])
2019-10-08 22:56:39 +00:00
if "config_file" in defaults:
2019-10-08 22:56:39 +00:00
defaults.pop("config_file")
if "role_dir" in defaults:
defaults.pop("role_dir")
defaults["logging"]["level"] = defaults["logging"]["level"].upper()
2019-10-08 22:56:39 +00:00
self.config = defaults
def _normalize_path(self, path):
if not os.path.isabs(path):
base = os.path.join(os.getcwd(), path)
return os.path.abspath(os.path.expanduser(os.path.expandvars(base)))
2019-10-08 09:30:31 +00:00
else:
2019-10-08 22:56:39 +00:00
return path
2019-10-08 09:30:31 +00:00
def _set_is_role(self):
if os.path.isdir(os.path.join(self.role_dir, "tasks")):
2019-10-08 09:30:31 +00:00
return True
def _validate(self, config):
try:
anyconfig.validate(config, self.schema, ac_schema_safe=False)
2019-10-08 09:30:31 +00:00
except jsonschema.exceptions.ValidationError as e:
schema_error = "Failed validating '{validator}' in schema{schema}\n{message}".format(
validator=e.validator,
2019-10-08 09:30:31 +00:00
schema=format_as_index(list(e.relative_schema_path)[:-1]),
message=e.message
)
2019-10-08 09:30:31 +00:00
raise ansibledoctor.Exception.ConfigError("Configuration error", schema_error)
2019-10-08 09:30:31 +00:00
return True
def _add_dict_branch(self, tree, vector, value):
key = vector[0]
tree[key] = value \
if len(vector) == 1 \
2019-10-08 22:56:39 +00:00
else self._add_dict_branch(tree[key] if key in tree else {}, vector[1:], value)
return tree
2019-10-07 06:52:00 +00:00
def get_annotations_definition(self, automatic=True):
annotations = {}
if automatic:
2019-10-08 22:56:39 +00:00
for k, item in self.ANNOTATIONS.items():
2019-10-07 06:52:00 +00:00
if "automatic" in item.keys() and item["automatic"]:
annotations[k] = item
return annotations
def get_annotations_names(self, automatic=True):
annotations = []
if automatic:
2019-10-08 22:56:39 +00:00
for k, item in self.ANNOTATIONS.items():
2019-10-07 06:52:00 +00:00
if "automatic" in item.keys() and item["automatic"]:
annotations.append(k)
return annotations
def get_template(self):
2019-10-07 06:52:00 +00:00
"""
Get the base dir for the template to use.
:return: str abs path
"""
template_dir = self.config.get("template_dir")
template = self.config.get("template")
return os.path.realpath(os.path.join(template_dir, template))
2019-10-07 06:52:00 +00:00
class SingleConfig(Config, metaclass=Singleton):
pass