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