From 051f8f63e267be65698f172c6c73e0c97d216380 Mon Sep 17 00:00:00 2001 From: Lorenzo Di Fuccia Date: Fri, 2 Feb 2018 12:14:36 +0100 Subject: [PATCH] Update: bug fix --- safaribooks.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/safaribooks.py b/safaribooks.py index 523bf53..2d98035 100644 --- a/safaribooks.py +++ b/safaribooks.py @@ -282,6 +282,9 @@ class SafariBooks: self.book_chapters = self.get_book_chapters() self.chapters_queue = self.book_chapters[:] + if len(self.book_chapters) > sys.getrecursionlimit(): + sys.setrecursionlimit(len(self.book_chapters)) + self.book_title = self.book_info["title"] self.base_url = self.book_info["web_url"] @@ -439,6 +442,9 @@ class SafariBooks: if "results" not in response or not len(response["results"]): self.display.exit("API: unable to retrieve book chapters.") + if response["count"] > sys.getrecursionlimit(): + sys.setrecursionlimit(response["count"]) + result = [] result.extend([c for c in response["results"] if "cover." in c["filename"]]) for c in result: