codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 支持 Radarr 的 X-Api-Key 请求头

Browse Source
This commit is contained in:
Edward 2024-02-12 04:41:55 +00:00 committed by GitHub
parent c4dc468479
commit ea8a4a3ec4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/core/security.py
View File

@ -5,11 +5,11 @@ import json
import os import os
import traceback import traceback
from datetime import datetime, timedelta from datetime import datetime, timedelta
from typing import Any, Union, Optional from typing import Any, Union, Optional, Annotated
import jwt import jwt
from Crypto.Cipher import AES from Crypto.Cipher import AES
from Crypto.Util.Padding import pad from Crypto.Util.Padding import pad
from fastapi import HTTPException, status, Depends from fastapi import HTTPException, status, Depends, Header
from fastapi.security import OAuth2PasswordBearer from fastapi.security import OAuth2PasswordBearer
from passlib.context import CryptContext from passlib.context import CryptContext
@ -68,11 +68,11 @@ def get_token(token: str = None) -> str:
return token return token
def get_apikey(apikey: str = None) -> str: def get_apikey(apikey: str = None, x_api_key: Annotated[str | None, Header()] = None) -> str:
""" """
从请求URL中获取apikey 从请求URL中获取apikey
""" """
return apikey return apikey or x_api_key
def verify_uri_token(token: str = Depends(get_token)) -> str: def verify_uri_token(token: str = Depends(get_token)) -> str: