24 lines
634 B
Python
24 lines
634 B
Python
# -*- coding: utf-8 -*-
|
|
# @version : 1.0
|
|
# @Creaet Time : 2021/12/5 8:45
|
|
# @File : save_file.py
|
|
# @IDE : PyCharm
|
|
# @desc : 保存图片到本地
|
|
|
|
from datetime import datetime
|
|
import os
|
|
from application.settings import TEMP_DIR
|
|
from fastapi import UploadFile
|
|
|
|
|
|
def save_tmp_file(file: UploadFile, data):
|
|
"""
|
|
保存临时文件
|
|
"""
|
|
date = datetime.strftime(datetime.now(), "%Y%m%d")
|
|
file_dir = os.path.join(TEMP_DIR, date)
|
|
if not os.path.exists(file_dir):
|
|
os.mkdir(file_dir)
|
|
with open(os.path.join(file_dir, file.filename), "wb") as f:
|
|
f.write(data)
|