48 lines
1.3 KiB
Python
48 lines
1.3 KiB
Python
import asyncio, aiohttp
|
|
|
|
class Autocomplete(object):
|
|
def __init__(self, query, **kwargs):
|
|
self.query = query
|
|
self.pretty = kwargs.get('pretty', 'None')
|
|
self.format = kwargs.get('format', 'None')
|
|
loop = asyncio.get_event_loop()
|
|
self.session = aiohttp.ClientSession(loop=loop)
|
|
|
|
async def getRequest(url, **kwargs):
|
|
async with self.session.get(url, **kwargs) as response:
|
|
return await response.json()
|
|
|
|
self.scryfallJson = loop.run_until_complete(getRequest(
|
|
url='https://api.scryfall.com/cards/autocomplete?',
|
|
params={'q':self.query, 'pretty':self.pretty, 'format':self.format}))
|
|
|
|
if self.scryfallJson['object'] == 'error':
|
|
self.session.close()
|
|
raise Exception(self.scryfallJson['details'])
|
|
|
|
self.session.close()
|
|
|
|
def __checkForKey(self, key):
|
|
try:
|
|
return self.scryfallJson[key]
|
|
except KeyError:
|
|
return None
|
|
|
|
def object(self):
|
|
if self.__checkForKey('object') is None:
|
|
return KeyError('This card has no associated object key.')
|
|
|
|
return self.scryfallJson['object']
|
|
|
|
def total_items(self):
|
|
if self.__checkForKey('total_items') is None:
|
|
return KeyError('This card has no associated total items key.')
|
|
|
|
return self.scryfallJson['total_items']
|
|
|
|
def data(self):
|
|
if self.__checkForKey('data') is None:
|
|
return KeyError('This card has no associated data key.')
|
|
|
|
return self.scryfallJson['data']
|