扩展图像字段以允许pdf(django)
我有我的形式的ImageField。 正如我发现它使用枕头来验证该文件实际上是一个图像。 这部分是伟大的,但我需要在这个表单字段也允许pdf。
所以它应该检查该文件是否为图像,如果不是,请检查它是否为pdf,然后加载并存储。
如果pdf检查可以真正检查文件格式,这很好,但只是扩展检查也足够了。
如果您在表单中使用forms.ImageField
,则无法执行此操作。 你需要使用forms.FileField
因为ImageField
唯一验证图像,并提出ValidationError
如果文件不是图像。
这是一个例子:
models.py
class MyModel(models.Model):
image = models.ImageField(upload_to='images')
forms.py
import os
from django import forms
from .models import MyModel
class MyModelForm(forms.ModelForm):
class Meta:
model = MyModel
fields = ['image']
image = forms.FileField()
def clean_image(self):
uploaded_file = self.cleaned_data['image']
try:
# create an ImageField instance
im = forms.ImageField()
# now check if the file is a valid image
im.to_python(uploaded_file)
except forms.ValidationError:
# file is not a valid image;
# so check if it's a pdf
name, ext = os.path.splitext(uploaded_file.name)
if ext not in ['.pdf', '.PDF']:
raise forms.ValidationError("Only images and PDF files allowed")
return uploaded_file
尽管上面的代码正确地验证了图像的有效性(通过调用ImageField.to_python()
方法),但要确定文件是否为PDF,它只会检查文件扩展名。 要真正验证PDF是否有效,可以尝试解决这个问题:检查PDF文件是否有效(Python)。 这种方法试图读取内存中的整个文件,如果文件太大,可能会消耗服务器的内存。
上一篇: Extend image field to allow pdf ( django )
下一篇: Problems trying to generate Kotlin application with Android Studio