tanner/wolframalpha
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Rely on inheritance to handle errors in Result and Pod.

Browse Source
This commit is contained in:
Jason R. Coombs 2016-09-03 15:36:52 -04:00
parent f0fc35c7c1
commit 7bee559740
1 changed files with 8 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
wolframalpha/__init__.py
View File

@ -56,16 +56,16 @@ class Client(object):
class ErrorHandler(object):
@staticmethod
def _handle_error(resp):
error = resp.get('error')
if not error:
def __init__(self, *args, **kwargs):
super(ErrorHandler, self).__init__(*args, **kwargs)
self._handle_error()
def _handle_error(self):
if not 'error' in self:
return
code = error['code']
msg = error['msg']
template = 'Error {code}: {msg}'
raise Exception(template.format(code=code, msg=msg))
template = 'Error {error[code]}: {error[msg]}'
raise Exception(template.format(**self))
class Document(dict):
@ -141,9 +141,6 @@ class Pod(ErrorHandler, Document):
numsubpods=int,
subpod=Subpod.from_doc,
)
def __init__(self, *args, **kwargs):
super(Pod, self).__init__(*args, **kwargs)
self._handle_error(self)
@property
def subpods(self):
@ -176,7 +173,6 @@ class Result(ErrorHandler, Document):
def __init__(self, stream):
doc = xmltodict.parse(stream, dict_constructor=dict)['queryresult']
super(Result, self).__init__(doc)
self._handle_error(self)
@property
def info(self):