From ccbdf67e36e7819548d75de9b89b2fabe47f3001 Mon Sep 17 00:00:00 2001 From: Vishwanatha Subbanna Date: Mon, 23 Jan 2017 14:39:06 +0530 Subject: Update parse_led.py to accept command line arguments Existing parse_led.py had a hardcoded reference to the current directory and the file name 'led.yaml'. This patch introduces changes through which an arbitrary directory and filename can be passed. What would still remain the older way is the generation of led-gen.hpp into the current source directorty Change-Id: I352dadd6aa99ef80192dfca3071357917d4593b8 Signed-off-by: Vishwanatha Subbanna --- parse_led.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/parse_led.py b/parse_led.py index c86764c..8ecfb94 100755 --- a/parse_led.py +++ b/parse_led.py @@ -1,10 +1,26 @@ #!/usr/bin/env python import yaml import os +import argparse if __name__ == '__main__': script_dir = os.path.dirname(os.path.realpath(__file__)) - with open(os.path.join(script_dir, 'led.yaml'), 'r') as f: + parser = argparse.ArgumentParser() + parser.add_argument("-f","--filename", default='led.yaml', help="Input File Name") + parser.add_argument("-d","--directory", default=script_dir, help="Input directory") + args = parser.parse_args() + + # Default to the one that is in the current. + yaml_dir = script_dir; + yaml_file = os.path.join(yaml_dir, 'led.yaml') + + if args.directory: + yaml_dir = args.directory + + if args.filename: + yaml_file = os.path.join(yaml_dir, args.filename) + + with open(yaml_file, 'r') as f: ifile = yaml.safe_load(f) with open(os.path.join(script_dir, 'led-gen.hpp'), 'w') as ofile: -- cgit v1.2.1