adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix ws exception when no token is provided

Browse Source
This commit is contained in:
Matthias
2022-09-10 14:29:58 +02:00
parent b344f78d00
commit 4250174de9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
freqtrade/rpc/api_server/api_auth.py
View File

@@ -53,13 +53,13 @@ def get_user_from_token(token, secret_key: str, token_type: str = "access") -> s
# https://github.com/tiangolo/fastapi/blob/master/fastapi/security/api_key.py # https://github.com/tiangolo/fastapi/blob/master/fastapi/security/api_key.py
async def validate_ws_token( async def validate_ws_token(
ws: WebSocket, ws: WebSocket,
ws_token: Union[str, None] = Query(..., alias="token"), ws_token: Union[str, None] = Query(default=None, alias="token"),
api_config: Dict[str, Any] = Depends(get_api_config) api_config: Dict[str, Any] = Depends(get_api_config)
): ):
secret_ws_token = api_config.get('ws_token', 'secret_ws_t0ken.') secret_ws_token = api_config.get('ws_token', 'secret_ws_t0ken.')
secret_jwt_key = api_config.get('jwt_secret_key', 'super-secret') secret_jwt_key = api_config.get('jwt_secret_key', 'super-secret')
if secrets.compare_digest(secret_ws_token, ws_token): if ws_token and secrets.compare_digest(secret_ws_token, ws_token):
# Just return the token if it matches # Just return the token if it matches
return ws_token return ws_token
else: else: