1de32f577SBrad Bishop#!/usr/bin/env python 2a90b01e2SPatrick Williamsimport sdbusplus 3a90b01e2SPatrick Williamsimport mako.lookup 4a90b01e2SPatrick Williamsimport argparse 5affeafadSPatrick Williamsimport sys 633906871SBrad Bishopimport os 7a90b01e2SPatrick Williams 8f5c3cde9SPatrick Williams 9a90b01e2SPatrick Williamsdef main(): 10ccacc4afSBrad Bishop module_path = os.path.dirname(sdbusplus.__file__) 11ccacc4afSBrad Bishop 127d7fda56SPatrick Williams valid_types = {'interface': sdbusplus.Interface, 137d7fda56SPatrick Williams 'error': sdbusplus.Error} 1404e007f1SPatrick Williams valid_processes = {'markdown': "markdown", 15831839adSPatrick Williams 'server-header': "server_header", 167d7fda56SPatrick Williams 'server-cpp': "server_cpp", 17*250654e8SPatrick Williams 'exception-header': "exception_header", 18*250654e8SPatrick Williams 'exception-cpp': "exception_cpp"} 19a90b01e2SPatrick Williams 20a90b01e2SPatrick Williams parser = argparse.ArgumentParser(description='Process sdbus++ YAML files.') 21a90b01e2SPatrick Williams 22a90b01e2SPatrick Williams parser.add_argument('-r', '--rootdir', dest='rootdir', default='example', 23a90b01e2SPatrick Williams type=str, help='Location of files to process.') 24a90b01e2SPatrick Williams parser.add_argument('-t', '--templatedir', dest='templatedir', 25ccacc4afSBrad Bishop default=os.path.join(module_path, 'templates'), 2633906871SBrad Bishop type=str, help='Location of templates files.') 27a90b01e2SPatrick Williams parser.add_argument('typeName', metavar='TYPE', type=str, 28a90b01e2SPatrick Williams choices=valid_types.keys(), help='Type to operate on.') 29a90b01e2SPatrick Williams parser.add_argument('process', metavar='PROCESS', type=str, 30a90b01e2SPatrick Williams choices=valid_processes.keys(), 31a90b01e2SPatrick Williams help='Process to apply.') 32a90b01e2SPatrick Williams parser.add_argument('item', metavar='ITEM', type=str, 33a90b01e2SPatrick Williams help='Item to process.') 34a90b01e2SPatrick Williams 35f5c3cde9SPatrick Williams args = parser.parse_args() 36a90b01e2SPatrick Williams 37affeafadSPatrick Williams if sys.version_info < (3, 0): 38affeafadSPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir], 39affeafadSPatrick Williams disable_unicode=True) 40affeafadSPatrick Williams else: 41a90b01e2SPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir]) 42a90b01e2SPatrick Williams 43a90b01e2SPatrick Williams instance = valid_types[args.typeName].load(args.item, args.rootdir) 4424e8ae6eSPatrick Williams function = getattr(instance, valid_processes[args.process]) 45a90b01e2SPatrick Williams print(function(lookup)) 46a90b01e2SPatrick Williams 47a90b01e2SPatrick Williamsif __name__ == '__main__': 48a90b01e2SPatrick Williams main() 49