From 8f5a223ed3cdf3644026938924724a0582f178e9 Mon Sep 17 00:00:00 2001
From: ilike2burnthing <59480337+ilike2burnthing@users.noreply.github.com>
Date: Mon, 29 Jul 2024 23:30:25 +0100
Subject: [PATCH] proxy fix

---
 src/utils.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/utils.py b/src/utils.py
index 4b4e879..9dbdb25 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -156,7 +156,7 @@ def get_webdriver(proxy: dict = None) -> ChromiumPage:
     proxy_extension_dir = None
     if proxy and all(key in proxy for key in ['url', 'username', 'password']):
         proxy_extension_dir = create_proxy_extension(proxy)
-        options.set_argument("--load-extension=%s" % os.path.abspath(proxy_extension_dir))
+        options.add_extension(proxy_extension_dir)
     elif proxy and 'url' in proxy:
         proxy_url = proxy['url']
         logging.debug("Using webdriver proxy: %s", proxy_url)
@@ -174,12 +174,12 @@ def get_webdriver(proxy: dict = None) -> ChromiumPage:
     if CHROME_EXE_PATH is not None:
         options.set_paths(browser_path=CHROME_EXE_PATH)
 
+    driver = ChromiumPage(addr_or_opts=options)
+
     # clean up proxy extension directory
     if proxy_extension_dir is not None:
         shutil.rmtree(proxy_extension_dir)
 
-    driver = ChromiumPage(addr_or_opts=options)
-
     return driver