fetchcargo: add type checking to cargo-vendor-normalise.py

This commit is contained in:
Jörg Thalheim 2018-09-11 21:49:16 +01:00 committed by Symphorien Gibol
parent f20b229aa1
commit 7bfa20198a

View File

@ -1,32 +1,41 @@
#!/usr/bin/env python #!/usr/bin/env python
import toml
import sys import sys
def escape(s): import toml
return '"'+s.replace('"', r'\"').replace("\n", r"\n").replace("\\", "\\\\")+'"'
data = toml.load(sys.stdin)
assert list(data.keys()) == [ "source" ] def quote(s: str) -> str:
escaped = s.replace('"', r"\"").replace("\n", r"\n").replace("\\", "\\\\")
return '"{}"'.format(escaped)
# this value is non deterministic
data["source"]["vendored-sources"]["directory"] = "@vendor@"
result = "" def main() -> None:
inner = data["source"] data = toml.load(sys.stdin)
for source in sorted(inner.keys()):
result += '[source.{}]\n'.format(escape(source)) assert list(data.keys()) == ["source"]
# this value is non deterministic
data["source"]["vendored-sources"]["directory"] = "@vendor@"
lines = []
inner = data["source"]
for source, attrs in sorted(inner.items()):
lines.append("[source.{}]".format(quote(source)))
if source == "vendored-sources": if source == "vendored-sources":
result += '"directory" = "@vendor@"\n' lines.append('"directory" = "@vendor@"\n')
else: else:
for key in sorted(inner[source].keys()): for key, value in sorted(attrs.items()):
result += '{} = {}\n'.format(escape(key), escape(inner[source][key])) attr = "{} = {}".format(quote(key), quote(value))
result += "\n" lines.append(attr)
lines.append("")
real = toml.loads(result) result = "\n".join(lines)
assert real == data, "output = {} while input = {}".format(real, data) real = toml.loads(result)
assert real == data, "output = {} while input = {}".format(real, data)
print(result) print(result)
if __name__ == "__main__":
main()