mirror of
https://github.com/open-webui/open-webui.git
synced 2026-08-24 22:44:50 -06:00
refac
This commit is contained in:
@@ -1162,10 +1162,11 @@ async def get_channel_message(
|
||||
if message.channel_id != id:
|
||||
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=ERROR_MESSAGES.DEFAULT())
|
||||
|
||||
message_user = Users.get_user_by_id(message.user_id, db=db)
|
||||
return MessageResponse(
|
||||
**{
|
||||
**message.model_dump(),
|
||||
'user': UserNameResponse(**Users.get_user_by_id(message.user_id, db=db).model_dump()),
|
||||
'user': UserNameResponse(**message_user.model_dump()) if message_user else None,
|
||||
}
|
||||
)
|
||||
|
||||
@@ -1245,10 +1246,11 @@ async def pin_channel_message(
|
||||
try:
|
||||
Messages.update_is_pinned_by_id(message_id, form_data.is_pinned, user.id, db=db)
|
||||
message = Messages.get_message_by_id(message_id, db=db)
|
||||
message_user = Users.get_user_by_id(message.user_id, db=db)
|
||||
return MessageUserResponse(
|
||||
**{
|
||||
**message.model_dump(),
|
||||
'user': UserNameResponse(**Users.get_user_by_id(message.user_id, db=db).model_dump()),
|
||||
'user': UserNameResponse(**message_user.model_dump()) if message_user else None,
|
||||
}
|
||||
)
|
||||
except Exception as e:
|
||||
|
||||
@@ -656,7 +656,7 @@ async def get_html_file_content_by_id(id: str, user=Depends(get_verified_user),
|
||||
)
|
||||
|
||||
file_user = Users.get_user_by_id(file.user_id, db=db)
|
||||
if not file_user.role == 'admin':
|
||||
if not file_user or file_user.role != 'admin':
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_404_NOT_FOUND,
|
||||
detail=ERROR_MESSAGES.NOT_FOUND,
|
||||
|
||||
Reference in New Issue
Block a user