Skip to content

Commit f81dc4d

Browse files
committed
Update API key desensitization
1 parent 37b253b commit f81dc4d

2 files changed

Lines changed: 6 additions & 1 deletion

File tree

crud/crud_api_key.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ async def get_by_key(self, db: AsyncSession, key: str) -> ApiKey | None:
4040
"""
4141
return await self.select_model_by_column(db, key=key)
4242

43-
async def get_select(self, user_id: int, is_superuser: bool, name: str | None, status: int | None) -> Select: # noqa: FBT001
43+
async def get_select(self, user_id: int | None, is_superuser: bool, name: str | None, status: int | None) -> Select: # noqa: FBT001
4444
"""
4545
获取 API Key 列表查询表达式
4646

service/api_key_service.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,10 @@ async def _get(*, db: AsyncSession, user_id: int, is_superuser: bool, pk: int) -
3131
api_key = await api_key_dao.get(db, pk)
3232
if not api_key:
3333
raise errors.NotFoundError(msg='API Key 不存在')
34+
35+
# 脱敏 key
36+
api_key.key = mask_key(api_key.key)
37+
3438
return api_key
3539

3640
async def get(self, *, db: AsyncSession, user_id: int, is_superuser: bool, pk: int) -> ApiKey:
@@ -67,6 +71,7 @@ async def get_list(
6771
api_key_select = await api_key_dao.get_select(user_id, is_superuser, name, status)
6872
page_data = await paging_data(db, api_key_select)
6973

74+
# 脱敏所有 key
7075
for item in page_data['items']:
7176
if item.get('key') is not None:
7277
item['key'] = mask_key(item.get('key'))

0 commit comments

Comments
 (0)