diff --git a/files/helpers/images.py b/files/helpers/images.py index 6fb63c1e6f..a4628c4cbc 100644 --- a/files/helpers/images.py +++ b/files/helpers/images.py @@ -11,9 +11,10 @@ CF_ZONE = environ.get("CLOUDFLARE_ZONE", "").strip() IMGUR_KEY = environ.get("IMGUR_KEY", "").strip() IBB_KEY = environ.get("IBB_KEY", "").strip() -def upload_ibb(filepath, resize=False): +def upload_ibb(file, resize=False): + if resize: - i = IImage.open(filepath) + i = IImage.open(file) size = 100, 100 frames = ImageSequence.Iterator(i) @@ -27,12 +28,12 @@ def upload_ibb(filepath, resize=False): om = next(frames) om.info = i.info - try: om.save(filepath, save_all=True, append_images=list(frames), loop=0, optimize=True, quality=30) + try: om.save(f"image.{i.format}", save_all=True, append_images=list(frames), loop=0, optimize=True, quality=30) except Exception as e: print(e) return try: - with open(filepath, 'rb') as f: + with open(file, 'rb') as f: data={'image': base64.b64encode(f.read())} req = requests.post(f'https://api.imgbb.com/1/upload?key={IBB_KEY}', data=data) resp = req.json()['data'] @@ -45,9 +46,10 @@ def upload_ibb(filepath, resize=False): return(url) -def upload_imgur(filepath, resize=False): +def upload_imgur(file, resize=False): + if resize: - i = IImage.open(filepath) + i = IImage.open(file) size = 100, 100 frames = ImageSequence.Iterator(i) @@ -61,12 +63,12 @@ def upload_imgur(filepath, resize=False): om = next(frames) om.info = i.info - try: om.save(filepath, save_all=True, append_images=list(frames), loop=0, optimize=True, quality=30) + try: om.save(f"image.{i.format}", save_all=True, append_images=list(frames), loop=0, optimize=True, quality=30) except Exception as e: print(e) return try: - with open(filepath, 'rb') as f: + with open(file, 'rb') as f: data={'image': base64.b64encode(f.read())} req = requests.post('https://api.imgur.com/3/upload.json', headers = {"Authorization": f"Client-ID {IMGUR_KEY}"}, data=data) resp = req.json()['data']