Limnoria/tools/noteConvert.py

51 lines
1.4 KiB
Python
Raw Normal View History

#!/usr/bin/env python
import os
import sys
import sqlite
import supybot.dbi as dbi
import supybot.plugins.Note as Note
def main():
if '-h' in sys.argv or '--help' in sys.argv or len(sys.argv) != 2:
print 'usage: %s <sqlitedb>' % sys.argv[0]
sys.exit(1)
sqldb = sys.argv[1]
if not os.path.exists(sqldb):
print 'Could not find %s' % sqldb
sys.exit(1)
TxtDb = Note.NoteDB()
db = sqlite.connect(sqldb)
cursor = db.cursor()
total = 0
failed = 0
success = 0
cursor.execute("""SELECT notified, read, from_id, to_id, public, note FROM notes ORDER BY id ASC""")
if cursor.rowcount != 0:
entries = cursor.fetchall()
for entry in entries:
try:
entry = tuple(entry)
total += 1
notified = bool(entry[0])
read = bool(entry[1])
frm = int(entry[2])
to = int(entry[3])
public = bool(entry[4])
text = entry[5]
id = TxtDb.send(frm, to, public, text)
if notified:
TxtDb.setNotified(id)
if read:
TxtDb.setRead(id)
success += 1
except dbi.Error:
failed += 1
print '%s/%s entries successfully added. %s failed.' % (success, total, failed)
db.close()
TxtDb.close()
sys.exit(0)
if __name__ == '__main__':
main()