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-23 17:51:36 +02:00
parent f47f590ce3
commit e98174b0be
2 changed files with 7 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
drama/helpers/aws.py
View file

@ -62,13 +62,14 @@ def upload_file(name, file, resize=None):
img = io.BytesIO() img = io.BytesIO()
i.save(img, format='PNG') i.save(img, format='PNG')
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())})
remove(tempname)
try: resp = req.json()['data'] try: resp = req.json()['data']
except Exception as e: except Exception as e:
print(e) print(e)
print(req) print(req)
print(req.text) print(req.text)
return if "File type invalid" in req.text: return "sex"
remove(tempname) else: return
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']
@ -76,7 +77,6 @@ def upload_file(name, file, resize=None):
print(e) print(e)
print(req) print(req)
print(req.text) print(req.text)
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(e)

5
drama/routes/posts.py
View file

@ -34,7 +34,10 @@ def resize():
file.write(chunk) file.write(chunk)
image = upload_from_file("resizing", "resizing", (100, 100)) image = upload_from_file("resizing", "resizing", (100, 100))
if image != None: if image == "sex":
u.resized = True
g.db.add(u)
elif image != None:
u.profileurl = image u.profileurl = image
u.resized = True u.resized = True
g.db.add(u) g.db.add(u)