2 """Reload a dump into DB
4 This file is a part of XSetBg.
8 __author__ = "Oleg Broytman <phd@phdru.name>"
9 __copyright__ = "Copyright (C) 2007-2015 PhiloSoft Design"
10 __license__ = "GNU GPL"
13 from sqlobject import SQLObjectNotFound
14 from sqlobject.sqlbuilder import Insert
15 from xsetbg_conf import xsetbg_conf
16 from xsetbg_db import recreate_db, SQLiteMassInsert
26 fs_encoding = xsetbg_conf.get("images", "fs_encoding")
27 dump_file = open(sys.argv[1], 'rU')
28 xsetbg_db = recreate_db()
29 count_new = count_old = 0
31 with SQLiteMassInsert() as txn:
32 for line in dump_file:
33 id, timestamp, filename = line.strip().split(None, 2)
35 timestamp = convert_str(timestamp)
36 filename = filename.decode(fs_encoding)
38 row = xsetbg_db.byFull_name(filename)
39 except SQLObjectNotFound:
40 values = {'last_shown': timestamp,
41 'full_name': filename.encode('utf-8')}
44 query = txn.sqlrepr(Insert(xsetbg_db.sqlmeta.table, values=values))
48 assert id is None or row.id == id
49 if row.last_shown is not None:
50 assert row.last_shown == timestamp
55 print "New images:", count_new
56 print "Existing images:", count_old