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

sneed

Browse source
This commit is contained in:
Aevann1 2021-07-22 23:03:27 +02:00
parent 6830239aa2
commit e0d4b62fbd
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
drama/helpers/aws.py
View file

@ -51,6 +51,7 @@ def upload_file(name, file, resize=None):
if resize: if resize:
tempname = name.replace("/", "_") tempname = name.replace("/", "_")
print(type(file))
file.save(tempname) file.save(tempname)
if tempname.split('.')[-1] in ['jpg', 'jpeg']: if tempname.split('.')[-1] in ['jpg', 'jpeg']:
@ -63,15 +64,24 @@ def upload_file(name, file, resize=None):
req = requests.post('https://api.imgur.com/3/upload.json', headers = {"Authorization": f"Client-ID {imgurkey}"}, data = {'image': base64.b64encode(img.getvalue())}) req = requests.post('https://api.imgur.com/3/upload.json', headers = {"Authorization": f"Client-ID {imgurkey}"}, data = {'image': base64.b64encode(img.getvalue())})
try: resp = req.json()['data'] try: resp = req.json()['data']
except Exception as e: except Exception as e:
print(e)
print(req)
print(req.text)
return return
remove(tempname) remove(tempname)
else: else:
req = requests.post('https://api.imgur.com/3/upload.json', headers = {"Authorization": f"Client-ID {imgurkey}"}, data = {'image': base64.b64encode(file.read())}) req = requests.post('https://api.imgur.com/3/upload.json', headers = {"Authorization": f"Client-ID {imgurkey}"}, data = {'image': base64.b64encode(file.read())})
try: resp = req.json()['data'] try: resp = req.json()['data']
except Exception as e: except Exception as e:
print(e)
print(req)
print(req.text)
return return
try: url = resp['link'].replace(".png", "_d.png").replace(".jpg", "_d.jpg").replace(".jpeg", "_d.jpeg") + "?maxwidth=9999" try: url = resp['link'].replace(".png", "_d.png").replace(".jpg", "_d.jpg").replace(".jpeg", "_d.jpeg") + "?maxwidth=9999"
except Exception as e: except Exception as e:
print(e)
print(req)
print(req.text)
return return
new_image = Image( new_image = Image(
@ -100,6 +110,9 @@ def upload_from_file(name, filename, resize=None):
remove(filename) remove(filename)
url = resp['link'].replace(".png", "_d.png").replace(".jpg", "_d.jpg").replace(".jpeg", "_d.jpeg") + "?maxwidth=9999" url = resp['link'].replace(".png", "_d.png").replace(".jpg", "_d.jpg").replace(".jpeg", "_d.jpeg") + "?maxwidth=9999"
except Exception as e: except Exception as e:
print(e)
print(req)
print(req.text)
return return
new_image = Image( new_image = Image(