1*de32f577SBrad 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(): 10a90b01e2SPatrick Williams valid_types = {'interface': sdbusplus.Interface} 1104e007f1SPatrick Williams valid_processes = {'markdown': "markdown", 12831839adSPatrick Williams 'server-header': "server_header", 13831839adSPatrick Williams 'server-cpp': "server_cpp"} 14a90b01e2SPatrick Williams 15a90b01e2SPatrick Williams parser = argparse.ArgumentParser(description='Process sdbus++ YAML files.') 16a90b01e2SPatrick Williams 17a90b01e2SPatrick Williams parser.add_argument('-r', '--rootdir', dest='rootdir', default='example', 18a90b01e2SPatrick Williams type=str, help='Location of files to process.') 19a90b01e2SPatrick Williams parser.add_argument('-t', '--templatedir', dest='templatedir', 2033906871SBrad Bishop default=os.path.join('sdbusplus', 'templates'), 2133906871SBrad Bishop type=str, help='Location of templates files.') 22a90b01e2SPatrick Williams parser.add_argument('typeName', metavar='TYPE', type=str, 23a90b01e2SPatrick Williams choices=valid_types.keys(), help='Type to operate on.') 24a90b01e2SPatrick Williams parser.add_argument('process', metavar='PROCESS', type=str, 25a90b01e2SPatrick Williams choices=valid_processes.keys(), 26a90b01e2SPatrick Williams help='Process to apply.') 27a90b01e2SPatrick Williams parser.add_argument('item', metavar='ITEM', type=str, 28a90b01e2SPatrick Williams help='Item to process.') 29a90b01e2SPatrick Williams 30f5c3cde9SPatrick Williams args = parser.parse_args() 31a90b01e2SPatrick Williams 32affeafadSPatrick Williams if sys.version_info < (3, 0): 33affeafadSPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir], 34affeafadSPatrick Williams disable_unicode=True) 35affeafadSPatrick Williams else: 36a90b01e2SPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir]) 37a90b01e2SPatrick Williams 38a90b01e2SPatrick Williams instance = valid_types[args.typeName].load(args.item, args.rootdir) 3924e8ae6eSPatrick Williams function = getattr(instance, valid_processes[args.process]) 40a90b01e2SPatrick Williams print(function(lookup)) 41a90b01e2SPatrick Williams 42a90b01e2SPatrick Williamsif __name__ == '__main__': 43a90b01e2SPatrick Williams main() 44