skip querying deleted vfolders when purge user

lablup · Jan 8, 2025 · d10dd16 · d10dd16
1 parent 368fdd6
commit d10dd16
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/src/ai/backend/manager/models/user.py b/src/ai/backend/manager/models/user.py
@@ -1107,7 +1107,13 @@ async def delete_vfolders(
 
         :return: number of deleted rows
         """
-        from . import VFolderDeletionInfo, initiate_vfolder_deletion, vfolder_permissions, vfolders
+        from . import (
+            VFolderDeletionInfo,
+            VFolderOperationStatus,
+            initiate_vfolder_deletion,
+            vfolder_permissions,
+            vfolders,
+        )
 
         async with engine.begin_session() as conn:
             await conn.execute(
@@ -1116,7 +1122,16 @@ async def delete_vfolders(
             result = await conn.execute(
                 sa.select([vfolders.c.id, vfolders.c.host, vfolders.c.quota_scope_id])
                 .select_from(vfolders)
-                .where(vfolders.c.user == user_uuid),
+                .where(
+                    sa.and_(
+                        vfolders.c.user == user_uuid,
+                        vfolders.c.status.not_in(
+                            VFolderOperationStatus.DELETE_ONGOING,
+                            VFolderOperationStatus.DELETE_COMPLETE,
+                            VFolderOperationStatus.DELETE_ERROR,
+                        ),
+                    )
+                ),
             )
             target_vfs = result.fetchall()