Fastapi上传下载文件

1.上传文件 upload.py

from typing import List
from fastapi import FastAPI, File, UploadFile
from fastapi.responses import HTMLResponse
upload = FastAPI()

＃ 第一种读取文件方法，读取二进制文件
@upload.post("/files/")
async def create_files(files: List[bytes] = File(...)):
    # 保存文件名
    # with open('保存文件名字', 'wb') as f:
    #     f.write(files[0])
    return {"file_sizes": [len(file) for file in files]}

# 第二种读取文件的方法，会报错文件名，文件本身等信息
@upload.post("/uploadfiles/")
async def create_upload_files(files: List[UploadFile] = File(...)):
    
    # 保存文件
    # data = await files[0].read()
    # with open('保存文件名字', 'wb') as f:
    #     f.write(data)
    return {"filenames": [file.filename for file in files]}

@upload.get("/")
async def main():
    content = """










    """
    return HTMLResponse(content=content)

if __name__ == '__main__':
    import uvicorn
    uvicorn.run('upload:upload', host='0.0.0.0', port=8900)

输入：http://0.0.0.0:8900/

上传文件

2.下载文件 download.py

import pandas as pd
from fastapi import FastAPI
from fastapi.responses import FileResponse,StreamingResponse,Response
import io

download = FastAPI()
file_path = "文件路径"

# 假设有个pandas DataFrame文件文件需要保存
@download.get("/from_data/")
def main():
    df = pd.DataFrame([1,2,3])
    stream = io.StringIO()
    df.to_csv(stream, index=False)
    response = StreamingResponse(iter([stream.getvalue()]), media_type="text/csv")
    response.headers["Content-Disposition"] = "attachment; filename=export.csv"
    return response

# 假设我们预定了文件的路径
@download.get("/from_path/")
def main():
    return FileResponse(path=file_path, filename=file_path, media_type='py')


if __name__ == '__main__':
    import uvicorn
    uvicorn.run('download:download', host='0.0.0.0', port=8800)

输入：http://0.0.0.0:8900/from_data/ 或者 http://0.0.0.0:8900/from_path/

下载文件

3.下载excel文件

# -*- coding: utf-8 -*-
"""
-------------------------------------------------
   File Name：     xxx
   Description :
   Author :       asdil
   date：          2022/5/11
-------------------------------------------------
   Change Activity:
                   2022/5/11:
-------------------------------------------------
"""
__author__ = 'Asdil'
import pandas as pd
from fastapi import FastAPI
from fastapi.responses import FileResponse,StreamingResponse,Response
import io

download = FastAPI()
file_path = "文件路径"

# 假设有个pandas DataFrame文件文件需要保存
@download.get("/from_data/")
def main():
    df = pd.DataFrame([1,2,3])
    stream = io.BytesIO()
    writer = pd.ExcelWriter(stream, engine='xlsxwriter')
    df.to_excel(writer, index=False)
    writer.save()
    xlsx_data = stream.getvalue()
    response = StreamingResponse(io.BytesIO(xlsx_data))
    response.media_type = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
    response.headers['Content-Disposition'] = f'attachment; filename="这里是你要保存的名字.xlsx"'.encode("utf-8").decode("latin1")
    return response


if __name__ == '__main__':
    import uvicorn
    uvicorn.run('xxx:download', host='0.0.0.0', port=8800)