【发布时间】:2020-04-17 20:15:30
【问题描述】:
我有一个模型文档,管理员可以将图像上传到 FileField。成功上传文档/图像后,我还会保存图像的 sha256“指纹”,以测试管理员是否尝试上传重复图像。如果检测到重复图像,我不会保存重复图像并通过消息框架向管理员显示错误消息。但是,我也收到文件已成功上传的消息。我怎样才能防止这种情况发生?
我的缩写形式的代码:
class Document(Model):
document_id = models.AutoField(primary_key=True)
computed_sha256 = models.CharField(editable=False, max_length=64, default="foobar")
storage_file_name = models.FileField('File name', upload_to=settings.DOCUMENT_FOLDER_ORIGINALS, default=settings.DEFAULT_IMAGE_XXXLARGE_PATH,)
class DocumentAdmin(admin.ModelAdmin):
def save_model(self, request, obj, form, change):
if form.is_valid():
if not change:
# Uploading one or more images
files = request.FILES.getlist('storage_file_name')
if files:
for f in files:
# Check if this file has been uploaded before by checking the fingerprint
_file = form.cleaned_data["storage_file_name"]
sha256 = image_processing_utils.compute_sha256(_file)
duplicate_files = Document.objects.filter(computed_sha256 = sha256)
if len(duplicate_files) > 0:
messages.add_message(request, messages.WARNING, 'Uploading a duplicate of "%s" and it will not be saved' % f.name)
break;
# more image processing stuff
else:
# some more image processing stuff
obj.metadata = form.cleaned_data['metadata']
super().save_model(request, obj, form, change)
我不确定 Django 在哪里添加了图片已正确上传的正面消息。如何删除它并仅显示未上传重复图像的消息?
谢谢!
标记
【问题讨论】:
标签: django django-models django-admin