Skip to content

Commit 3429e7f

Browse files
committed
feat: add progress bar
1 parent ee30a5a commit 3429e7f

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

videodb/_utils/_http_client.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def _make_request(
8787

8888
def _handle_request_error(self, e: requests.exceptions.RequestException) -> None:
8989
"""Handle request errors"""
90-
90+
self.show_progress = False
9191
if isinstance(e, requests.exceptions.HTTPError):
9292
try:
9393
error_message = e.response.json().get("message", "Unknown error")
@@ -198,8 +198,11 @@ def get(
198198
self.show_progress = show_progress
199199
return self._make_request(method=self.session.get, path=path, **kwargs)
200200

201-
def post(self, path: str, data=None, **kwargs) -> requests.Response:
201+
def post(
202+
self, path: str, data=None, show_progress: Optional[bool] = False, **kwargs
203+
) -> requests.Response:
202204
"""Make a post request"""
205+
self.show_progress = show_progress
203206
return self._make_request(self.session.post, path, json=data, **kwargs)
204207

205208
def put(self, path: str, data=None, **kwargs) -> requests.Response:

videodb/video.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,7 @@ def index_spoken_words(self, force: bool = False) -> None:
129129
"index_type": IndexType.semantic,
130130
"force": force,
131131
},
132+
show_progress=True,
132133
)
133134

134135
def index_scenes(

0 commit comments

Comments
 (0)