X-Git-Url: http://git.treefish.org/~alex/photosort.git/blobdiff_plain/0c3d67ff5da26de2b8b531caaaf4b65ec8dc135c..17a3b0beb3e13a00515d22fa147f121582adc6df:/src/registry.py

diff --git a/src/registry.py b/src/registry.py
index 3d8cfd3..a2bad86 100644
--- a/src/registry.py
+++ b/src/registry.py
@@ -15,27 +15,27 @@ class Registry:
             self._conn = sqlite3.connect(db_file, check_same_thread=False)
 
     def is_registered(self, name):
-        if not db_file: return False
+        if not self._conn: return False
         c = self._conn.cursor()
         c.execute("SELECT COUNT(*) FROM cache WHERE name=?", (name,))
-        return c.fetchone() != None
+        return c.fetchone()[0] == 1
 
     def clean(self):
-        if not db_file: return
+        if not self._conn: return
         c = self._conn.cursor()
         c.execute("DELETE FROM cache WHERE access_time<?", (int(time.time()) - 604800,))
         self._conn.commit()
 
 
     def register(self, name):
-        if not db_file: return
+        if not self._conn: return
         c = self._conn.cursor()
         c.execute( '''INSERT INTO cache (name, access_time) VALUES (?, ?)''',
                    (name, int(time.time())) )
         self._conn.commit()
 
     def refresh(self, name):
-        if not db_file: return
+        if not self._conn: return
         c = self._conn.cursor()
         c.execute( "UPDATE cache SET access_time=? WHERE name=?",
                    (int(time.time()), name) )