cyonjan/PyWxDump
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
xaoyaoo 2024-08-28 17:46:21 +08:00
parent 644537b9b6
commit c5b9c009df
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pywxdump/api/__init__.py
View File

@ -73,6 +73,7 @@ def gen_fastapi_app():
async def serve_file(filename: str): async def serve_file(filename: str):
# 构建完整的文件路径 # 构建完整的文件路径
file_path = os.path.join(web_path, filename) file_path = os.path.join(web_path, filename)
file_path = os.path.abspath(file_path)
# 检查文件是否存在 # 检查文件是否存在
if os.path.isfile(file_path): if os.path.isfile(file_path):
@ -81,6 +82,7 @@ def gen_fastapi_app():
# 如果 MIME 类型为空，则默认为 application/octet-stream # 如果 MIME 类型为空，则默认为 application/octet-stream
if mime_type is None: if mime_type is None:
mime_type = "application/octet-stream" mime_type = "application/octet-stream"
server_loger.info(f"[+] 无法获取文件 MIME 类型，使用默认值：{mime_type}")
# 返回文件 # 返回文件
return FileResponse(file_path, media_type=mime_type) return FileResponse(file_path, media_type=mime_type)