File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change 7
7
import typing
8
8
import platform
9
9
10
+ from webdriver_manager .core .driver_cache import DriverCacheManager
10
11
from botcity .web import WebBot , Browser , By , browsers
11
12
from webdriver_manager .chrome import ChromeDriverManager
12
13
from webdriver_manager .firefox import GeckoDriverManager
@@ -97,7 +98,8 @@ def download_driver(request):
97
98
folder_driver = tempfile .mkdtemp ()
98
99
browser = request .config .getoption ("--browser" ) or Browser .CHROME
99
100
manager = factory_driver_manager (browser = browser )
100
- installed_driver = manager (path = folder_driver ).install ()
101
+ cache_manager = DriverCacheManager (root_dir = folder_driver )
102
+ installed_driver = manager (cache_manager = cache_manager ).install ()
101
103
yield installed_driver
102
104
shutil .rmtree (folder_driver )
103
105
You can’t perform that action at this time.
0 commit comments