Brad Bishop | de32f57 | 2016-11-10 16:23:15 -0500 | [diff] [blame] | 1 | #!/usr/bin/env python |
Patrick Williams | a90b01e | 2016-10-09 16:19:52 -0500 | [diff] [blame] | 2 | import sdbusplus |
| 3 | import mako.lookup |
| 4 | import argparse |
Patrick Williams | affeafa | 2016-10-17 11:40:13 -0500 | [diff] [blame] | 5 | import sys |
Brad Bishop | 3390687 | 2016-11-10 09:29:09 -0500 | [diff] [blame] | 6 | import os |
Patrick Williams | a90b01e | 2016-10-09 16:19:52 -0500 | [diff] [blame] | 7 | |
Patrick Williams | f5c3cde | 2016-10-21 11:49:01 -0500 | [diff] [blame] | 8 | |
Patrick Williams | a90b01e | 2016-10-09 16:19:52 -0500 | [diff] [blame] | 9 | def main(): |
Brad Bishop | ccacc4a | 2016-11-10 17:10:46 -0500 | [diff] [blame^] | 10 | module_path = os.path.dirname(sdbusplus.__file__) |
| 11 | |
Patrick Williams | f5c3cde | 2016-10-21 11:49:01 -0500 | [diff] [blame] | 12 | valid_types = {'interface': sdbusplus.Interface} |
| 13 | valid_processes = {'markdown': "markdown", |
| 14 | 'server-header': "server_header", |
| 15 | 'server-cpp': "server_cpp"} |
Patrick Williams | a90b01e | 2016-10-09 16:19:52 -0500 | [diff] [blame] | 16 | |
| 17 | parser = argparse.ArgumentParser(description='Process sdbus++ YAML files.') |
| 18 | |
| 19 | parser.add_argument('-r', '--rootdir', dest='rootdir', default='example', |
| 20 | type=str, help='Location of files to process.') |
| 21 | parser.add_argument('-t', '--templatedir', dest='templatedir', |
Brad Bishop | ccacc4a | 2016-11-10 17:10:46 -0500 | [diff] [blame^] | 22 | default=os.path.join(module_path, 'templates'), |
Brad Bishop | 3390687 | 2016-11-10 09:29:09 -0500 | [diff] [blame] | 23 | type=str, help='Location of templates files.') |
Patrick Williams | a90b01e | 2016-10-09 16:19:52 -0500 | [diff] [blame] | 24 | parser.add_argument('typeName', metavar='TYPE', type=str, |
| 25 | choices=valid_types.keys(), help='Type to operate on.') |
| 26 | parser.add_argument('process', metavar='PROCESS', type=str, |
| 27 | choices=valid_processes.keys(), |
| 28 | help='Process to apply.') |
| 29 | parser.add_argument('item', metavar='ITEM', type=str, |
| 30 | help='Item to process.') |
| 31 | |
Patrick Williams | f5c3cde | 2016-10-21 11:49:01 -0500 | [diff] [blame] | 32 | args = parser.parse_args() |
Patrick Williams | a90b01e | 2016-10-09 16:19:52 -0500 | [diff] [blame] | 33 | |
Patrick Williams | f5c3cde | 2016-10-21 11:49:01 -0500 | [diff] [blame] | 34 | if sys.version_info < (3, 0): |
Patrick Williams | affeafa | 2016-10-17 11:40:13 -0500 | [diff] [blame] | 35 | lookup = mako.lookup.TemplateLookup(directories=[args.templatedir], |
| 36 | disable_unicode=True) |
| 37 | else: |
| 38 | lookup = mako.lookup.TemplateLookup(directories=[args.templatedir]) |
Patrick Williams | a90b01e | 2016-10-09 16:19:52 -0500 | [diff] [blame] | 39 | |
| 40 | instance = valid_types[args.typeName].load(args.item, args.rootdir) |
Patrick Williams | 24e8ae6 | 2016-10-14 18:24:38 -0500 | [diff] [blame] | 41 | function = getattr(instance, valid_processes[args.process]) |
Patrick Williams | a90b01e | 2016-10-09 16:19:52 -0500 | [diff] [blame] | 42 | print(function(lookup)) |
| 43 | |
| 44 | if __name__ == '__main__': |
| 45 | main() |