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 12*7d7fda56SPatrick Williams valid_types = {'interface': sdbusplus.Interface, 13*7d7fda56SPatrick Williams 'error': sdbusplus.Error} 1404e007f1SPatrick Williams valid_processes = {'markdown': "markdown", 15831839adSPatrick Williams 'server-header': "server_header", 16*7d7fda56SPatrick Williams 'server-cpp': "server_cpp", 17*7d7fda56SPatrick Williams 'exception-header': "exception_header"} 18a90b01e2SPatrick Williams 19a90b01e2SPatrick Williams parser = argparse.ArgumentParser(description='Process sdbus++ YAML files.') 20a90b01e2SPatrick Williams 21a90b01e2SPatrick Williams parser.add_argument('-r', '--rootdir', dest='rootdir', default='example', 22a90b01e2SPatrick Williams type=str, help='Location of files to process.') 23a90b01e2SPatrick Williams parser.add_argument('-t', '--templatedir', dest='templatedir', 24ccacc4afSBrad Bishop default=os.path.join(module_path, 'templates'), 2533906871SBrad Bishop type=str, help='Location of templates files.') 26a90b01e2SPatrick Williams parser.add_argument('typeName', metavar='TYPE', type=str, 27a90b01e2SPatrick Williams choices=valid_types.keys(), help='Type to operate on.') 28a90b01e2SPatrick Williams parser.add_argument('process', metavar='PROCESS', type=str, 29a90b01e2SPatrick Williams choices=valid_processes.keys(), 30a90b01e2SPatrick Williams help='Process to apply.') 31a90b01e2SPatrick Williams parser.add_argument('item', metavar='ITEM', type=str, 32a90b01e2SPatrick Williams help='Item to process.') 33a90b01e2SPatrick Williams 34f5c3cde9SPatrick Williams args = parser.parse_args() 35a90b01e2SPatrick Williams 36affeafadSPatrick Williams if sys.version_info < (3, 0): 37affeafadSPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir], 38affeafadSPatrick Williams disable_unicode=True) 39affeafadSPatrick Williams else: 40a90b01e2SPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir]) 41a90b01e2SPatrick Williams 42a90b01e2SPatrick Williams instance = valid_types[args.typeName].load(args.item, args.rootdir) 4324e8ae6eSPatrick Williams function = getattr(instance, valid_processes[args.process]) 44a90b01e2SPatrick Williams print(function(lookup)) 45a90b01e2SPatrick Williams 46a90b01e2SPatrick Williamsif __name__ == '__main__': 47a90b01e2SPatrick Williams main() 48