2 """Reload a dump into DB
4 This file is a part of XSetBg.
9 from sqlobject import SQLObjectNotFound
10 from sqlobject.sqlbuilder import Insert
11 from xsetbg_conf import xsetbg_conf
12 from xsetbg_db import recreate_db, SQLiteMassInsert
22 fs_encoding = xsetbg_conf.get("images", "fs_encoding")
23 dump_file = open(sys.argv[1], 'rU')
24 xsetbg_db = recreate_db()
25 count_new = count_old = 0
27 with SQLiteMassInsert() as txn:
28 for line in dump_file:
29 id, timestamp, filename = line.strip().split(None, 2)
31 timestamp = convert_str(timestamp)
32 filename = filename.decode(fs_encoding)
34 row = xsetbg_db.byFull_name(filename)
35 except SQLObjectNotFound:
36 values = {'last_shown': timestamp,
37 'full_name': filename.encode('utf-8')}
40 query = txn.sqlrepr(Insert(xsetbg_db.sqlmeta.table, values=values))
44 assert id is None or row.id == id
45 if row.last_shown is not None:
46 assert row.last_shown == timestamp
51 print "New images:", count_new
52 print "Existing images:", count_old