convert urllib code to aiohttp

This commit is contained in:
Abigail Gold 2019-10-05 19:59:19 -04:00
parent 640ad9f8e3
commit 96d251fe47
No known key found for this signature in database
GPG Key ID: 80A676456AB6B045
1 changed files with 6 additions and 3 deletions

View File

@ -12,7 +12,7 @@ import discord.ext.commands as commands
import random import random
import json import json
import urllib import aiohttp
class StudyCog(commands.Cog): class StudyCog(commands.Cog):
def __init__(self, bot): def __init__(self, bot):
@ -51,8 +51,11 @@ class StudyCog(commands.Cog):
'(Note that only the US question pools are available).') '(Note that only the US question pools are available).')
return return
with urllib.request.urlopen(f'https://hamstudy.org/pools/{selected_pool}') as url: async with aiohttp.ClientSession() as session:
pool = json.loads(url.read().decode())['pool'] async with session.get(f'https://hamstudy.org/pools/{selected_pool}') as resp:
if resp.status != 200:
return await ctx.send('Could not load questions...')
pool = json.loads(await resp.read())['pool']
# Select a question # Select a question
pool_section = random.choice(pool)['sections'] pool_section = random.choice(pool)['sections']