github/ansible-doctor
0
0
Fork 0
mirror of https://github.com/thegeeklab/ansible-doctor.git synced 2024-11-25 06:10:43 +00:00
Code Issues Projects Releases Wiki Activity

fix handling of working dir and rename base_dir to role_dir

Browse Source
This commit is contained in:
Robert Kaussow 2019-10-09 23:19:57 +02:00
parent e5b64c094c
commit beb81f348f
4 changed files with 33 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ansibledoctor/Cli.py
View File

@ -35,7 +35,7 @@ class AnsibleDoctor:
# TODO: add function to print to stdout instead of file # TODO: add function to print to stdout instead of file
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
description="Generate documentation from annotated Ansible roles using templates") description="Generate documentation from annotated Ansible roles using templates")
parser.add_argument("base_dir", nargs="?", help="role directory, (default: current working dir)") parser.add_argument("role_dir", nargs="?", help="role directory, (default: current working dir)")
parser.add_argument("-c", "--config", nargs="?", dest="config_file", help="location of configuration file") parser.add_argument("-c", "--config", nargs="?", dest="config_file", help="location of configuration file")
parser.add_argument("-o", "--output", action="store", dest="output_dir", type=str, parser.add_argument("-o", "--output", action="store", dest="output_dir", type=str,
help="output base dir") help="output base dir")

41
ansibledoctor/Config.py
View File

@ -35,9 +35,9 @@ class Config():
"env": "CONFIG_FILE", "env": "CONFIG_FILE",
"type": environs.Env().str "type": environs.Env().str
}, },
"base_dir": { "role_dir": {
"default": "", "default": "",
"env": "BASE_DIR", "env": "ROLE_DIR",
"type": environs.Env().str "type": environs.Env().str
}, },
"dry_run": { "dry_run": {
@ -131,7 +131,7 @@ class Config():
self._args = args self._args = args
self._schema = None self._schema = None
self.config_file = default_config_file self.config_file = default_config_file
self.base_dir = os.getcwd() self.role_dir = os.getcwd()
self.config = None self.config = None
self._set_config() self._set_config()
self.is_role = self._set_is_role() or False self.is_role = self._set_is_role() or False
@ -186,18 +186,19 @@ class Config():
# preset config file path # preset config file path
if envs.get("config_file"): if envs.get("config_file"):
self.config_file = self._normalize_path(envs.get("config_file")) self.config_file = self._normalize_path(envs.get("config_file"))
if envs.get("base_dir"): if envs.get("role_dir"):
self.base_dir = self._normalize_path(envs.get("base_dir")) self.role_dir = self._normalize_path(envs.get("role_dir"))
if args.get("config_file"): if args.get("config_file"):
self.config_file = self._normalize_path(args.get("config_file")) self.config_file = self._normalize_path(args.get("config_file"))
if args.get("base_dir"): if args.get("role_dir"):
self.base_dir = self._normalize_path(args.get("base_dir")) self.role_dir = self._normalize_path(args.get("role_dir"))
source_files = [] source_files = []
source_files.append(self.config_file) source_files.append(self.config_file)
source_files.append(os.path.join(self.base_dir, ".ansibledoctor")) source_files.append(os.path.join(os.getcwd(), ".ansibledoctor"))
source_files.append(os.path.join(self.base_dir, ".ansibledoctor.yml")) source_files.append(os.path.join(os.getcwd(), ".ansibledoctor.yml"))
source_files.append(os.path.join(self.base_dir, ".ansibledoctor.yaml")) source_files.append(os.path.join(os.getcwd(), ".ansibledoctor.yaml"))
for config in source_files: for config in source_files:
if config and os.path.exists(config): if config and os.path.exists(config):
@ -219,25 +220,29 @@ class Config():
if self._validate(args): if self._validate(args):
anyconfig.merge(defaults, args, ac_merge=anyconfig.MS_DICTS) anyconfig.merge(defaults, args, ac_merge=anyconfig.MS_DICTS)
defaults["output_dir"] = self._normalize_path(defaults["output_dir"]) fix_files = ["output_dir", "template_dir", "custom_header"]
defaults["template_dir"] = self._normalize_path(defaults["template_dir"]) for file in fix_files:
defaults["custom_header"] = self._normalize_path(defaults["custom_header"]) if defaults[file] and defaults[file] != "":
defaults[file] = self._normalize_path(defaults[file])
if defaults.get("config_file"): if "config_file" in defaults:
defaults.pop("config_file") defaults.pop("config_file")
if defaults.get("base_dir"): if "role_dir" in defaults:
defaults.pop("base_dir") defaults.pop("role_dir")
defaults["logging"]["level"] = defaults["logging"]["level"].upper()
self.config = defaults self.config = defaults
def _normalize_path(self, path): def _normalize_path(self, path):
if not os.path.isabs(path): if not os.path.isabs(path):
return os.path.abspath(os.path.expanduser(os.path.expandvars(path))) base = os.path.join(os.getcwd(), path)
return os.path.abspath(os.path.expanduser(os.path.expandvars(base)))
else: else:
return path return path
def _set_is_role(self): def _set_is_role(self):
if os.path.isdir(os.path.join(self.base_dir, "tasks")): if os.path.isdir(os.path.join(self.role_dir, "tasks")):
return True return True
def _validate(self, config): def _validate(self, config):

6
ansibledoctor/DocumentationGenerator.py
View File

@ -41,10 +41,10 @@ class Generator:
:return: None :return: None
""" """
base_dir = self.config.get_template() role_dir = self.config.get_template()
for file in glob.iglob(base_dir + "/**/*." + self.extension, recursive=True): for file in glob.iglob(role_dir + "/**/*." + self.extension, recursive=True):
relative_file = file[len(base_dir) + 1:] relative_file = file[len(role_dir) + 1:]
if ntpath.basename(file)[:1] != "_": if ntpath.basename(file)[:1] != "_":
self.logger.debug("Found template file: " + relative_file) self.logger.debug("Found template file: " + relative_file)
self.template_files.append(relative_file) self.template_files.append(relative_file)

12
ansibledoctor/FileRegistry.py
View File

@ -35,18 +35,18 @@ class Registry:
:return: None :return: None
""" """
extensions = YAML_EXTENSIONS extensions = YAML_EXTENSIONS
base_dir = self.config.base_dir role_dir = self.config.role_dir
role_name = os.path.basename(base_dir) role_name = os.path.basename(role_dir)
excludes = self.config.config.get("exclude_files") excludes = self.config.config.get("exclude_files")
excludespec = pathspec.PathSpec.from_lines("gitwildmatch", excludes) excludespec = pathspec.PathSpec.from_lines("gitwildmatch", excludes)
self.log.debug("Scan for files: " + base_dir) self.log.debug("Scan for files: " + role_dir)
for extension in extensions: for extension in extensions:
pattern = os.path.join(base_dir, "**/*." + extension) pattern = os.path.join(role_dir, "**/*." + extension)
for filename in glob.iglob(pattern, recursive=True): for filename in glob.iglob(pattern, recursive=True):
if not excludespec.match_file(filename): if not excludespec.match_file(filename):
self.log.debug("Adding file to '{}': {}".format(role_name, os.path.relpath(filename, base_dir))) self.log.debug("Adding file to '{}': {}".format(role_name, os.path.relpath(filename, role_dir)))
self._doc.append(filename) self._doc.append(filename)
else: else:
self.log.debug("Excluding file: {}".format(os.path.relpath(filename, base_dir))) self.log.debug("Excluding file: {}".format(os.path.relpath(filename, role_dir)))