在Django中如何限制上传文件的类型与大小

在Django中如何限制上传文件的类型与大小

Django是一个庞大有丰富的框架, 它内置了很多的模块, 今天这篇文章讲的就是有关文件的操作——在Django中如何限制上传文件的类型与大小

之前在项目中有用到element-ui的文件限制, 但终究是在前端浏览器做的限制, 后端上总觉得该再加一层处理, 先看看element-ui吧, 个人觉得还是很棒的。

 <el-upload
     class="upload-demo"
     drag
     style="margin-top: 130px"
     action="http://127.0.0.1:8000/uploads/"
     :before-upload="beforeUpload"  
     multiple
     accept=".md,.txt,.doc,.docx,.ppt,.pdf"
     :on-success="successInfo"
     :on-error="errorInfo"
	>
    <i class="el-icon-upload"></i>
    <div class="el-upload__text">将文件拖到此处，或<em>点击上传</em></div>
    <div class="el-upload__tip" slot="tip">只能上传 md、txt、doc、docx、ppt、pdf文件，且不超过2M</div>
</el-upload>

<script>
methods: {
			// 在上传之前校验
            beforeUpload(file) {
                var fileSuffix = file.name.substring(file.name.lastIndexOf(".") + 1);
                // 准许上传的类型
                var whiteList = ["md", "txt", "doc", "docx", "ppt", "pdf"];
                // 准许上传的大小
                var isLt2M = file.size / 1024 / 1024 < 2;
                
				// 校验
                if (this.formData.length == 0) {
                    this.formData = new FormData();
                } else if (whiteList.indexOf(fileSuffix) === -1) {
                    this.errormsg = "上传文件只能是 md、txt、doc、docx、ppt格式"
                    this.errorInfo()
                    return false;
                } else if (!isLt2M) {
                    this.errormsg = "上传文件大小不能超过 2MB"
                    this.errorInfo()
                    return false;
                } else {
                    this.formData.append('file', file.file);
                    // 执行上传操作
                    this.onSubmit()
                }

            },
</script>

今天在Flask中看到了它对上传文件类型和大小的限制, 相关可看Flask专栏Flask之文件上传操作。

from flask import Flask
app = Flask(__name__)

# 文件上传目录
app.config['UPLOAD_FOLDER'] = 'static/uploads'
# 支持上传的文件格式
app.config['ALLOWED_EXTENSIONS'] = {'png', 'jpg', 'jpeg', 'gif'} 
# 支持上传的文件大小
app.config['MAX_CONTENT_LENGTH'] = 1024 * 1024

我就在想Django是否有关于限制上传文件类型和大小的操作, 结果还真有(虽然显而易见), 话不多说, 直接上代码。

• 写一个限制类RestrictedFileField继承FileField, 用于扩展对文件的操作, 并包含了额外的两个可选参数: 可接受的内容类型content_types和max_upload_size最大上传尺寸(比如5242880=5MB)

  from django.db.models import FileField
  from django.forms import forms
  from django.template.defaultfilters import filesizeformat
  
  
  class RestrictedFileField(FileField):
      """ max_upload_size:
              2.5MB - 2621440
              5MB - 5242880
              10MB - 10485760
              20MB - 20971520
              50MB - 5242880
              100MB 104857600
              250MB - 214958080
              500MB - 429916160
      """
      def __init__(self, *args, **kwargs):
          self.content_types = kwargs.pop("content_types", [])
          self.max_upload_size = kwargs.pop("max_upload_size", [])
          super().__init__(*args, **kwargs)
  
      def clean(self, *args, **kwargs):
          data = super().clean(*args, **kwargs)   # clean()方法来自于FileField的父类Field, 用于验证
          file = data.file
  
          try:
              content_type = file.content_type
              # 自定义验证
              if content_type in self.content_types:
                  if file.size > self.max_upload_size:
                      raise forms.ValidationError('Please keep filesize under {}. Current filesize {}'.format(filesizeformat(self.max_upload_size), filesizeformat(file.size)))
              else:
                  raise forms.ValidationError('This file type is not allowed.')
          except AttributeError:
              pass
          return data
  

• 该功能可以用于文件，也可以用于图片。与ImageField和FileField相比，它多了content_types和max_upload_size选项

  from django.db import models
  from .fields import RestrictedFileField
   
   
  class File(models.Model):
      file = RestrictedFileField(upload_to=user_directory_path, max_length=100, content_types=['application/pdf', 'application/excel', 'application/msword', 'text/plain', 'text/csv', 'application/zip'], max_upload_size=5242880)
   
   
  class Image(models.Model):
      file = RestrictedFileField(upload_to=user_directory_path, max_length=100, content_types=['image/jpeg', 'image/gif', 'image/bmp', 'image/tiff'], max_upload_size=5242880,)