Skip to content

Commit 3ad3c76

Browse files
committed
multiple token support
1 parent f68bc94 commit 3ad3c76

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

api/app/services/security/customBearerCheck.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,17 @@
11
import os
2+
import json
23
import secrets
34
from fastapi.security import HTTPBearer,OAuth2AuthorizationCodeBearer,HTTPBasicCredentials
45
from fastapi import Depends, HTTPException, status
5-
66
security = HTTPBearer()
77

88
def validate_token(credentials: HTTPBasicCredentials = Depends(security)):
9-
correct_token = secrets.compare_digest(credentials.credentials, os.environ.get('FILE_MANAGER_BEARER_TOKEN'))
10-
if not (correct_token):
9+
10+
for eachKey in os.environ.get('FILE_MANAGER_BEARER_TOKEN').split(','):
11+
if secrets.compare_digest(credentials.credentials, eachKey):
12+
return True
13+
else:
1114
raise HTTPException(
1215
status_code=status.HTTP_401_UNAUTHORIZED,
1316
detail="Incorrect token"
1417
)
15-
return True

0 commit comments

Comments
 (0)