/
githubmirror
/
nbs
Обзор
Документация
Войти
/
githubmirror
/
nbs
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
build/scripts/gen_tasklet_reg.py
48 строк
1 KB
iaz1607
`build/scripts` ya style --py
30 ноя 2023, 13:12
30 ноя 2023, 13:12
a6ee229
Код
Авторство
О чём код?
import argparse TEMPLATE = '''\ {includes}\ #include <tasklet/v1/runtime/lib/{language}_wrapper.h> #include <tasklet/v1/runtime/lib/registry.h> static const NTasklet::TRegHelper REG( "{name}", new NTasklet::{wrapper} ); ''' WRAPPER = { 'cpp': 'TCppWrapper<{impl}>()', 'js': 'TJsWrapper("{impl}")', 'go': 'TGoWrapper("{impl}")', 'py': 'TPythonWrapper("{impl}")', 'java': 'TJavaWrapper("{impl}", "{py_wrapper}")', } def parse_args(): parser = argparse.ArgumentParser() parser.add_argument('name') parser.add_argument('output') parser.add_argument('-l', '--lang', choices=WRAPPER, required=True) parser.add_argument('-i', '--impl', required=True) parser.add_argument('-w', '--wrapper', required=False) parser.add_argument('includes', nargs='*') return parser.parse_args() if __name__ == '__main__': args = parse_args() includes = ''.join('#include <{}>\n'.format(include) for include in args.includes) code = TEMPLATE.format( includes=includes, language=args.lang, name=args.name, wrapper=WRAPPER[args.lang].format(impl=args.impl, py_wrapper=args.wrapper), ) with open(args.output, 'w') as f: f.write(code)