cgiacofei/TandoorRecipes
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fixed import image error

Browse Source
This commit is contained in:
vabene1111
2020-12-15 22:54:10 +01:00
parent cdcca80196
commit f72fc699f8
1 changed files with 15 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
cookbook/views/data.py
View File

@ -3,7 +3,7 @@ from datetime import datetime
from io import BytesIO from io import BytesIO
import requests import requests
from PIL import Image from PIL import Image, UnidentifiedImageError
from django.contrib import messages from django.contrib import messages
from django.core.files import File from django.core.files import File
from django.db.transaction import atomic from django.db.transaction import atomic
@ -144,6 +144,7 @@ def import_url(request):
print(ingredient) print(ingredient)
if data['image'] != '': if data['image'] != '':
try:
response = requests.get(data['image']) response = requests.get(data['image'])
img = Image.open(BytesIO(response.content)) img = Image.open(BytesIO(response.content))
@ -157,6 +158,8 @@ def import_url(request):
img.save(im_io, 'PNG', quality=70) img.save(im_io, 'PNG', quality=70)
recipe.image = File(im_io, name=f'{uuid.uuid4()}_{recipe.pk}.png') recipe.image = File(im_io, name=f'{uuid.uuid4()}_{recipe.pk}.png')
recipe.save() recipe.save()
except UnidentifiedImageError:
pass
return HttpResponse(reverse('view_recipe', args=[recipe.pk])) return HttpResponse(reverse('view_recipe', args=[recipe.pk]))