diff --git a/app.py b/app.py index 3b294de..c2552be 100644 --- a/app.py +++ b/app.py @@ -8,6 +8,7 @@ from flask import Flask, jsonify, request from pygrocy import EntityType, Grocy import atexit import base64 +import mimetypes import loguru import redis import requests @@ -301,10 +302,16 @@ def consume(): def upload_image(url): # download image + url = url.split('?')[0] + image_name = url.split('/')[-1] img = requests.get(url) + mime_type = mimetypes.guess_type(image_name)[0] + if not mime_type: + mime_type = 'image/jpeg' # compress image compress_url = 'https://upload-tinypng.tunpok.com/upload/' - resp = requests.post(compress_url, files={'file': img.content}) + files = {'file': ('img.jpeg', img.content, mime_type,)} + resp = requests.post(compress_url, files=files, ) if resp.status_code == 200: return resp.json()['cdn_url'] return None