Commit 76c948e0 authored by 杨明橙's avatar 杨明橙

修改查询基金 权限验证

parent 05e0cb9e
...@@ -93,13 +93,20 @@ async def update( ...@@ -93,13 +93,20 @@ async def update(
fund_id: str, fund_id: str,
update_fund_data: UpdateFund, update_fund_data: UpdateFund,
user: User = Depends(get_current_user), user: User = Depends(get_current_user),
permission_user_collect: AgnosticCollection = Depends(get_permission_user_collect),
permission_role_collect: AgnosticCollection = Depends(get_permission_role_collect),
fund_collect: AgnosticCollection = Depends(get_fund_collect) fund_collect: AgnosticCollection = Depends(get_fund_collect)
): ):
# 验证权限
if not await check_permission(['data_permission.fund.update_info'], fund_id, user.email, permission_user_collect,
permission_role_collect):
raise FundPermissionError()
db_update_data = update_fund_data.dict(exclude_unset=True) db_update_data = update_fund_data.dict(exclude_unset=True)
db_update_data.update({ db_update_data.update({
"update_time": int(datetime.datetime.utcnow().timestamp()) "update_time": int(datetime.datetime.utcnow().timestamp())
}) })
data = await fund_collect.find_one_and_update({'id': fund_id, 'user_id': user.id}, {'$set': db_update_data}, data = await fund_collect.find_one_and_update({'id': fund_id}, {'$set': db_update_data},
return_document=ReturnDocument.AFTER) return_document=ReturnDocument.AFTER)
assert data, NotFundError() assert data, NotFundError()
response_model = fund_type_map[data['fund_type']] response_model = fund_type_map[data['fund_type']]
...@@ -120,7 +127,7 @@ async def get( ...@@ -120,7 +127,7 @@ async def get(
permission_role_collect): permission_role_collect):
raise FundPermissionError() raise FundPermissionError()
data = await fund_collect.find_one({'id': fund_id, 'user_id': user.id}) data = await fund_collect.find_one({'id': fund_id})
assert data, NotFundError() assert data, NotFundError()
response_model = fund_type_map[data['fund_type']] response_model = fund_type_map[data['fund_type']]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment