Merge pull request #102 from ScilifelabDataCentre/develop

Missing db migration
ScilifelabDataCentre · Jun 22, 2021 · 6abeec4 · 6abeec4
2 parents d3ed572 + e9aefbf
commit 6abeec4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/backend/db_management.py b/backend/db_management.py
@@ -6,7 +6,7 @@
 import utils
 from migrations import MIGRATIONS
 
-DB_VERSION = 2
+DB_VERSION = 3
 
 
 def check_db(config: dict):

diff --git a/backend/migrations.py b/backend/migrations.py
@@ -28,5 +28,15 @@ def migrate_v1_to_v2(db):
     db["datasets"].update_many({}, {"$unset": {"cross_references": ""}})
 
 
+def migrate_v2_to_v3(db):
+    """
+    Update the database fields to match the changes in the data structure.
+
+    * Remove the ``DATA_LIST`` and ``STATISTICS`` permissions from all users.
+    """
+    logging.info("Remove the DATA_LIST and STATISTICS permissions")
+    db["users"].update_many({}, {"$pull": {"permissions": {"$in": ["STATISTICS", "DATA_LIST"]}}})
+
+
 # Position 0 is empty since the first release is 1
-MIGRATIONS = [None, migrate_v1_to_v2]
+MIGRATIONS = [None, migrate_v1_to_v2, migrate_v2_to_v3]