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

Merge pull request #1474 from WangEdward/main

Browse Source
This commit is contained in:
jxxghp
2024-02-12 15:19:03 +08:00
committed by GitHub
parent c4dc468479 ea8a4a3ec4
commit 928aaf0c19
1 changed files with 4 additions and 4 deletions
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: