mirror of
https://github.com/quatalog/quatalog.git
synced 2024-11-15 19:42:44 +00:00
30 lines
835 B
Python
30 lines
835 B
Python
import json
|
|
import sys
|
|
import collections
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) != 3:
|
|
print(
|
|
f"USAGE: python {sys.argv[0]} <json from scraper> <by-course output file>",
|
|
file=sys.stderr,
|
|
)
|
|
return 1
|
|
with open(sys.argv[1], "r") as scraper_json:
|
|
by_institution = json.load(scraper_json)
|
|
|
|
by_rpi_course = collections.defaultdict(list)
|
|
for inst in by_institution:
|
|
for xfer in inst["transfers"]:
|
|
for rpi_course in xfer["rpi"]["courses"]:
|
|
for a in ["institution", "city", "state"]:
|
|
xfer[a] = inst[a]
|
|
by_rpi_course[rpi_course["id"]].append(xfer)
|
|
|
|
with open(sys.argv[2], "w") as out_json:
|
|
json.dump(by_rpi_course, out_json, sort_keys=True, indent=2)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
exit(main())
|