Add proxy support

pyinstalive/auth.py

@@ -45,7 +45,7 @@ def onlogin_callback(api, cookie_file):
         logger.separator()
 
 
-def authenticate(username, password, force_use_login_args=False):
+def authenticate(username, password, force_use_login_args=False, proxy=None):
     ig_api = None
     try:
         if force_use_login_args:
@@ -60,7 +60,7 @@ def authenticate(username, password, force_use_login_args=False):
             # login new
             ig_api = Client(
                 username, password,
-                on_login=lambda x: onlogin_callback(x, cookie_file))
+                on_login=lambda x: onlogin_callback(x, cookie_file), proxy=proxy)
         else:
             with open(cookie_file) as file_data:
                 cached_settings = json.load(file_data, object_hook=from_json)
@@ -71,7 +71,7 @@ def authenticate(username, password, force_use_login_args=False):
             try:
                 ig_api = Client(
                     username, password,
-                    settings=cached_settings)
+                    settings=cached_settings, proxy=proxy)
 
             except ClientCookieExpiredError as e:
                 logger.warn('The current cookie file has expired, creating a new one.')
@@ -79,7 +79,7 @@ def authenticate(username, password, force_use_login_args=False):
                 ig_api = Client(
                     username, password,
                     device_id=device_id,
-                    on_login=lambda x: onlogin_callback(x, cookie_file))
+                    on_login=lambda x: onlogin_callback(x, cookie_file), proxy=proxy)
 
     except (ClientLoginError, ClientError) as e:
         logger.separator()

pyinstalive/pil.py

@@ -74,3 +74,4 @@ def initialize():
     clear_temp_files = False
     has_guest = None
     do_heartbeat = False
+    proxy = None

pyinstalive/startup.py

@@ -82,6 +82,7 @@ def validate_inputs(config, args, unknown_args):
         pil.ffmpeg_path = config.get('pyinstalive', 'ffmpeg_path')
         pil.args = args
         pil.config = config
+        pil.proxy = config.get('pyinstalive', 'proxy')
 
         if args.configpath:
             pil.config_path = args.configpath
@@ -268,13 +269,13 @@ def run():
 
     if validate_inputs(config, args, unknown_args):
         if not args.username and not args.password:
-            pil.ig_api = auth.authenticate(username=pil.ig_user, password=pil.ig_pass)
+            pil.ig_api = auth.authenticate(username=pil.ig_user, password=pil.ig_pass, proxy=pil.proxy)
         elif (args.username and not args.password) or (args.password and not args.username):
             logger.warn("Missing --username or --password argument. Falling back to config file.")
             logger.separator()
-            pil.ig_api = auth.authenticate(username=pil.ig_user, password=pil.ig_pass)
+            pil.ig_api = auth.authenticate(username=pil.ig_user, password=pil.ig_pass, proxy=pil.proxy)
         elif args.username and args.password:
-            pil.ig_api = auth.authenticate(username=args.username, password=args.password, force_use_login_args=True)
+            pil.ig_api = auth.authenticate(username=args.username, password=args.password, force_use_login_args=True, proxy=pil.proxy)
 
         if pil.ig_api:
             if pil.dl_user or pil.args.downloadfollowing: