skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

re

Browse source
This commit is contained in:
Aevann1 2021-09-13 17:58:52 +02:00
parent df028d68bd
commit 76f2312c93
1940 changed files with 206 additions and 252 deletions
Download patch file Download diff file Expand all files Collapse all files

64
files/helpers/images.py
View file

@ -11,7 +11,7 @@ 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=None, file=None, resize=False):
def upload(ibb=False, filepath=None, file=None, resize=False):
if file:
format = file.filename.split('.')[-1].lower().replace('jpg','png').replace('jpeg','png')
@ -47,66 +47,20 @@ def upload_ibb(filepath=None, file=None, resize=False):
try:
with open(filepath, 'rb') as f:
data={'image': base64.b64encode(f.read())}
req = requests.post(f'https://api.imgbb.com/1/upload?key={IBB_KEY}', data=data)
if ibb: req = requests.post(f'https://api.imgbb.com/1/upload?key={IBB_KEY}', data=data)
else: req = requests.post('https://api.imgur.com/3/upload.json', headers = {"Authorization": f"Client-ID {IMGUR_KEY}"}, data=data)
resp = req.json()['data']
url = resp['url']
if ibb: url = resp['url']
else: url = resp['link']
url = url.replace(".png", ".webp").replace(".jpg", ".webp").replace(".jpeg", ".webp")
except Exception as e:
print(e)
print(req.text)
return
return url
def upload_imgur(filepath=None, file=None, resize=False):
if file:
format = file.filename.split('.')[-1].lower().replace('jpg','png').replace('jpeg','png')
filepath = f"image.{format}"
file.save(filepath)
else: format = filepath.split('.')[-1].lower().replace('jpg','png').replace('jpeg','png')
if resize:
i = IImage.open(filepath)
size = 100, 100
frames = ImageSequence.Iterator(i)
def thumbnails(frames):
for frame in frames:
thumbnail = frame.copy()
thumbnail.thumbnail(size, IImage.ANTIALIAS)
yield thumbnail
frames = thumbnails(frames)
om = next(frames)
om.info = i.info
filepath = f"image.{i.format}".lower().replace('jpg','png').replace('jpeg','png')
try: om.save(filepath, save_all=True, append_images=list(frames), loop=0, optimize=True, quality=30)
except Exception as e:
print(e)
return
elif format != "gif":
i = IImage.open(filepath)
filepath = f"image.{i.format}".lower().replace('jpg','png').replace('jpeg','png')
i.save(filepath, optimize=True, quality=30)
try:
with open(filepath, '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']
url = resp['link']
if not "_d." in url:
url = url.replace(".png", "_d.png").replace(".jpg", "_d.jpg").replace(".jpeg", "_d.jpeg")
if "_d." in url: url += "?maxwidth=9999"
except Exception as e:
print(e)
print(req.text)
return
new_image = Image(text=url, deletehash=resp["deletehash"])
g.db.add(new_image)
if not ibb:
new_image = Image(text=url, deletehash=resp["deletehash"])
g.db.add(new_image)
return url