From de5884d28dda23d902baac529c0a727331b57eba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8C=BF=E5=B0=8F=E5=A4=A9?= <1638245306@qq.com>
Date: Thu, 7 Sep 2023 23:24:21 +0800
Subject: [PATCH] =?UTF-8?q?1.=E4=BF=AE=E5=A4=8D=E5=8D=95=E4=BE=8B=E6=9F=A5?=
 =?UTF-8?q?=E8=AF=A2=E6=97=B6,=E6=8A=A5404=E9=97=AE=E9=A2=98;?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/dvadmin/utils/filters.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/backend/dvadmin/utils/filters.py b/backend/dvadmin/utils/filters.py
index b68ec9e..dbd921f 100644
--- a/backend/dvadmin/utils/filters.py
+++ b/backend/dvadmin/utils/filters.py
@@ -111,8 +111,10 @@ class DataLevelPermissionsFilter(BaseFilterBackend):
         # (2, "本部门数据权限"),
         # (3, "全部数据权限"),
         # (4, "自定数据权限")
-        replace_str = re.compile('\d')
-        re_api = replace_str.sub('{id}', api)
+        re_api = api
+        _pk = request.parser_context["kwargs"].get('pk')
+        if _pk: # 判断是否是单例查询
+            re_api = re.sub(_pk,'{id}', api)
         role_id_list = request.user.role.values_list('id', flat=True)
         role_permission_list=RoleMenuButtonPermission.objects.filter(
             role__in=role_id_list,