fix: only match books/chapters in playlist

lorenzodifuccia · Apr 19, 2019 · 45758eb · 45758eb
1 parent 27b78ba
commit 45758eb
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/safaribooks.py b/safaribooks.py
@@ -391,8 +391,11 @@ def get_playlist_books(self):
         self.books_dir = os.path.join(PATH, base_dir, self.escape_dirname(pdir))
 
         ourns = [book.get("ourn") for book in response.get("content")]
-        ids = [re.search('urn:orm:book:([^:]*)(?::.+)', ourn).group(1) for ourn in ourns]
-
+        ids = []
+        for ourn in ourns:
+            result = re.search('urn:orm:book:([^:]*)(?::.+)', ourn)
+            if result:
+                ids.append(result.group(1))
         return ids
 
     def return_cookies(self):