diff options
Diffstat (limited to 'src/ee/digikey')
| -rw-r--r-- | src/ee/digikey/__init__.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/src/ee/digikey/__init__.py b/src/ee/digikey/__init__.py index cc57771..a9d49cd 100644 --- a/src/ee/digikey/__init__.py +++ b/src/ee/digikey/__init__.py @@ -11,10 +11,10 @@ from typing import List, Optional from urllib.parse import urlparse, parse_qs from lxml import html -from selenium import webdriver +from selenium.webdriver.remote.webdriver import WebDriver import ee._utils -from ee import EeException +from ee import EeException, _utils from ee.money import Money, get_default_context from ee.part import Category from ee.tools import mk_parents @@ -261,7 +261,7 @@ class DigikeyClient(object): self.baseurl = baseurl self.on_download = on_download or self.__nop self.cache = ee._utils.maybe_cache(cache_dir) - self._driver: Optional[webdriver.Chrome] = None + self._driver: Optional[WebDriver] = None def search(self, query: str) -> str: return self.product_search(query) @@ -344,10 +344,9 @@ class DigikeyClient(object): return self._req(url, cache_key, params) @property - def driver(self) -> webdriver.Chrome: + def driver(self) -> WebDriver: if self._driver is None: - options = webdriver.ChromeOptions() - self._driver = webdriver.Chrome(chrome_options=options) + self._driver = _utils.get_web_driver() return self._driver |