Refactored autocomplete to be more consistent with the rest of the library.

This commit is contained in:
Nanda Scott 2018-02-13 16:35:01 -05:00
parent 3d5be5a24e
commit 91a5f04113
1 changed files with 165 additions and 24 deletions

View File

@ -1,26 +1,17 @@
import asyncio, aiohttp from .cards_object import CardsObject
import urllib.parse
class Autocomplete(object): class Autocomplete(CardsObject):
def __init__(self, query, **kwargs): def __init__(self, **kwargs):
self.query = query self.dict = {
self.pretty = kwargs.get('pretty', 'None') 'q': kwargs.get('query', ''),
self.format = kwargs.get('format', 'None') 'pretty': kwargs.get('pretty', 'false'),
loop = asyncio.get_event_loop() 'format': kwargs.get('format', 'json')
self.session = aiohttp.ClientSession(loop=loop) }
async def getRequest(url, **kwargs): self.args = urllib.parse.urlencode(self.dict)
async with self.session.get(url, **kwargs) as response: self.url = 'cards/autocomplete?' + self.args
return await response.json() super(Autocomplete, self).__init__(self.url)
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): def __checkForKey(self, key):
try: try:
@ -30,18 +21,168 @@ class Autocomplete(object):
def object(self): def object(self):
if self.__checkForKey('object') is None: if self.__checkForKey('object') is None:
return KeyError('This card has no associated object key.') raise KeyError('This card has no key \'object\'')
return self.scryfallJson['object'] return self.scryfallJson['object']
def total_items(self): def total_items(self):
if self.__checkForKey('total_items') is None: if self.__checkForKey('total_items') is None:
return KeyError('This card has no associated total items key.') raise KeyError('This card has no key \'total_items\'')
return self.scryfallJson['total_items'] return self.scryfallJson['total_items']
def data(self): def data(self):
if self.__checkForKey('data') is None: if self.__checkForKey('data') is None:
return KeyError('This card has no associated data key.') raise KeyError('This card has no key \'data\'')
return self.scryfallJson['data'] return self.scryfallJson['data']
#The following attributes are only to override the inherited class attributes.
#This class has no matching attributes but we still need the getRequest function from CardsObject
def id(self):
raise AttributeError('Search object has no attribute \'id\'')
def multiverse_ids(self):
raise AttributeError('Search object has no attribute \'multiverse_ids\'')
def mtgo_id(self):
raise AttributeError('Search object has no attribute \'mtgo_id\'')
def mtgo_foil_id(self):
raise AttributeError('Search object has no attribute \'mtgo_foil_id\'')
def name(self):
raise AttributeError('Search object has no attribute \'name\'')
def uri(self):
raise AttributeError('Search object has no attribute \'uri\'')
def scryfall_uri(self):
raise AttributeError('Search object has no attribute \'scryfall_uri\'')
def layout(self):
raise AttributeError('Search object has no attribute \'layout\'')
def highres_image(self):
raise AttributeError('Search object has no attribute \'highres_image\'')
def image_uris(self):
raise AttributeError('Search object has no attribute \'image_uris\'')
def cmc(self):
raise AttributeError('Search object has no attribute \'cmc\'')
def type_line(self):
raise AttributeError('Search object has no attribute \'type_line\'')
def oracle_text(self):
raise AttributeError('Search object has no attribute \'oracle_text\'')
def mana_cost(self):
raise AttributeError('Search object has no attribute \'mana_cost\'')
def colors(self):
raise AttributeError('Search object has no attribute \'colors\'')
def color_identity(self):
raise AttributeError('Search object has no attribute \'color_identity\'')
def legalities(self):
raise AttributeError('Search object has no attribute \'legalities\'')
def reserved(self):
raise AttributeError('Search object has no attribute \'reserved\'')
def reprint(self):
raise AttributeError('Search object has no attribute \'reprint\'')
def set_code(self):
raise AttributeError('Search object has no attribute \' def set_code\'')
def set_name(self):
raise AttributeError('Search object has no attribute \' def set_name\'')
def set_uri(self):
raise AttributeError('Search object has no attribute \'set_uri\'')
def set_search_uri(self):
raise AttributeError('Search object has no attribute \'set_search_uri\'')
def scryfall_set_uri(self):
raise AttributeError('Search object has no attribute \'scryfall_set_uri\'')
def rulings_uri(self):
raise AttributeError('Search object has no attribute \'rulings_uri\'')
def prints_search_uri(self):
raise AttributeError('Search object has no attribute \'prints_search_uri\'')
def collector_number(self):
raise AttributeError('Search object has no attribute \'collector_number\'')
def digital(self):
raise AttributeError('Search object has no attribute \'digital\'')
def rarity(self):
raise AttributeError('Search object has no attribute \'rarity\'')
def illustration_id(self):
raise AttributeError('Search object has no attribute \'illustration_id\'')
def artist(self):
raise AttributeError('Search object has no attribute \'artist\'')
def frame(self):
raise AttributeError('Search object has no attribute \'frame\'')
def full_art(self):
raise AttributeError('Search object has no attribute \'full_art\'')
def border_color(self):
raise AttributeError('Search object has no attribute \'border_color\'')
def timeshifted(self):
raise AttributeError('Search object has no attribute \'timeshifted\'')
def colorshifted(self):
raise AttributeError('Search object has no attribute \'colorshifted\'')
def futureshifted(self):
raise AttributeError('Search object has no attribute \'futureshifted\'')
def edhrec_rank(self):
raise AttributeError('Search object has no attribute \'edhrec_rank\'')
def currency(self, mode):
raise AttributeError('Search object has no attribute \'currency(self,\'')
def related_uris(self):
raise AttributeError('Search object has no attribute \'related_uris\'')
def purchase_uris(self):
raise AttributeError('Search object has no attribute \'purchase_uris\'')
def life_modifier(self):
raise AttributeError('Search object has no attribute \'life_modifier\'')
def hand_modifier(self):
raise AttributeError('Search object has no attribute \'hand_modifier\'')
def color_indicator(self):
raise AttributeError('Search object has no attribute \'color_indicator\'')
def all_parts(self):
raise AttributeError('Search object has no attribute \'all_parts\'')
def card_faces(self):
raise AttributeError('Search object has no attribute \'card_faces\'')
def watermark(self):
raise AttributeError('Search object has no attribute \'watermark\'')
def story_spotlight_number(self):
raise AttributeError('Search object has no attribute \'story_spotlight_number\'')
def story_spotlight_uri(self):
raise AttributeError('Search object has no attribute \'story_spotlight_uri\'')