mirror of https://github.com/tailix/libkernaux.git
89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
from jinja2 import Environment, FileSystemLoader
|
|
from os import path
|
|
from sys import argv
|
|
from yaml import safe_load
|
|
|
|
def main(test_filepath, template_filepath, cases_reg_filepath, cases_orig_filepath):
|
|
cases_reg = safe_load(open(cases_reg_filepath))
|
|
cases_orig = safe_load(open(cases_orig_filepath))
|
|
|
|
cases = cases_reg + cases_orig
|
|
|
|
jinja_env = Environment(
|
|
keep_trailing_newline=True,
|
|
loader=FileSystemLoader(path.dirname(template_filepath)),
|
|
)
|
|
jinja_template = jinja_env.get_template(path.basename(template_filepath))
|
|
|
|
result = jinja_template.render(
|
|
cases=cases,
|
|
escape_str=escape_str,
|
|
fmt=fmt,
|
|
values=values,
|
|
)
|
|
|
|
with open(test_filepath, 'w') as f:
|
|
f.write(result)
|
|
|
|
def escape_char(c):
|
|
return "'" + c + "'"
|
|
|
|
def escape_str(s):
|
|
return '"' + s + '"'
|
|
|
|
def fmt(args):
|
|
fmt = ''
|
|
for arg in args:
|
|
if type(arg) is list:
|
|
fmt += arg[0]
|
|
else:
|
|
fmt += arg
|
|
return fmt
|
|
|
|
def values(args):
|
|
values = ''
|
|
|
|
for arg in args:
|
|
if type(arg) is list and len(arg) >= 2:
|
|
if len(arg) >= 3:
|
|
values += ', ' + str(arg[1])
|
|
arg[1] = arg[2]
|
|
|
|
if type(arg[1]) is str:
|
|
values += ', ' + escape_str(arg[1])
|
|
elif type(arg[1]) is list:
|
|
if len(arg[1]) == 1:
|
|
values += ', ' + escape_char(arg[1][0])
|
|
elif arg[1][0] == 'long long':
|
|
values += ', (long long)' + str(arg[1][1])
|
|
else:
|
|
raise RuntimeError('unknown format: ' + str(args))
|
|
else:
|
|
values += ', ' + str(arg[1])
|
|
|
|
return values
|
|
|
|
if __name__ == '__main__':
|
|
print(argv)
|
|
|
|
template_filepath = argv[1]
|
|
cases_reg_filepath = argv[2]
|
|
cases_orig_filepath = argv[3]
|
|
test_filepath = argv[4]
|
|
|
|
print('test_filepath: %s' % test_filepath)
|
|
print('template_filepath: %s' % template_filepath)
|
|
print('cases_reg_filepath: %s' % cases_reg_filepath)
|
|
print('cases_orig_filepath: %s' % cases_orig_filepath)
|
|
|
|
if path.exists(test_filepath) and not path.isfile(test_filepath):
|
|
raise RuntimeError('invalid test file path')
|
|
if not path.isfile(template_filepath):
|
|
raise RuntimeError('invalid template file path')
|
|
if not path.isfile(cases_reg_filepath):
|
|
raise RuntimeError('invalid regular cases file path')
|
|
if not path.isfile(cases_orig_filepath):
|
|
raise RuntimeError('invalid original cases file path')
|
|
|
|
main(test_filepath, template_filepath, cases_reg_filepath, cases_orig_filepath)
|