CodeMirror/Bonfire
1
0
Fork 0
mirror of synced 2024-06-03 11:14:33 +12:00
Code Issues Releases Wiki Activity

Corrected how to wrap the BytesIO object

Browse source
This commit is contained in:
Phxntxm 2016-12-09 21:34:17 -06:00
parent 6383b0bced
commit 821f33f5d1
1 changed files with 6 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
cogs/utils/utilities.py
View file

@ -54,15 +54,12 @@ def find_command(bot, command):
async def download_image(url): async def download_image(url):
"""Returns a file-like object based on the URL provided""" """Returns a file-like object based on the URL provided"""
headers = {'User-Agent': config.user_agent} headers = {'User-Agent': config.user_agent}
try: # Simply download the image
# Simply download the image with aiohttp.ClientSession(headers=headers) as session:
with aiohttp.ClientSession(headers=headers) as session: async with session.get(url) as r:
async with session.get(url) as r: # Then wrap it in a BytesIO object, to be used like an actual file
image_bytes = r.read() image = io.BytesIO(await r.read())
# Then wrap it in a BytesIO object, to be used like an actual file return image
return io.BytesIO(image_bytes)
except:
return
async def update_records(key, winner, loser): async def update_records(key, winner, loser):
# We're using the Harkness scale to rate # We're using the Harkness scale to rate