puppetlabs/infinitory
2
0
Fork 0
mirror of https://github.com/puppetlabs/infinitory.git synced 2026-01-26 10:18:41 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding index json document so this can be hit like an API, and used by other apps

Browse source
This commit is contained in:
Mikker Gimenez-Peterson 2017-12-05 15:59:55 -08:00
parent 3aead36578
commit a4b1a86031
4 changed files with 13 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
infinitory/cli.py
View file

@ -4,6 +4,7 @@ import click
import csv
from datetime import datetime
import jinja2
import json
import logging
import os
import markdown2
@ -90,6 +91,8 @@ def output_html(inventory, directory):
for node in nodes:
csv_writer.writerow([cell.body_csv(node) for cell in all_columns])
write_json(nodes, directory, "index")
for node in nodes:
path = "{}/nodes/{}.html".format(directory, node["certname"])
with open(path, "w", encoding="utf-8") as html:
@ -127,7 +130,6 @@ def output_html(inventory, directory):
generation_time=generation_time,
service=service))
def render_template(template_name, **kwargs):
data_path = os.path.dirname(os.path.abspath(__file__))
environment = jinja2.Environment(
@ -174,6 +176,11 @@ def set_up_logging(level=logging.WARNING):
logging.getLogger("paramiko").setLevel(logging.FATAL)
def write_json(nodes, directory, filename):
path = "{}/nodes/{}.json".format(directory, filename)
with open(path, "w", encoding="utf-8") as json_out:
json_out.write(json.dumps(nodes))
@click.command()
@click.option("--output", "-o", required=True, metavar="PATH", help="Directory to put report in. WARNING: this directory will be removed if it already exists.")
@click.option("--host", "-h", default="localhost", metavar="HOST", help="PuppetDB host to query")
@ -182,7 +189,6 @@ def set_up_logging(level=logging.WARNING):
@click.version_option()
def main(host, output, verbose, debug):
"""Generate SRE inventory report"""
if debug:
set_up_logging(logging.DEBUG)
elif verbose: