Added proxy cmd line arg

README.md

@@ -63,6 +63,7 @@ optional arguments:
                        "[email protected]:password01" `.
   --login              Prompt for credentials used to perform the auth login
                        on Safari Books Online.
+  --proxy PROXY        Add proxy URL and port (e.g. `https://127.0.0.1:8080`)
   --no-cookies         Prevent your session data to be saved into
                        `cookies.json` file.
   --kindle             Add some CSS rules that block overflow on `table` and

safaribooks.py

@@ -31,10 +31,6 @@
 API_ORIGIN_URL = "https://" + API_ORIGIN_HOST
 PROFILE_URL = SAFARI_BASE_URL + "/profile/"
 
-# DEBUG
-USE_PROXY = False
-PROXIES = {"https": "https://127.0.0.1:8080"}
-
 
 class Display:
     BASE_FORMAT = logging.Formatter(
@@ -315,9 +311,9 @@ def __init__(self, args):
         self.display.intro()
 
         self.session = requests.Session()
-        if USE_PROXY:  # DEBUG
-            self.session.proxies = PROXIES
-            self.session.verify = False
+        if args_parsed.proxies:
+            self.session.proxies = args_parsed.proxies
+            # self.session.verify = False
 
         self.session.headers.update(self.HEADERS)
 
@@ -1062,7 +1058,10 @@ def create_epub(self):
         "--login", action='store_true',
         help="Prompt for credentials used to perform the auth login on Safari Books Online."
     )
-
+    arguments.add_argument(
+        "--proxy",
+        help="Add proxy URL and port (e.g. `https://127.0.0.1:8080`)"
+    )
     arguments.add_argument(
         "--no-cookies", dest="no_cookies", action='store_true',
         help="Prevent your session data to be saved into `cookies.json` file."
@@ -1109,6 +1108,20 @@ 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 args_parsed.proxy:
+        proxy_regex = r"http[s]?://[a-zA-Z0-9.-]+:\d{4}"          # Matches proxy URL
+        pattern = re.compile(proxy_regex)
+        match = re.search(pattern, args_parsed.proxy)
+        if match:
+            result = match.group()
+            args_parsed.proxies = {
+                "http": result,
+                "https": result
+            }
+        else:
+            arguments.error(f"Incorrect proxy format (should match the regex: `{proxy_regex}`)")
+
+
     SafariBooks(args_parsed)
     # Hint: do you want to download more then one book once, initialized more than one instance of `SafariBooks`...
     sys.exit(0)