Added regex for checking book URL

safaribooks.py

@@ -32,8 +32,8 @@
 PROFILE_URL = SAFARI_BASE_URL + "/profile/"
 
 # DEBUG
-USE_PROXY = False
-PROXIES = {"https": "https://127.0.0.1:8080"}
+USE_PROXY = True
+PROXIES = {"https": "http://de001-surf.zone2.proxy.allianz:8080"}
 
 
 class Display:
@@ -317,7 +317,7 @@ def __init__(self, args):
         self.session = requests.Session()
         if USE_PROXY:  # DEBUG
             self.session.proxies = PROXIES
-            self.session.verify = False
+            self.session.verify = True
 
         self.session.headers.update(self.HEADERS)
 
@@ -1110,7 +1110,14 @@ def create_epub(self):
             arguments.error("invalid option: `--no-cookies` is valid only if you use the `--cred` option")
 
     if len(args_parsed.bookid) > 0:
-        bookID = args_parsed.bookid.split("/")[-1]          # Only get book ID from URL
+        bookid_regex = r"['\"]*http[s]?://[a-zA-Z0-9.\-/]+(\d{10,15})/*['\"]*"          # Matches book URL
+        pattern = re.compile(bookid_regex)
+        match = re.search(pattern, args_parsed.bookid)
+        if match:
+            bookID = match.group(1)
+        else:
+            bookID = None
+            arguments.error("Invalid book ID or URL")
         if str.isdecimal(bookID):
             args_parsed.bookid = bookID
         else: