99import click
1010import requests
1111
12+ from requests import Response , HTTPError
13+
1214
1315class URLGenerator :
1416 def __init__ (self , base_url : str ):
@@ -62,7 +64,7 @@ def ingest_tiles(context: click.Context, document_id: Optional[UUID], container:
6264 "filename" : container .name ,
6365 },
6466 )
65- response . raise_for_status ( )
67+ handle_response_errors ( response )
6668 response_data = response .json ()
6769 upload_url = response_data ["upload_url" ]
6870 ingest_id = response_data ["id" ]
@@ -75,7 +77,7 @@ def ingest_tiles(context: click.Context, document_id: Optional[UUID], container:
7577 click .echo ("Processing" )
7678 http .post (process_url )
7779 response = http .get (task_url )
78- response . raise_for_status ( )
80+ handle_response_errors ( response )
7981 response_data = response .json ()
8082
8183 delay = 1
@@ -86,7 +88,7 @@ def ingest_tiles(context: click.Context, document_id: Optional[UUID], container:
8688 if delay < 60 :
8789 delay += 1
8890 response = http .get (task_url )
89- response . raise_for_status ( )
91+ handle_response_errors ( response )
9092 response_data = response .json ()
9193
9294 if response_data ["state" ] == "completed" :
@@ -134,3 +136,11 @@ def upload_file(file: Path, url: str):
134136 chunk = None
135137 else :
136138 response .raise_for_status ()
139+
140+ def handle_response_errors (response : Response ):
141+ try :
142+ response .raise_for_status ()
143+ except HTTPError as error :
144+ print (error )
145+ print (error .response .json ()["errors" ][0 ]["msg" ])
146+ exit ()
0 commit comments