OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-29 09:42:12 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-10-15 02:51:08 -07:00
parent 97bb9d41a6
commit b4cd503a02
2 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
backend/open_webui/apps/webui/routers/auths.py
View File

@ -64,8 +64,10 @@ async def get_session_user(
expires_delta=expires_delta, expires_delta=expires_delta,
) )
datetime_expires_at = datetime.datetime.fromtimestamp( datetime_expires_at = (
expires_at, datetime.timezone.utc datetime.datetime.fromtimestamp(expires_at, datetime.timezone.utc)
if expires_at
else None
) )
# Set the cookie token # Set the cookie token
@ -193,8 +195,10 @@ async def signin(request: Request, response: Response, form_data: SigninForm):
expires_delta=expires_delta, expires_delta=expires_delta,
) )
datetime_expires_at = datetime.datetime.fromtimestamp( datetime_expires_at = (
expires_at, datetime.timezone.utc datetime.datetime.fromtimestamp(expires_at, datetime.timezone.utc)
if expires_at
else None
) )
# Set the cookie token # Set the cookie token
@ -276,8 +280,10 @@ async def signup(request: Request, response: Response, form_data: SignupForm):
expires_delta=expires_delta, expires_delta=expires_delta,
) )
datetime_expires_at = datetime.datetime.fromtimestamp( datetime_expires_at = (
expires_at, datetime.timezone.utc datetime.datetime.fromtimestamp(expires_at, datetime.timezone.utc)
if expires_at
else None
) )
# Set the cookie token # Set the cookie token

2
backend/open_webui/utils/utils.py
View File

@ -104,7 +104,7 @@ def get_current_user(
Users.update_user_last_active_by_id(user.id) Users.update_user_last_active_by_id(user.id)
return user return user
else: else:
response.delete_cookie("token") response.set_cookie("token", "", expires=0)
raise HTTPException( raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, status_code=status.HTTP_401_UNAUTHORIZED,