Fixed OSError: [Errno 22] Invalid argument error for book ID

safaribooks.py

@@ -1119,6 +1119,40 @@ def create_epub(self):
         if args_parsed.no_cookies:
             arguments.error("invalid option: `--no-cookies` is valid only if you use the `--cred` option")
 
+    if len(args_parsed.bookid) > 0:
+        book_url_regex = r"['\"]*http[s]?:\/\/[a-zA-Z0-9.\-/]+\/(\d{10,15})\/*['\"]*"       # Matches book URL
+        pattern = re.compile(book_url_regex)
+        matchURL = re.search(pattern, args_parsed.bookid)
+
+        book_id_regex = r"['\"]*(\d{10,15})/*['\"]*"                                   # Matches book ID
+        pattern = re.compile(book_id_regex)
+        matchID = re.search(pattern, args_parsed.bookid)
+
+        if matchURL:
+            bookID = matchURL.group(1)
+        elif matchID:
+            bookID = matchID.group(1)
+        else:
+            bookID = None
+            arguments.error("Invalid book ID or URL")
+        if str.isdecimal(bookID):
+            args_parsed.bookid = bookID
+        else:
+            arguments.error("Invalid book ID")
+    else:
+        arguments.error("Book ID must not be empty")
+
+
+    if len(args_parsed.bookid) > 0:
+        bookID = args_parsed.bookid.split("/")[-1]          # Only get book ID from URL
+        if str.isdecimal(bookID):
+            args_parsed.bookid = bookID
+        else:
+            arguments.error("Invalid book ID")
+    else:
+        arguments.error("Book ID must not be empty")
+
+
     SafariBooks(args_parsed)
     # Hint: do you want to download more then one book once, initialized more than one instance of `SafariBooks`...
     sys.exit(0)