Moving new table definition into migration section

This ensures that upgrades from existing databases will be seamless
mintsoft · Dec 11, 2015 · 8ad7617 · 8ad7617
1 parent ece555c
commit 8ad7617
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/resources/data/models/Catalogue.py b/resources/data/models/Catalogue.py
@@ -53,10 +53,6 @@ def __init__(self, database_path):
                 CREATE TABLE cache_status (
                     etag TEXT
                 ) ''')
-            database.execute('''
-                CREATE TABLE cookies (
-                    cookieblob TEXT
-            )''')
             database.execute('''
                 CREATE TABLE auth (
                     token TEXT
@@ -127,6 +123,12 @@ def __init__(self, database_path):
             database.commit()
         else:
             database = sqlite3.connect(database_path)
+            # Deviations from the original schema should be
+            # defined here so that upgrades will work correctly
+            database.execute('''
+                CREATE TABLE IF NOT EXISTS cookies (
+                    cookieblob TEXT
+            )''')
 
         database.row_factory = sqlite3.Row
         self.database = database