diff options
-rwxr-xr-x | parse_led.py | 18 |
1 files changed, 17 insertions, 1 deletions
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: |