Skip to content

Commit 20ff0ea

Browse files
authored
Add better search errors (#107)
1 parent 785b263 commit 20ff0ea

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

src/pystow/github.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -91,10 +91,12 @@ def search_code(
9191

9292
inner_tqdm = partial(tqdm, disable=not inner_progress, unit="record", leave=False)
9393

94-
initial_response = _search_code_helper(page_size=page_size, page=page, query=query).json()
95-
total = initial_response["total_count"]
94+
initial_res = _search_code_helper(page_size=page_size, page=page, query=query)
95+
initial_res.raise_for_status()
96+
initial_res_json = initial_res.json()
97+
total = initial_res_json["total_count"]
9698
total_pages = 1 + (total // page_size)
97-
yield from inner_tqdm(initial_response["items"], desc="Page 1")
99+
yield from inner_tqdm(initial_res_json["items"], desc="Page 1")
98100

99101
with tqdm(
100102
total=total_pages, unit="page", disable=not progress, desc="Paginating code search results"
@@ -104,10 +106,10 @@ def search_code(
104106
while page_size * page < total:
105107
page += 1
106108
tbar.update(1)
107-
successive_response = _search_code_helper(
108-
page=page, page_size=page_size, query=query
109-
).json()
110-
yield from inner_tqdm(successive_response["items"], desc=f"Page {page}")
109+
res = _search_code_helper(page=page, page_size=page_size, query=query)
110+
res.raise_for_status()
111+
res_json = res.json()
112+
yield from inner_tqdm(res_json["items"], desc=f"Page {page}")
111113

112114

113115
def _search_code_helper(page_size: int, page: int, query: str) -> requests.Response:

0 commit comments

Comments
 (0)