Pre-merge edit

This commit is contained in:
IgorVolochay
2026-01-10 13:48:18 +03:00
parent 997204f414
commit e92ade6c94
6 changed files with 27 additions and 32 deletions
+4 -2
View File
@@ -29,7 +29,7 @@ logger.add(sys.stdout,
disable_docs = os.getenv("DISABLE_DOCS", "true").lower() == "true"
app: FastAPI = FastAPI(title="DropMeFiles analog",
summary="OpenAPI schema for \"DropMeFiles analog\" project!",
version="0.1",
version="1.0",
contact={"GitHub": "https://github.com/IgorVolochay/Drop-me-files-analog"},
docs_url=None if disable_docs else "/docs",
redoc_url=None if disable_docs else "/redoc",
@@ -38,6 +38,8 @@ app: FastAPI = FastAPI(title="DropMeFiles analog",
s3_worker = S3Worker()
redis_worker = RedisWorker()
max_file_size = int(os.getenv('MAX_FILES_SIZE', 1048576)) # 1 MB if None
class RealIPMiddleware(BaseHTTPMiddleware):
async def dispatch(self, request: Request, call_next):
forwarded = request.headers.get("X-Forwarded-For", "")
@@ -55,7 +57,7 @@ app.add_middleware(RealIPMiddleware)
async def get_upload_token(file_name: str, file_type: str, file_size: int, response: Response, request: Request) -> BaseResponse:
user_ip = request.state.client_ip
logger.debug(f"User IP: {user_ip}; Endpoint: /upload_token; File Name: {file_name}; File Type: {file_type}; File Size: {file_size}")
if file_size > int(os.getenv('MAX_FILES_SIZE')):
if file_size > max_file_size:
response.status_code = status.HTTP_413_CONTENT_TOO_LARGE
logger.warning(f"User IP: {user_ip}; File Name: {file_name}; Exception: The file is too large")
return BaseResponse(result="The uploaded file is too large", error=True)