mirror of https://github.com/tailix/libkernaux.git
81 lines
2.1 KiB
Python
81 lines
2.1 KiB
Python
from jinja2 import Environment, FileSystemLoader
|
|
from os import path
|
|
from yaml import SafeLoader, safe_load
|
|
|
|
CASES_ORIG_FILENAME = 'printf_orig.yml'
|
|
CASES_REG_FILENAME = 'printf.yml'
|
|
TEMPLATE_FILENAME = 'printf_gen.jinja'
|
|
TEST_FILENAME = 'test_printf_gen.c'
|
|
|
|
ROOT_DIRPATH = path.dirname(path.dirname(path.join(path.abspath(__file__))))
|
|
|
|
COMMON_DIRPATH = path.join(ROOT_DIRPATH, 'common')
|
|
TESTS_DIRPATH = path.join(ROOT_DIRPATH, 'tests')
|
|
|
|
CASES_ORIG_FILEPATH = path.join(COMMON_DIRPATH, CASES_ORIG_FILENAME)
|
|
CASES_REG_FILEPATH = path.join(COMMON_DIRPATH, CASES_REG_FILENAME)
|
|
TEST_FILEPATH = path.join(TESTS_DIRPATH, TEST_FILENAME)
|
|
|
|
def main():
|
|
cases_orig = safe_load(open(CASES_ORIG_FILEPATH))
|
|
cases_reg = safe_load(open(CASES_REG_FILEPATH))
|
|
|
|
cases = cases_reg + cases_orig
|
|
|
|
jinja_env = Environment(
|
|
keep_trailing_newline=True,
|
|
loader=FileSystemLoader(TESTS_DIRPATH),
|
|
)
|
|
jinja_template = jinja_env.get_template(TEMPLATE_FILENAME)
|
|
|
|
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__':
|
|
main()
|