X-Git-Url: http://git.cascardo.info/?p=cascardo%2Fipsilon.git;a=blobdiff_plain;f=ipsilon%2Fproviders%2Fopenid%2Fstore.py;h=40e8b8eb4148027851fa7cf5421ad1e2c6f12ec6;hp=9b2bc4f9b38434ec1617758f40c450631135f259;hb=cff71af03913b5b0987171205ef0c460b2f6fff8;hpb=61de77fc7eae1f844944bd692d13bf27a1fda6fe

diff --git a/ipsilon/providers/openid/store.py b/ipsilon/providers/openid/store.py
index 9b2bc4f..40e8b8e 100644
--- a/ipsilon/providers/openid/store.py
+++ b/ipsilon/providers/openid/store.py
@@ -84,4 +84,14 @@ class OpenIDStore(Store, OpenIDStoreInterface):
         q.create()
 
     def _upgrade_schema(self, old_version):
-        raise NotImplementedError()
+        if old_version == 1:
+            # In schema version 2, we added indexes and primary keys
+            # pylint: disable=protected-access
+            table = self._query(self._db, 'association', UNIQUE_DATA_TABLE,
+                                trans=False)._table
+            self._db.add_constraint(table.primary_key)
+            for index in table.indexes:
+                self._db.add_index(index)
+            return 2
+        else:
+            raise NotImplementedError()