Skip to content

Commit ef5b70b

Browse files
committed
add default exception params, add type check for speech audio
1 parent 656d062 commit ef5b70b

2 files changed

Lines changed: 6 additions & 4 deletions

File tree

auroraapi/api.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,13 +9,13 @@
99

1010
class APIException(Exception):
1111
""" Raise an exception when querying the API """
12-
def __init__(self, id, status, code, type, message):
12+
def __init__(self, id=None, status=None, code=None, type=None, message=None):
1313
self.id = id
1414
self.status = status
1515
self.code = code
1616
self.type = type
1717
self.message = message
18-
super(APIException, self).__init__("[{}] {}".format(code, message))
18+
super(APIException, self).__init__("[{}] {}".format(code if code != None else status, message))
1919

2020
def __repr__(self):
2121
return json.dumps({
@@ -38,8 +38,8 @@ def handle_error(r):
3838
if "application/json" in r.headers["content-type"]:
3939
raise APIException(**r.json())
4040
if r.status_code == 413:
41-
raise APIException(code="RequestEntityTooLarge", message="Request entity too large")
42-
raise APIException(message=r.text)
41+
raise APIException(code="RequestEntityTooLarge", message="Request entity too large", status=413, type="RequestEntityTooLarge")
42+
raise APIException(message=r.text, status=r.status_code)
4343

4444
def get_tts(text):
4545
r = requests.get(TTS_URL, params=[("text", text)], headers=get_headers(), stream=True)

auroraapi/speech.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@ def __init__(self, audio):
2323
:param audio an audio file
2424
:type audio auroraapi.audio.AudioFile
2525
"""
26+
if not isinstance(audio, AudioFile):
27+
raise TypeError("audio must be an instance of auroraapi.audio.AudioFile")
2628
self.audio = audio
2729

2830
def text(self):

0 commit comments

Comments
 (0)