|
3 | 3 | import jinja2 |
4 | 4 |
|
5 | 5 |
|
6 | | -loader = jinja2.FileSystemLoader(searchpath=Path(__file__).parent / 'templates') |
| 6 | +loader = jinja2.FileSystemLoader(searchpath=Path(__file__).parent / "templates") |
7 | 7 |
|
8 | 8 | environment = jinja2.Environment( |
9 | 9 | loader=loader, |
10 | 10 | trim_blocks=True, |
11 | | - block_start_string='@@', |
12 | | - block_end_string='@@', |
13 | | - variable_start_string='@=', |
14 | | - variable_end_string='=@' |
| 11 | + block_start_string="@@", |
| 12 | + block_end_string="@@", |
| 13 | + variable_start_string="@=", |
| 14 | + variable_end_string="=@", |
15 | 15 | ) |
16 | 16 |
|
17 | | -library_dir = Path(__file__).parent.parent.parent.parent / 'FMI' |
18 | | - |
19 | | -for variable_type, prefix in product(['Real', 'Integer', 'Boolean', 'String'], ['Get', 'Set']): |
| 17 | +library_dir = Path(__file__).parent.parent.parent.parent / "FMI" |
20 | 18 |
|
| 19 | +for variable_type, prefix in product( |
| 20 | + ["Real", "Integer", "Boolean", "String"], ["Get", "Set"] |
| 21 | +): |
21 | 22 | for package, level in [ |
22 | | - (library_dir / 'FMI2' / 'Functions', ''), |
23 | | - (library_dir / 'Internal' / 'FMI2', 'Internal') |
| 23 | + (library_dir / "FMI2" / "Functions", ""), |
| 24 | + (library_dir / "Internal" / "FMI2", "Internal"), |
24 | 25 | ]: |
| 26 | + template = environment.get_template(f"FMI2_{level}{prefix}.mo") |
25 | 27 |
|
26 | | - template = environment.get_template(f'FMI2_{level}{prefix}.mo') |
27 | | - |
28 | | - class_text = template.render( |
29 | | - variable_type=variable_type |
30 | | - ) |
| 28 | + class_text = template.render(variable_type=variable_type) |
31 | 29 |
|
32 | | - function_name = f'FMI2{prefix}{variable_type}' |
| 30 | + function_name = f"FMI2{prefix}{variable_type}" |
33 | 31 |
|
34 | | - with open(package / f'{function_name}.mo', 'w') as f: |
| 32 | + with open(package / f"{function_name}.mo", "w") as f: |
35 | 33 | f.write(class_text) |
36 | 34 |
|
37 | | - package_order_file = package / 'package.order' |
| 35 | + package_order_file = package / "package.order" |
38 | 36 |
|
39 | | - with open(package_order_file, 'r') as f: |
| 37 | + with open(package_order_file, "r") as f: |
40 | 38 | package_order = list(map(lambda line: line.strip(), f.readlines())) |
41 | 39 |
|
42 | 40 | if function_name not in package_order: |
43 | | - with open(package_order_file, 'a') as f: |
44 | | - f.write(function_name + '\n') |
| 41 | + with open(package_order_file, "a") as f: |
| 42 | + f.write(function_name + "\n") |
45 | 43 |
|
46 | 44 | for variable_type, (prefix, suffix) in product( |
47 | | - ['Float32', 'Float64', 'Int8', 'UInt8', 'Int16', 'UInt16', 'Int32', 'UInt32', 'Int64', 'UInt64', 'Boolean', 'String'], |
48 | | - [('Get', ''), ('Set', ''), ('Get', 'Matrix'), ('Set', 'Matrix')] |
| 45 | + [ |
| 46 | + "Float32", |
| 47 | + "Float64", |
| 48 | + "Int8", |
| 49 | + "UInt8", |
| 50 | + "Int16", |
| 51 | + "UInt16", |
| 52 | + "Int32", |
| 53 | + "UInt32", |
| 54 | + "Int64", |
| 55 | + "UInt64", |
| 56 | + "Boolean", |
| 57 | + "String", |
| 58 | + ], |
| 59 | + [("Get", ""), ("Set", ""), ("Get", "Matrix"), ("Set", "Matrix")], |
49 | 60 | ): |
50 | | - |
51 | 61 | for package, level in [ |
52 | | - (library_dir / 'FMI3' / 'Functions', ''), |
53 | | - (library_dir / 'Internal' / 'FMI3', 'Internal') |
| 62 | + (library_dir / "FMI3" / "Functions", ""), |
| 63 | + (library_dir / "Internal" / "FMI3", "Internal"), |
54 | 64 | ]: |
| 65 | + template = environment.get_template(f"FMI3_{level}{prefix}{suffix}.mo") |
55 | 66 |
|
56 | | - template = environment.get_template(f'FMI3_{level}{prefix}{suffix}.mo') |
57 | | - |
58 | | - class_text = template.render( |
59 | | - variable_type=variable_type |
60 | | - ) |
| 67 | + class_text = template.render(variable_type=variable_type) |
61 | 68 |
|
62 | | - function_name = f'FMI3{prefix}{variable_type}{suffix}' |
| 69 | + function_name = f"FMI3{prefix}{variable_type}{suffix}" |
63 | 70 |
|
64 | | - with open(package / f'{function_name}.mo', 'w') as f: |
| 71 | + with open(package / f"{function_name}.mo", "w") as f: |
65 | 72 | f.write(class_text) |
66 | 73 |
|
67 | | - package_order_file = package / 'package.order' |
| 74 | + package_order_file = package / "package.order" |
68 | 75 |
|
69 | | - with open(package_order_file, 'r') as f: |
| 76 | + with open(package_order_file, "r") as f: |
70 | 77 | package_order = list(map(lambda line: line.strip(), f.readlines())) |
71 | 78 |
|
72 | 79 | if function_name not in package_order: |
73 | | - with open(package_order_file, 'a') as f: |
74 | | - f.write(function_name + '\n') |
| 80 | + with open(package_order_file, "a") as f: |
| 81 | + f.write(function_name + "\n") |
0 commit comments