修复：装填前复核仓库权限

2026-07-26 18:16:34 +08:00
parent f6e994d95f
commit 66890ca63c
5 changed files with 184 additions and 14 deletions
@@ -14,11 +14,22 @@ class AccessService:
        db: AsyncSession,
        current_user: CurrentUserModel,
        warehouse_id: int,
+        *,
+        for_update: bool = False,
    ) -> None:
        """校验当前用户是否有权访问指定仓库。"""
        if current_user.user.admin:
            return
-        if not await AccessDao.user_has_warehouse(db, current_user.user.user_id, warehouse_id):
+        if for_update:
+            has_warehouse = await AccessDao.user_has_warehouse(
+                db,
+                current_user.user.user_id,
+                warehouse_id,
+                for_update=True,
+            )
+        else:
+            has_warehouse = await AccessDao.user_has_warehouse(db, current_user.user.user_id, warehouse_id)
+        if not has_warehouse:
            raise PermissionException(data='', message='无权访问该仓库')

    @classmethod