63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
|
import sys
|
|
import difflib
|
|
|
|
from pathlib import Path
|
|
|
|
import jinja2
|
|
|
|
|
|
def render_workflow(scheme_path):
|
|
result = 0
|
|
scheme_name = scheme_path.name
|
|
scheme_type = str(scheme_path.parent.name).split("_")[1]
|
|
target_name = f"{scheme_type}_{scheme_name}.yml"
|
|
|
|
tpl = TPL_ENV.get_template("template.yml.j2")
|
|
new_contents = tpl.render(scheme_type=scheme_type, scheme_name=scheme_name)
|
|
if not TEST_MODE:
|
|
with open(target_name, "w") as f:
|
|
f.write(new_contents)
|
|
with open("BADGES.md", "a") as f:
|
|
f.write(
|
|
f"![Test {scheme_name}]"
|
|
f"(https://github.com/PQClean/PQClean/workflows/Test%20{scheme_name}/badge.svg?branch=master)\n")
|
|
else:
|
|
if (Path(".") / target_name).exists():
|
|
with open(target_name, "r") as f:
|
|
file_contents = f.read()
|
|
else:
|
|
file_contents = ""
|
|
|
|
if file_contents != new_contents:
|
|
sys.stdout.writelines(difflib.unified_diff(
|
|
file_contents.splitlines(keepends=True), new_contents.splitlines(keepends=True),
|
|
fromfile=f"old/{target_name}", tofile=f"new/{target_name}")
|
|
)
|
|
result = 1
|
|
|
|
return result
|
|
|
|
|
|
if __name__ == "__main__":
|
|
TEST_MODE = len(sys.argv) > 1 and sys.argv[1] == "test"
|
|
|
|
loader = jinja2.FileSystemLoader(".")
|
|
TPL_ENV = jinja2.Environment(
|
|
loader=loader, variable_start_string="{-", variable_end_string="-}"
|
|
)
|
|
|
|
if not TEST_MODE:
|
|
with open("BADGES.md", "w") as f:
|
|
f.truncate()
|
|
|
|
with open("template.yml.j2") as f:
|
|
TEMPLATE = f.read()
|
|
root = Path("../..")
|
|
|
|
paths = root.glob("crypto_*/*")
|
|
for path in paths:
|
|
result = render_workflow(path)
|
|
|
|
sys.exit(result)
|