Update data

[xsetbg.git] / reload_db.py

diff --git a/reload_db.py b/reload_db.py
index 41a3c941fcc3b406f1f74f9ca65dc8e126a5b900..93eeac9fd82f40af6e0259161c649f2a1ee32211 100755 (executable)
--- a/reload_db.py
+++ b/reload_db.py
@@ -10,7 +10,7 @@ __copyright__ = "Copyright (C) 2007-2015 PhiloSoft Design"
 __license__ = "GNU GPL"
 
 import sys
-from sqlobject import sqlhub
+from sqlobject import sqlhub, SQLObjectNotFound
 from sqlobject.sqlbuilder import Insert
 from xsetbg_conf import xsetbg_conf
 from xsetbg_db import recreate_db
@@ -41,11 +41,17 @@ for line in dump_file:
     timestamp = convert_str(timestamp)
     if fs_encoding != 'utf-8':
         filename = filename.decode(fs_encoding).encode('utf-8')
-    values = {'last_shown': timestamp, 'full_name': filename}
-    if id:
-        values['id'] = id
-    query = txn.sqlrepr(Insert(xsetbg_db.sqlmeta.table, values=values))
-    txn.query(query)
+    try:
+        row = xsetbg_db.byFull_name(filename)
+    except SQLObjectNotFound:
+        values = {'last_shown': timestamp, 'full_name': filename}
+        if id:
+            values['id'] = id
+        query = txn.sqlrepr(Insert(xsetbg_db.sqlmeta.table, values=values))
+        txn.query(query)
+    else:
+        assert row.id == id
+        assert row.last_shown == timestamp
 
 txn.commit()
 sqlhub.processConnection = connection