django限制表单上传图片的大小

django的ImageField没有提供控制上传图片的内置方法，我们可以在表单验证的过程中用clean函数来控制，搬运博客园 python小童鞋，可以通过重写ImageField的方法来控制上传图片大小。

1、在APP内建立fields.py文件

 1 from django.db.models import ImageField
 2 from django.forms import forms
 3 from django.template.defaultfilters import filesizeformat
 4 
 5 class RImageField(ImageField):
 6     def __init__(self, *args, **kwargs):
 7         self.max_upload_size = kwargs.pop("max_upload_size", [])
 8         super().__init__(*args, **kwargs)
 9 
10     def clean(self, *args, **kwargs):
11         data = super().clean(*args, **kwargs)
12         file = data.file
13 
14         try:
15             if file.size > self.max_upload_size:
16                 raise forms.ValidationError('上传图片大小不能超过 {}， 当期图片大小 {}。'
17                                             .format(filesizeformat(self.max_upload_size),
18                                                     filesizeformat(file.size)))
19         except AttributeError:
20             pass
21         return data

2、在models.py中引用：

from APP.fields import RImageField 创建models的时候字段如下写就OK了，

from APP.fields import RImageField
......
image = RImageField(upload_to=upload_path_handler, max_length=256, max_upload_size=2621440, verbose_name='图片') #以字节为单位

posted on 2023-02-21 22:56 ldx-wsj 阅读(102) 评论(0) 收藏举报