Sea-Studio/PerToolBoxServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

v1.0.0后端最终修复版

Browse Source
This commit is contained in:
canglan
2026-04-01 16:22:44 +08:00
parent a38306307c
commit c4ad386b9f
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
backend/middleware/rate_limit.py
View File

@@ -11,7 +11,6 @@ from fastapi import Request, HTTPException
from slowapi import Limiter, _rate_limit_exceeded_handler from slowapi import Limiter, _rate_limit_exceeded_handler
from slowapi.util import get_remote_address from slowapi.util import get_remote_address
from slowapi.errors import RateLimitExceeded from slowapi.errors import RateLimitExceeded
from slowapi.extension import RateLimitDecorator
from ..config import settings from ..config import settings
# 创建限流器 # 创建限流器
@@ -24,11 +23,12 @@ def setup_rate_limit(app):
app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler) app.add_exception_handler(RateLimitExceeded, _rate_limit_exceeded_handler)
def rate_limit(requests: int = None, period: int = None): def rate_limit(requests: int = None, period: int = None):
"""限流装饰器工厂 - 修复版本""" """限流装饰器工厂"""
if not settings.RATE_LIMIT_ENABLED: if not settings.RATE_LIMIT_ENABLED:
return lambda func: func return lambda func: func
req = requests or settings.RATE_LIMIT_REQUESTS req = requests or settings.RATE_LIMIT_REQUESTS
per = period or settings.RATE_LIMIT_PERIOD per = period or settings.RATE_LIMIT_PERIOD
# 直接返回 slowapi 的限流装饰器
return limiter.limit(f"{req}/{per} seconds") return limiter.limit(f"{req}/{per} seconds")

1
requirements.txt
View File

@@ -15,3 +15,4 @@ passlib[bcrypt]==1.7.4
aiosmtplib==2.0.2 aiosmtplib==2.0.2
tencentcloud-sdk-python==3.0.1000 tencentcloud-sdk-python==3.0.1000
pycryptodome==3.20.0 pycryptodome==3.20.0
httpx==0.25.1