Make Scryfall errors easier to catch and handle

This commit is contained in:
Matt Cooper 2019-08-31 06:53:00 -04:00 committed by GitHub
parent 49fd9bd112
commit 20256fa38e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 2 deletions

View File

@ -2,6 +2,12 @@ import aiohttp
import asyncio
import urllib
class ScryfallError(Exception):
def __init__(self, error_obj, *args, **kwargs):
super(self.__class__, self).__init__(*args, **kwargs)
self.error_details = **error_obj
class FoundationObject(object):
def __init__(self, _url, override=False, **kwargs):
@ -28,7 +34,7 @@ class FoundationObject(object):
loop.run_until_complete(main(loop))
if self.scryfallJson['object'] == 'error':
raise Exception(self.scryfallJson['details'])
raise ScryfallError(self.scryfallJson, message=self.scryfallJson['details'])
def _checkForKey(self, key):
"""Checks for a key in the scryfallJson object.
@ -58,4 +64,4 @@ class FoundationObject(object):
KeyError: If key is not found.
"""
if not key in self.scryfallJson[parent][num]:
raise KeyError('This tuple has no key \'{}\''.format(key))
raise KeyError('This tuple has no key \'{}\''.format(key))