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

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

擴(kuò)展FileField字段

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

如何使用擴(kuò)展后的RestrictedFileField

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

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 用來(lái)保存實(shí)際變量的地址。

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

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

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