1*18ce803dSPatrick Williams#!/usr/bin/env python3 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", 17250654e8SPatrick Williams 'exception-header': "exception_header", 18ddc0eba2SLei YU 'exception-cpp': "exception_cpp", 19ddc0eba2SLei YU 'client-header': "client_header"} 20a90b01e2SPatrick Williams 21a90b01e2SPatrick Williams parser = argparse.ArgumentParser(description='Process sdbus++ YAML files.') 22a90b01e2SPatrick Williams 237bfb70acSPatrick Williams parser.add_argument('-r', '--rootdir', dest='rootdir', default='.', 24a90b01e2SPatrick Williams type=str, help='Location of files to process.') 25a90b01e2SPatrick Williams parser.add_argument('-t', '--templatedir', dest='templatedir', 26ccacc4afSBrad Bishop default=os.path.join(module_path, 'templates'), 2733906871SBrad Bishop type=str, help='Location of templates files.') 28a90b01e2SPatrick Williams parser.add_argument('typeName', metavar='TYPE', type=str, 29a90b01e2SPatrick Williams choices=valid_types.keys(), help='Type to operate on.') 30a90b01e2SPatrick Williams parser.add_argument('process', metavar='PROCESS', type=str, 31a90b01e2SPatrick Williams choices=valid_processes.keys(), 32a90b01e2SPatrick Williams help='Process to apply.') 33a90b01e2SPatrick Williams parser.add_argument('item', metavar='ITEM', type=str, 34a90b01e2SPatrick Williams help='Item to process.') 35a90b01e2SPatrick Williams 36f5c3cde9SPatrick Williams args = parser.parse_args() 37a90b01e2SPatrick Williams 38a90b01e2SPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir]) 39a90b01e2SPatrick Williams 40a90b01e2SPatrick Williams instance = valid_types[args.typeName].load(args.item, args.rootdir) 4124e8ae6eSPatrick Williams function = getattr(instance, valid_processes[args.process]) 42a90b01e2SPatrick Williams print(function(lookup)) 43a90b01e2SPatrick Williams 44a90b01e2SPatrick Williamsif __name__ == '__main__': 45a90b01e2SPatrick Williams main() 46