1a90b01e2SPatrick Williams#!/bin/env python 2a90b01e2SPatrick Williamsimport sdbusplus 3a90b01e2SPatrick Williamsimport mako.lookup 4a90b01e2SPatrick Williamsimport argparse 5*affeafadSPatrick Williamsimport sys 6a90b01e2SPatrick Williams 7a90b01e2SPatrick Williamsdef main(): 8a90b01e2SPatrick Williams valid_types = { 'interface': sdbusplus.Interface } 904e007f1SPatrick Williams valid_processes = { 'markdown' : "markdown", 10831839adSPatrick Williams 'server-header' : "server_header", 11831839adSPatrick Williams 'server-cpp' : "server_cpp" } 12a90b01e2SPatrick Williams 13a90b01e2SPatrick Williams parser = argparse.ArgumentParser(description='Process sdbus++ YAML files.') 14a90b01e2SPatrick Williams 15a90b01e2SPatrick Williams parser.add_argument('-r', '--rootdir', dest='rootdir', default='example', 16a90b01e2SPatrick Williams type=str, help='Location of files to process.') 17a90b01e2SPatrick Williams parser.add_argument('-t', '--templatedir', dest='templatedir', 18a90b01e2SPatrick Williams default='templates', type=str, 19a90b01e2SPatrick Williams help='Location of templates files.') 20a90b01e2SPatrick Williams parser.add_argument('typeName', metavar='TYPE', type=str, 21a90b01e2SPatrick Williams choices=valid_types.keys(), help='Type to operate on.') 22a90b01e2SPatrick Williams parser.add_argument('process', metavar='PROCESS', type=str, 23a90b01e2SPatrick Williams choices=valid_processes.keys(), 24a90b01e2SPatrick Williams help='Process to apply.') 25a90b01e2SPatrick Williams parser.add_argument('item', metavar='ITEM', type=str, 26a90b01e2SPatrick Williams help='Item to process.') 27a90b01e2SPatrick Williams 28a90b01e2SPatrick Williams args = parser.parse_args(); 29a90b01e2SPatrick Williams 30*affeafadSPatrick Williams if sys.version_info < (3,0): 31*affeafadSPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir], 32*affeafadSPatrick Williams disable_unicode=True) 33*affeafadSPatrick Williams else: 34a90b01e2SPatrick Williams lookup = mako.lookup.TemplateLookup(directories=[args.templatedir]) 35a90b01e2SPatrick Williams 36a90b01e2SPatrick Williams instance = valid_types[args.typeName].load(args.item, args.rootdir) 3724e8ae6eSPatrick Williams function = getattr(instance, valid_processes[args.process]) 38a90b01e2SPatrick Williams print(function(lookup)) 39a90b01e2SPatrick Williams 40a90b01e2SPatrick Williamsif __name__ == '__main__': 41a90b01e2SPatrick Williams main() 42