forked from smogon/usage-stats
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathPS-Extractor.py
35 lines (25 loc) · 902 Bytes
/
PS-Extractor.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#!/usr/bin/env python
"""The goal of this script is to take a json version of the "exports" object
used on Pokemon Showdown and pull out the data that's needed for the scripts."""
import json
import pickle
from onix import contexts
ctx = contexts.get_standard_context(force_refresh=True)
keyLookup = {}
baseStats = {}
types = {}
for item in ctx.items.values():
keyLookup[item['id']] = item['name']
for key, move in ctx.moves.items():
keyLookup[key] = move['name']
for ability in ctx.abilities.values():
keyLookup[ability['id']] = ability['name']
for k, v in ctx.pokedex.items():
baseStats[k] = v['baseStats']
keyLookup[k] = v['species']
types[k] = v['types']
for k, v in ctx.natures.items():
keyLookup[k] = v['name']
json.dump(baseStats, open('baseStats.json', 'w'))
json.dump(types, open('types.json', 'w'))
pickle.dump(keyLookup, open('keylookup.pickle', 'wb'))