puppet-lint.py

#!/usr/bin/env python3

import concurrent.futures
import subprocess
import sys


def colored(txt, clr):
    if clr == "red":
        clr_code = "\033[31m"
    elif clr == "green":
        clr_code = "\033[32m"
    return "{}{}\033[0m".format(clr_code, txt)

paths = ["manifests",
         "modules/bayncore",
         "modules/dtg",
         "modules/exim",
         "modules/gpg",
         "modules/munin",
         "modules/nagios"]


def puppet_lint(path, doc=False):
    args = ["puppet-lint", "--fail-on-warnings", "--error-level", "warning"]
    if doc:
        args += ["--only-checks", "documentation"]
    else:
        args += ["--no-documentation-check"]
    args += [path]
    return path, subprocess.run(args,
                                stdout=subprocess.PIPE,
                                stderr=subprocess.STDOUT)

exit_code = 0

if sys.argv[1] == "--doc":
    doc = True
    mode_txt = "doc"
elif sys.argv[1] == "--err":
    doc = False
    mode_txt = "err"
else:
    print("Error, must specify either --err or --doc")
    sys.exit(1)

with concurrent.futures.ThreadPoolExecutor(max_workers=6) as ex:
    for ftr in [ex.submit(puppet_lint, path, doc) for path in paths]:
        path, res = ftr.result()

        if res.returncode == 0:
            print(colored("puppet-lint-{}({}): SUCCESS".format(mode_txt,
                                                               path),
                          "green"))
        else:
            print(colored("puppet-lint-{}({}): FAILURE".format(mode_txt,
                                                               path),
                          "red"))
            exit_code = 1
            print(res.stdout.decode("UTF-8").strip())
        sys.stdout.flush()
sys.exit(exit_code)