c-kzg-4844/scripts/convert_trusted_setup.py

45 lines
1.1 KiB
Python
Raw Normal View History

#!/usr/bin/python3
import argparse
import json
from typing import TextIO
def convert(ts_json: TextIO, ts_text: TextIO) -> None:
"""Convert trusted setup to text format."""
trusted_setup = json.load(ts_json)
g1_values = trusted_setup["g1_lagrange"]
g2_values = trusted_setup["g2_monomial"]
print(len(g1_values), file=ts_text)
print(len(g2_values), file=ts_text)
for g1 in g1_values:
print(g1.replace("0x", ""), file=ts_text)
for g2 in g2_values:
print(g2.replace("0x", ""), file=ts_text)
if __name__ == "__main__":
parser = argparse.ArgumentParser(
description="Convert trusted setup from JSON to text format.",
)
parser.add_argument(
"--input",
required=True,
type=argparse.FileType("r"),
help="the trusted setup in JSON format",
)
parser.add_argument(
"--output",
required=True,
type=argparse.FileType("w"),
help="the trusted setup in text format",
)
args = parser.parse_args()
try:
convert(args.input, args.output)
finally:
args.input.close()
args.output.close()