digikey: Better webdriver creation.

Making the instantiation of the web driver a bit more generic, starting on
making it possible to try out other browsers.

1 files changed, 19 insertions, 0 deletions

diff --git a/src/ee/_utils.py b/src/ee/_utils.py
index 5960161..0aa12ae 100644
--- a/src/ee/_utils.py
+++ b/src/ee/_utils.py
@@ -79,6 +79,25 @@ def maybe_cache(path: Optional[Path], **kwargs) -> HttpCache:
     return HttpCache(path, **kwargs) if path is not None else EmptyHttpCache()
 
 
+def get_web_driver():  # selenium.webdriver.remote.webdriver.WebDriver
+    import selenium.common.exceptions
+    from selenium import webdriver
+
+    drivers = [
+        ("chrome", webdriver.Chrome),
+        # ("firefox", webdriver.Firefox),
+        # ("webkit", webdriver.WebKitGTK),
+    ]
+
+    for key, constructor in drivers:
+        try:
+            return constructor()
+        except selenium.common.exceptions.WebDriverException:
+            pass
+
+    return webdriver.Chrome()
+
+
 def gen_rst_table(header: List[str], data: List[List[str]]):
     column_widths = []
     for i in range(len(header)):