diff --git a/pokeapi_ditto/commands/clone.py b/pokeapi_ditto/commands/clone.py index 6b9a688b0..6ac83f7a6 100644 --- a/pokeapi_ditto/commands/clone.py +++ b/pokeapi_ditto/commands/clone.py @@ -60,10 +60,15 @@ def _crawl_index(self) -> List[URL]: def _crawl_resource_list(self, url: URL) -> List[URL]: zero_url = url.with_query({"limit": 0, "offset": 0}) - count = self._crawl(zero_url, save=False)["count"] - full_url = url.with_query({"limit": count, "offset": 0}) - resource_list = self._crawl(full_url) - return [URL(resource_ref["url"]) for resource_ref in resource_list["results"]] + payload = self._crawl(zero_url, save=False) + if "count" in payload: + count = payload["count"] + full_url = url.with_query({"limit": count, "offset": 0}) + resource_list = self._crawl(full_url) + return [URL(resource_ref["url"]) for resource_ref in resource_list["results"]] + else: + self._crawl(url) + return [] def clone_single(self, endpoint_and_id: Tuple[str, str]) -> None: endpoint, id = endpoint_and_id