FastAPI学习笔记(一)-15.上传文件

1.小文件上传使用内存,文件以bytes的类型读入内存。
2.File是对文件进行验证
3.多个小文件使用List[bytes]
4.上传大文件使用UploadFile类,多个大文件使用List[UploadFile]
5.上传大文件需要时间等待，因此需要使用 async + await的关键字来进行异步读取文件。

'''
@author:invoker
@project:fastapi202108
@file: chapter044.py
@contact:invoker2021@126.com
@descript:
@Date:2021/8/6 18:17
@version: Python 3.7.8
'''

from fastapi import APIRouter, File, UploadFile
from typing import List

app044 = APIRouter()

"""
单文件和多文件上传
"""


# 小文件上传使用内存,文件以bytes的类型读入内存。
# File是对文件进行验证
@app044.post("/file")
async def file_1(file: bytes = File(...)):
    """

    :param file:
    :return:
    """
    return {"file_size": len(file)}


# 多个小文件上传
@app044.post("/file_2")
async def file_2(file: List[bytes] = File(...)):
    """

    :param file:
    :return:
    """
    return {"file1_size": len(file[0]), "file2_size": len(file[1])}


@app044.post("/upload_files")
async def upload_files(files: List[UploadFile] = File(...)):
    """
     使用UploadFile类的优势：
     1.文件刚开始也是存储在内存中，当内存到达阀值后，将被保存在硬盘中，不会出现内容爆表的问题
     2.适合上传大图片或则视频
     3.可以读取到文件的元数据，比如文件名，创建时间等。
     4.有文件对象异步接口，在异步的函数（async）使用异步的方式（await）去读取或则更改文件
     5.上传文件是Python文件对象，可以使用write(),read(),seek(),close()操作
    :param file:
    :return:
    """

    for file in files:
        contents = await file.read()
        # await等待 file.read()读完文件才会赋值给contents
        print(contents)
        return {"filename":files[0].filename,
                "content_type":files[0].content_type}