queryfeats.py 661 B

123456789101112131415161718192021222324252627282930
  1. #!/usr/bin/env python3
  2. import pickle
  3. import pprint
  4. with open('feats.pickle', 'rb') as f:
  5. allfeats = pickle.load(f)
  6. prefeats = {}
  7. for f in allfeats:
  8. if 'feat' in f['prereqs']:
  9. fs = f['prereqs']['feat']
  10. if isinstance(fs, list):
  11. for pf in fs:
  12. if pf in prefeats:
  13. prefeats[pf] += 1
  14. else:
  15. prefeats[pf] = 1
  16. else:
  17. if fs in prefeats:
  18. prefeats[fs] += 1
  19. else:
  20. prefeats[fs] = 1
  21. outfeats = sorted(prefeats, key=lambda x: prefeats[x])
  22. for of in outfeats:
  23. print("%s, %s" % (of, prefeats[of]))