文件上传系统

概述

在一个CMS系统中，除了增删改查的业务，可能最刚需的功能的就是图片上传了，而图片又是属于文件的范畴，所以，为了避免开发者重复编写上传相关的业务代码，Lin 为开发者提供了一个文件上传系统。

在 flask 框架中，虽说已经内置好了文件上传操作，但如果你想实现一个功能比较完备的文件上传系统，可能还需要考虑到诸如文件扩展名校验、文件大小校验、文件数量校验、文件存储位置等等问题。

lin 提供的文件上传系统，已经封装好了上述比较棘手耗时的功能，这可以帮助你简单快速地实现文件上传的相关业务。

使用

模式

我们把 lin 的文件上传接口定到了cms/file/这个路径上，你可以通过 post 方法上传文件。lin 默认的文件上传实现LocalUploader这个类，意思是前端上传的文件会保存到服务器本地，但更多的时候我们需要将文件保存到云服务器上，如阿里云或腾讯云，lin 并没有局限在此，你可以自己实现存储文件的逻辑，将文件保存到你希望的地方。

对于文件上传(file)这种可以有多种实现方式的功能，我们把其诸多实现归入到app/extensions/file这个目录下。如local_uploader.py这个文件实现了上传到本地的LocalUploader类，config.py文件是与其相关的配置文件。

  1. app/extensions/
  2. └── file
  3.     ├── config.py  # 配置文件
  4.     └── local_uploader.py # LocalUploader类实现文件

默认的情况下在，cms/file/这个接口使用LocalUploader这个文件上传实现类。

  1. @file_api.route('/', methods=['POST'])
  2. @login_required
  3. def post_file():
  4.     # 解析前端上传的文件
  5.     files = request.files
  6.     # 将解析后的字典作为参数传入LoaclUploader类
  7.     uploader = LocalUploader(files)
  8.     ret = uploader.upload()
  9.     return jsonify(ret)

在这段代码中， 我们首先调用了flask的请求对象request上的files字典，它会为我们自动解析客户端传输过来的文件，然后实例化了LocalUploader类，并将files作为构造参数传递给 LocalUploader，在LocalUploader构造方法中，我们校验了客户端传输过来的文件是否符合我们约定的配置，最后调用其upload方法。

记住这一步，它是固定的，任何继承自Uploader这个基类的实现类必须重写upload方法，如LocalUploader中：

  1. class LocalUploader(Uploader):
  3.     def upload(self):
  4.         pass

当需要实现其它的上传类时，如上传到阿里云 OSS，我们只需要重新定义一个Aliyun类，在类中重新实现这个upload方法，而后在视图函数中，更换LocalUploader为Aliyun即可快速切换。

实操

在具体的实践之前，我们需要了解一下file的具体配置。

  1. FILE = {
  2.     "STORE_DIR": 'app/assets',       # 文件的存储路径
  3.     "SINGLE_LIMIT": 1024 * 1024 * 2, # 单个文件的大小限制，默认2M
  4.     "TOTAL_LIMIT": 1024 * 1024 * 20, # 所有文件的大小限制，默认20M
  5.     "NUMS": 10,                      # 文件数量限制，默认10
  6.     "INCLUDE": [],                   # 文件后缀名的排除项，默认排除[]，即允许所有类型的文件上传
  7.     "EXCLUDE": []                    # 文件后缀名的包括项
  8. }

TIP

为了提高配置的灵活性，你也可以在实例化Uploader的实现类的时候，指定 config 参数来传入你的自定义配置。方法如下：

  1. # config 参数为一个字典。注意：当字典的key值和上述6个配置项key值相同时，才会被解析
  2. uploader = LocalUploader(files, config={'NUMS': 2, 'SINGLE_LIMIT': 1024 * 1024 * 1})

在单个配置的后面，笔者已经以注释的方式解释了每项的作用。当然还需要着重解释一下exclude和include这两项，默认情况下，只会读取它们中的一项，如果两项皆为空列表的话，所有的文件类型都会被允许上传，如果二者都不是空列表则取include为有效配置，总而言之，系统只会支持使用一项。

注意了！！！在app/extensions/下的配置文件，系统不会帮我们做自动加载，不同于app/config。因此，你必须在app/app.py的create_app方法中，显示的加载该配置文件

  1. ## 手动加载配置文件
  2. def create_app(register_all=True):
  3.     app = Flask(__name__, static_folder='./assets')
  4.     app.config.from_object('app.config.setting')
  5.     app.config.from_object('app.config.secure')
  6.     # 加载其他配置文件：
  7.     app.config.from_object('app.extensions.file.config')
  8.     ...

当做好了这些后，我们使用 postman 测试一下文件上传。

文件上传系统 - 图1

上传成功后，我们会得到如下的结果：

  1. [
  2.   {
  3.     key: 'one',
  4.     id: 13,
  5.     url:
  6.       'http://localhost:5000/assets/2019/05/19/3428c02f-dfb5-47a0-82cf-2d6c05285473.png'
  7.   },
  8.   {
  9.     key: 'two',
  10.     id: 14,
  11.     url:
  12.       'http://localhost:5000/assets/2019/05/19/bfcff6f4-8dd7-4dd8-af9d-660781d0e076.jpg'
  13.   }
  14. ];

由于上传了两个文件，因此我们得到了两个元素的数组，它们的结构如下：

name说明类型
key文件上传的 keystring
id文件存储到数据库的 idstring
url可访问的 urlstring

TIP

系统会自动帮助我们上传的文件做 md5，因此你大可不必担心文件重复上传，如果你上传了重复的文件，服务器会返回已传文件的数据。