mirror of https://github.com/tailix/libkernaux.git
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
|
from jinja2 import Environment, FileSystemLoader
|
||
|
|
from os import path
|
||
|
|
from yaml import SafeLoader, safe_load
|
||
|
|
|
||
|
|
CASES_FILENAME = 'cmdline.yml'
|
||
|
|
TEMPLATE_FILENAME = 'cmdline_gen.jinja'
|
||
|
|
TEST_FILENAME = 'test_cmdline_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_FILEPATH = path.join(COMMON_DIRPATH, CASES_FILENAME)
|
||
|
|
TEST_FILEPATH = path.join(TESTS_DIRPATH, TEST_FILENAME)
|
||
|
|
|
||
|
|
def main():
|
||
|
|
cases = safe_load(open(CASES_FILEPATH))
|
||
|
|
|
||
|
|
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,
|
||
|
|
escape_int=escape_int,
|
||
|
|
len=len,
|
||
|
|
)
|
||
|
|
|
||
|
|
with open(TEST_FILEPATH, 'w') as f:
|
||
|
|
f.write(result)
|
||
|
|
|
||
|
|
def escape_int(n):
|
||
|
|
return str(n)
|
||
|
|
|
||
|
|
def escape_str(s):
|
||
|
|
return '"' + s + '"'
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|