Database Migration 4

Author: hoffmabc

db/datastore.py

@@ -10,7 +10,7 @@
 from protos import objects
 from protos.objects import Listings, Followers, Following
 from os.path import join
-from db.migrations import migration1, migration2, migration3
+from db.migrations import *
 
 
 class Database(object):
@@ -180,16 +180,19 @@ def _run_migrations(self):
         cursor.execute('''PRAGMA user_version''')
         version = cursor.fetchone()[0]
         conn.close()
+        
         if version == 0:
             migration1.migrate(self.PATH)
             migration2.migrate(self.PATH)
             migration3.migrate(self.PATH)
+            migration4.migrate(self.PATH)
         elif version == 1:
             migration2.migrate(self.PATH)
             migration3.migrate(self.PATH)
+            migration4.migrate(self.PATH)
         elif version == 2:
             migration3.migrate(self.PATH)
-
+            migration4.migrate(self.PATH)
 
 class HashMap(object):
     """

db/migrations/__init__.py

@@ -1 +1,2 @@
 __author__ = 'chris'
+__all__ = ['migration1', 'migration2', 'migration3', 'migration4']

db/migrations/migration4.py

@@ -3,7 +3,7 @@
 
 
 def migrate(database_path):
-    print "migrating to db version 3"
+    print "migrating to db version 4"
     conn = sqlite3.connect(database_path)
     conn.text_factory = str
     cursor = conn.cursor()
@@ -28,6 +28,6 @@ def migrate(database_path):
                            (follower.guid.encode("hex"), follower.SerializeToString().encode("hex"),))
 
     # update version
-    cursor.execute('''PRAGMA user_version = 3''')
+    cursor.execute('''PRAGMA user_version = 4''')
     conn.commit()
     conn.close()