mirror of
https://github.com/haidaraM/ansible-playbook-grapher
synced 2024-09-20 22:22:00 +00:00
91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
import sys
|
|
import os
|
|
import ntpath
|
|
|
|
from ansible.cli import CLI
|
|
from ansible.errors import AnsibleOptionsError
|
|
|
|
from ansibleplaybookgrapher.grapher import Grapher
|
|
|
|
try:
|
|
from __main__ import display
|
|
except ImportError:
|
|
from ansible.utils.display import Display
|
|
|
|
display = Display()
|
|
|
|
from ansibleplaybookgrapher import __prog__, __version__
|
|
|
|
|
|
class PlaybookGrapherCLI(CLI):
|
|
"""
|
|
Playbook grapher CLI
|
|
"""
|
|
|
|
def run(self):
|
|
super(PlaybookGrapherCLI, self).run()
|
|
|
|
playbook = self.args[0]
|
|
|
|
loader, inventory, variable_manager = self._play_prereqs(self.options)
|
|
|
|
grapher = Grapher(data_loader=loader, inventory_manager=inventory, variable_manager=variable_manager,
|
|
playbook_filename=playbook, output_filename=self.options.output_filename)
|
|
|
|
grapher.make_graph(include_role_tasks=self.options.include_role_tasks, tags=self.options.tags,
|
|
skip_tags=self.options.skip_tags)
|
|
|
|
grapher.render_graph(save_dot_file=self.options.save_dot_file)
|
|
|
|
grapher.post_process_svg()
|
|
|
|
def parse(self):
|
|
# create parser for CLI options
|
|
parser = CLI.base_parser(
|
|
usage="%s [options] playbook.yml" % __prog__,
|
|
subset_opts=True,
|
|
vault_opts=True,
|
|
runtask_opts=True,
|
|
desc="Make graph from your Playbook.",
|
|
)
|
|
|
|
parser.add_option('-i', '--inventory', dest='inventory', action="append",
|
|
help="specify inventory host path (default=[%s]) or comma separated host list. ")
|
|
|
|
parser.add_option("--include-role-tasks", dest="include_role_tasks", action='store_true', default=False,
|
|
help="Include the tasks of the role in the graph.")
|
|
|
|
parser.add_option("-s", "--save-dot-file", dest="save_dot_file", action='store_true', default=False,
|
|
help="Save the dot file used to generate the graph.")
|
|
|
|
parser.add_option("-o", "--ouput-file-name", dest='output_filename',
|
|
help="Output filename without the '.svg' extension. Default: <playbook>.svg")
|
|
|
|
parser.version = "%s %s" % (__prog__, __version__)
|
|
|
|
self.parser = parser
|
|
|
|
super(PlaybookGrapherCLI, self).parse()
|
|
|
|
if len(self.args) == 0:
|
|
raise AnsibleOptionsError("You must specify a playbook file to graph.")
|
|
|
|
if len(self.args) > 1:
|
|
raise AnsibleOptionsError("You must specify only one playbook file to graph.")
|
|
|
|
display.verbosity = self.options.verbosity
|
|
|
|
if self.options.output_filename is None:
|
|
self.options.output_filename = os.path.splitext(ntpath.basename(self.args[0]))[0]
|
|
|
|
|
|
def main():
|
|
cli = PlaybookGrapherCLI(sys.argv)
|
|
|
|
cli.parse()
|
|
|
|
cli.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|