Add backfill MANAGE_SHARES migration script - fix existing permissions

data-dot-all · Nov 18, 2024 · 69c49ba · 69c49ba
1 parent 90d5705
commit 69c49ba
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/backend/migrations/versions/af2e1362d4cb_add_tenant_share_permissions.py b/backend/migrations/versions/af2e1362d4cb_add_tenant_share_permissions.py
@@ -52,7 +52,15 @@ def upgrade():
             group_uri=group.groupUri,
             tenant_name=TENANT_NAME,
         )
-        TenantPolicyService.associate_permission_to_tenant_policy(session, policy, MANAGE_SHARES)
+        already_associated = TenantPolicyRepository.has_group_tenant_permission(
+            session,
+            group_uri=group.groupUri,
+            permission_name=MANAGE_SHARES,
+            tenant_name=TENANT_NAME,
+        )
+
+        if not already_associated:
+            TenantPolicyService.associate_permission_to_tenant_policy(session, policy, MANAGE_SHARES)
 
 
 def downgrade():