mirror of
https://github.com/Mikaela/Limnoria.git
synced 2024-12-21 02:02:53 +01:00
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
|
#!/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()
|