mirror of
https://github.com/nikdoof/dropbot.git
synced 2025-12-17 19:59:24 +00:00
Add support for nearest offices.
This commit is contained in:
49
gen_reference_data.py
Normal file
49
gen_reference_data.py
Normal file
@@ -0,0 +1,49 @@
|
||||
import sys
|
||||
import sqlite3
|
||||
from json import dumps
|
||||
|
||||
def main():
|
||||
if len(sys.argv) > 1:
|
||||
dbfile = sys.argv[1]
|
||||
try:
|
||||
conn = sqlite3.connect(dbfile)
|
||||
conn.text_factory = str
|
||||
except:
|
||||
print("Unable to open the SDE file at %s\n" % dbfile)
|
||||
else:
|
||||
sys.stderr.write("Usage: {} <db file>\n".format(sys.argv[0]))
|
||||
return 1
|
||||
|
||||
sys.stderr.write("Importing Types...\n")
|
||||
data = {}
|
||||
for row in conn.execute("""SELECT typeID, typeName FROM invTypes"""):
|
||||
pk, name = row
|
||||
try:
|
||||
x = name.decode('utf8')
|
||||
except:
|
||||
continue
|
||||
data[long(pk)] = name
|
||||
|
||||
with open('types.json', 'wb') as f:
|
||||
f.write(dumps(data))
|
||||
|
||||
sys.stderr.write("Importing Stations...\n")
|
||||
data = {}
|
||||
for row in conn.execute("""SELECT stationID, solarSystemID FROM staStations"""):
|
||||
pk, val = row
|
||||
data[long(pk)] = long(val)
|
||||
|
||||
with open('stations.json', 'wb') as f:
|
||||
f.write(dumps(data))
|
||||
|
||||
sys.stderr.write("Importing Map...\n")
|
||||
|
||||
from dropbot.map import Map
|
||||
map = Map()
|
||||
map.from_sde(conn)
|
||||
|
||||
with open('map.json', 'wb') as f:
|
||||
f.write(map.to_json())
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Reference in New Issue
Block a user