Skip to content

Commit a573207

Browse files
committed
perf: 支持设置用户权限验证属性名称
1 parent 9728407 commit a573207

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

fastapi_user_auth/mixins/schemas.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,13 +54,14 @@ class UserSelectPerm(SelectPerm):
5454
"""所属用户选择数据集,只能选择匹配当前用户的数据"""
5555

5656
user_column: str = "user_id"
57+
user_attr: str = "id"
5758

5859
async def _call(self, admin: ModelAdmin, request: Request, sel: Select) -> Select:
59-
user_id = await admin.site.auth.get_current_user_identity(request, name="id")
60-
if not user_id: # 未登录
60+
user = await admin.site.auth.get_current_user(request)
61+
if not user: # 未登录
6162
return sel.where(False)
6263
column = getattr(admin.model, self.user_column)
63-
return sel.where(column == user_id)
64+
return sel.where(column == getattr(user, self.user_attr))
6465

6566

6667
@dataclass

0 commit comments

Comments
 (0)