Skip to content

GitLab

Commit 1066096b authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

fix

parent 398b2076
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
backend/utils/utils.py
View file @ 1066096b
...@@ -2,7 +2,6 @@ from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials ...@@ -2,7 +2,6 @@ from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
from fastapi import HTTPException, status, Depends from fastapi import HTTPException, status, Depends
from apps.web.models.users import Users from apps.web.models.users import Users
from apps.web.models.auths import Auths
from pydantic import BaseModel from pydantic import BaseModel
from typing import Union, Optional from typing import Union, Optional
...@@ -78,6 +77,8 @@ def get_http_authorization_cred(auth_header: str): ...@@ -78,6 +77,8 @@ def get_http_authorization_cred(auth_header: str):
def get_current_user( def get_current_user(
auth_token: HTTPAuthorizationCredentials = Depends(bearer_security), auth_token: HTTPAuthorizationCredentials = Depends(bearer_security),
): ):
print(auth_token)
# auth by api key # auth by api key
if auth_token.credentials.startswith("sk-"): if auth_token.credentials.startswith("sk-"):
return get_current_user_by_api_key(auth_token.credentials) return get_current_user_by_api_key(auth_token.credentials)
...@@ -99,7 +100,7 @@ def get_current_user( ...@@ -99,7 +100,7 @@ def get_current_user(
def get_current_user_by_api_key(api_key: str): def get_current_user_by_api_key(api_key: str):
user = Auths.authenticate_user_by_api_key(api_key) user = Users.get_user_by_api_key(api_key)
if user is None: if user is None:
raise HTTPException( raise HTTPException(
status_code=status.HTTP_401_UNAUTHORIZED, status_code=status.HTTP_401_UNAUTHORIZED,
......
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