40 lines
836 B
Python
40 lines
836 B
Python
import os
|
|
from fastapi import UploadFile
|
|
|
|
|
|
def file_path(*path):
|
|
"""
|
|
拼接返回文件路径
|
|
:param path:
|
|
:return:
|
|
"""
|
|
return_path = os.path.join(*path)
|
|
return return_path
|
|
|
|
|
|
def create_folder(*path):
|
|
"""根据路径创建文件夹"""
|
|
folder_path = os.path.join(*path)
|
|
try:
|
|
os.makedirs(folder_path, exist_ok=True)
|
|
except Exception as e:
|
|
print(f"创建文件夹时错误： {e}")
|
|
return folder_path
|
|
|
|
|
|
def save_images(*path, file: UploadFile):
|
|
"""
|
|
保存上传的图片
|
|
:param path: 路径
|
|
:param file: 文件
|
|
:return:
|
|
"""
|
|
save_path = os.path.join(*path, file.filename)
|
|
|
|
os.makedirs(os.path.dirname(save_path), exist_ok=True)
|
|
with open(save_path, "wb") as f:
|
|
for line in file.file:
|
|
f.write(line)
|
|
return save_path
|
|
|