64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import json
|
||
|
|
import sys
|
||
|
|
|
||
|
|
def dump_json(obj, out, pretty):
|
||
|
|
if pretty:
|
||
|
|
json.dump(obj, out, indent=2, sort_keys=True)
|
||
|
|
else:
|
||
|
|
json.dump(obj, out, separators=(",", ":"), sort_keys=True)
|
||
|
|
return
|
||
|
|
|
||
|
|
def main():
|
||
|
|
parser = argparse.ArgumentParser(description="Set is_prefixed=false for all asset owners")
|
||
|
|
parser.add_argument("-o", "--output", metavar="OUT", default="-", help="output filename (default: stdout)")
|
||
|
|
parser.add_argument("-i", "--input", metavar="IN", default="-", help="input filename (default: stdin)")
|
||
|
|
parser.add_argument("-f", "--from", metavar="PREFIX", default="", help="initial prefix")
|
||
|
|
parser.add_argument("-t", "--to", metavar="PREFIX", default="", help="new prefix")
|
||
|
|
parser.add_argument("-p", "--pretty", action="store_true", default=False, help="pretty print output")
|
||
|
|
opts = parser.parse_args()
|
||
|
|
|
||
|
|
if opts.input == "-":
|
||
|
|
genesis = json.load(sys.stdin)
|
||
|
|
else:
|
||
|
|
with open(opts.input, "r") as f:
|
||
|
|
genesis = json.load(f)
|
||
|
|
|
||
|
|
frum = opts.__dict__["from"] # from is a language keyword and cannot be an attribute name
|
||
|
|
|
||
|
|
def convert(k):
|
||
|
|
if not k.startswith(frum):
|
||
|
|
print("key {k} does not start with prefix {p}".format(k=repr(k), p=repr(frum)))
|
||
|
|
raise RuntimeError()
|
||
|
|
return opts.to + k[len(frum):]
|
||
|
|
|
||
|
|
for account in genesis["initial_accounts"]:
|
||
|
|
account["owner_key"] = convert(account["owner_key"])
|
||
|
|
account["active_key"] = convert(account["active_key"])
|
||
|
|
|
||
|
|
for asset in genesis["initial_assets"]:
|
||
|
|
for cr in asset.get("collateral_records", []):
|
||
|
|
cr["owner"] = convert(cr["owner"])
|
||
|
|
|
||
|
|
for balance in genesis["initial_balances"]:
|
||
|
|
balance["owner"] = convert(balance["owner"])
|
||
|
|
|
||
|
|
for vb in genesis["initial_vesting_balances"]:
|
||
|
|
vb["owner"] = convert(vb["owner"])
|
||
|
|
|
||
|
|
for witness in genesis["initial_witness_candidates"]:
|
||
|
|
witness["block_signing_key"] = convert(witness["block_signing_key"])
|
||
|
|
|
||
|
|
if opts.output == "-":
|
||
|
|
dump_json( genesis, sys.stdout, opts.pretty )
|
||
|
|
sys.stdout.flush()
|
||
|
|
else:
|
||
|
|
with open(opts.output, "w") as f:
|
||
|
|
dump_json( genesis, f, opts.pretty )
|
||
|
|
return
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|