Skip to content

Instantly share code, notes, and snippets.

@aDotInTheVoid
Created December 5, 2020 22:27
Show Gist options
  • Save aDotInTheVoid/2dfce0d241338def3f033f941b7c183d to your computer and use it in GitHub Desktop.
Save aDotInTheVoid/2dfce0d241338def3f033f941b7c183d to your computer and use it in GitHub Desktop.
#!/usr/bin/env python3
# python3 bless_local.py nested.json nested.expected
# Where nested.json is raw rustdoc output
# Your'll need to add $TEST_BASE_DIR manualy for now
import json
import sys
def keep(k, v) -> bool:
# Ensure it's a local (not std and frends), and also not an impl.
return k.split(":")[0] == "0" and v["kind"] != "impl"
with open(sys.argv[1]) as f:
raw = json.load(f)
raw["index"] = {k: v for (k, v) in raw["index"].items() if keep(k, v)}
# Beautifull magestic glorious hack
idxstr = repr(raw["index"])
raw["paths"] = {k: v for (k, v) in raw["paths"].items() if k in idxstr}
crates_needed = {i["crate_id"] for i in raw["paths"].values()}
raw["external_crates"] = {k: v for (k, v) in raw["external_crates"].items() if k in crates_needed}
with open(sys.argv[2], 'w') as f:
json.dump(raw, f, indent=2, sort_keys=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment