Limnoria/tools/fundbConvert.py

59 lines
1.8 KiB
Python
Raw Normal View History

#!/usr/bin/env python
import os
import sys
import sqlite
import supybot.dbi as dbi
import supybot.plugins.Lart as Lart
import supybot.plugins.Praise as Praise
def usage():
print 'usage: %s <sqlitedb> <channel> [<botname>]' % sys.argv[0]
print '<botname> is an optional parameter used if any db entries are '\
'missing a "created by" user.'
def main():
if '-h' in sys.argv or '--help' in sys.argv or len(sys.argv) not in (3, 4):
usage()
sys.exit(1)
sqldb = sys.argv[1]
channel = sys.argv[2]
if len(sys.argv) >= 4:
botname = sys.argv[3]
if not os.path.exists(sqldb):
print 'Unable to open %s' % sqldb
sys.exit(1)
praises = Praise.Praise().db
larts = Lart.Lart().db
db = sqlite.connect(sqldb)
cursor = db.cursor()
total = 0
failed = 0
success = 0
for (table, plugin) in (('larts', larts), ('praises', praises))
cursor.execute("""SELECT * FROM %s ORDER BY id ASC""" % table)
table = table[:-1]
if cursor.rowcount != 0:
entries = cursor.fetchall()
for entry in entries:
try:
total += 1
text = entry[1]
by = entry[2]
if by is None:
by = botname
plugin.add(channel, table, text, by)
success += 1
except dbi.Error:
failed += 1
print '%s/%s entries successfully added. %s failed.' % (success, total,
failed)
print 'Dbs are at: %s' % ', '.join((praises.filename, larts.filename))
db.close()
praises.close()
larts.close()
sys.exit(0)
if __name__ == '__main__':
main()