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 <vishwa@linux.vnet.ibm.com>
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: