mirror of
https://github.com/jlu5/SupyPlugins.git
synced 2025-04-26 04:51:08 -05:00
26 lines
826 B
Python
Executable File
26 lines
826 B
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Migrates user locations from the Weather plugin to NuWeather.
|
|
"""
|
|
|
|
import argparse
|
|
import sqlite3
|
|
import pickle
|
|
|
|
parser = argparse.ArgumentParser(prog='weather-migrate', description=__doc__)
|
|
parser.add_argument('infile', type=str, help='input filename (BOT_DATA_DIR/Weather.db)')
|
|
parser.add_argument('outfile', type=str, help='output filename (e.g. BOT_DATA_DIR/NuWeather.db)')
|
|
args = parser.parse_args()
|
|
|
|
conn = sqlite3.connect(args.infile)
|
|
c = conn.cursor()
|
|
# We only care about nick and location
|
|
c.execute('SELECT nick, location from users')
|
|
out = dict(c.fetchall())
|
|
print("OK read %d entries from %s" % (len(out), args.infile))
|
|
c.close() # No need to commit since we only read
|
|
|
|
with open(args.outfile, 'wb') as outf:
|
|
pickle.dump(out, outf)
|
|
print("OK wrote output to %s" % args.outfile)
|