一、Django限制用戶上傳文件格式與大小的優(yōu)異處理方式

一個處理該問題的優(yōu)異方式是擴展Django的FileFiled字段，在創(chuàng)建模型時直接設(shè)置可以接受的文件類型，并限定可以上傳的文件的最大尺寸。本文教你如何擴展Django的FileField字段，并展示如何在模型中使用它，從而以優(yōu)異方式限制用戶上傳文件的格式與大小。

擴展FileField字段

首先在你APP文件夾內(nèi)新建fields.py, 并添加如下代碼(來自stackoverflow)。新擴展的FileField叫RestrictedFileField，繼承了FileField類，并包含了額外的兩個可選參數(shù): 可接受的內(nèi)容類型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)

??????? 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

如何使用擴展后的RestrictedFileField

首先你要從創(chuàng)建的fields.py里導(dǎo)入RestrictedFileField，然后在模型中按如下代碼使用它。該字段可以用于文件，也可以用于圖片。與ImageField和FileField相比，它多了content_types和max_upload_size選項，從此你再也不用擔(dān)心用戶上傳文件時為所欲為啦。 這個字段是如此有用，說不定Django某一天會把它變成默認(rèn)字段哦。小編我期待有那一天的到來。

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=[‘a(chǎn)pplication/pdf’, ‘a(chǎn)pplication/excel’, ‘a(chǎn)pplication/msword’,

‘text/plain’, ‘text/csv’, ‘a(chǎn)pplication/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/gif’, ‘image/bmp’, ‘image/tiff’],

max_upload_size=5242880,)

延伸閱讀：

二、data是什么

data 用來保存實際變量的地址。

data 中的內(nèi)容會根據(jù)實際情況變化，因為 golang 在函數(shù)傳參和賦值時是 值傳遞 的，所以：

1.? 如果實際類型是一個值，那么 interface 會保存這個值的一份拷貝。interface 會在堆上為這個值分配一塊內(nèi)存，然后 data 指向它。

2.? 如果實際類型是一個指針，那么 interface 會保存這個指針的一份拷貝。由于 data 的長度恰好能保存這個指針的內(nèi)容，所以 data 中存儲的就是指針的值。它和實際數(shù)據(jù)指向的是同一個變量。