Merge pull request #173 from myd7349/fix-key-error

fix: handle potential KeyError
yihong0618 · Jun 18, 2024 · 9013bcf · 9013bcf
2 parents 7d8a3ab + 2ddeb14
commit 9013bcf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/kindle_download_helper/kindle.py b/kindle_download_helper/kindle.py
@@ -334,7 +334,11 @@ def get_all_books(self, start_index=0, filetype="EBOK"):
             if not result.get("success", True):
                 logger.error("get all books error: %s", result.get("error"))
                 break
-            items = result["OwnershipData"]["items"]
+            try:
+                items = result["OwnershipData"]["items"]
+            except KeyError:
+                logger.error("get all books error: %s", result.get("error"))
+                break
             for item in items:
                 if filetype == "PDOC":
                     item["title"] = html.unescape(item["title"])