diff options
Diffstat (limited to 'scripts/collect_yaml.py')
-rwxr-xr-x | scripts/collect_yaml.py | 74 |
1 files changed, 0 insertions, 74 deletions
diff --git a/scripts/collect_yaml.py b/scripts/collect_yaml.py deleted file mode 100755 index 3147a0c..0000000 --- a/scripts/collect_yaml.py +++ /dev/null @@ -1,74 +0,0 @@ -#!/usr/bin/env python3 -# -# Copyright (c) 2017 Weitian LI <liweitianux@live.com> -# MIT license -# -# Weitian LI -# 2017-02-12 - -""" -Collect YAML manifest files, and convert collected results to CSV -format for later use. -""" - -import sys -import argparse -import csv - -from _context import acispy -from acispy.manifest import Manifest - - -def main(): - parser = argparse.ArgumentParser(description="Collect YAML manifest files") - parser.add_argument("-k", "--keys", dest="keys", required=True, - help="YAML keys to be collected (in order); " + - "can be comma-separated string, or a file " + - "containing the keys one-per-line") - parser.add_argument("-b", "--brief", dest="brief", - action="store_true", - help="be brief and do not print header") - parser.add_argument("-v", "--verbose", dest="verbose", - action="store_true", - help="show verbose information") - parser.add_argument("-o", "--outfile", dest="outfile", default=sys.stdout, - help="output CSV file to save collected data") - parser.add_argument("-i", "--infile", dest="infile", - nargs="+", required=True, - help="list of input YAML manifest files") - args = parser.parse_args() - - try: - keys = [k.strip() for k in open(args.keys).readlines()] - except FileNotFoundError: - keys = [k.strip() for k in args.keys.split(",")] - - if args.verbose: - print("keys:", keys, file=sys.stderr) - print("infile:", args.infile, file=sys.stderr) - print("outfile:", args.outfile, file=sys.stderr) - - results = [] - for fp in args.infile: - manifest = Manifest(fp) - res = manifest.gets(keys, splitlist=True) - if args.verbose: - print("FILE:{0}: {1}".format(fp, list(res.values())), - file=sys.stderr) - results.append(res) - - try: - of = open(args.outfile, "w") - except TypeError: - of = args.outfile - writer = csv.writer(of) - if not args.brief: - writer.writerow(results[0].keys()) - for res in results: - writer.writerow(res.values()) - if of is not sys.stdout: - of.close() - - -if __name__ == "__main__": - main() |