zooeytsai 2 years ago
parent
commit
385fb200a4
1 changed files with 3 additions and 8 deletions
  1. 3 8
      web/main.py

+ 3 - 8
web/main.py

@@ -130,18 +130,13 @@ async def func_expand(kw: str = Form(...),kw2:Optional[str] = Form(None),kw3:Opt
 
 
 def restart_browser():
-    os.system('docker container stop p4444')
-    time.sleep(1)
-    os.system('docker container rm p4444')
-    time.sleep(1)
-    os.system('docker run -d -p ' + '4458' + ':4444 --name p4444 --add-host=host.docker.internal:host-gateway --shm-size="2g" selenium/standalone-chrome:103.0')
-    time.sleep(18)
+    global driver
     options = webdriver.ChromeOptions()
     options.add_argument("--no-sandbox")
     options.add_argument("--disable-dev-shm-usage")
     options.add_argument('--headless')
-    ddriver = webdriver.Remote(command_executor='http://127.0.0.1:4458/wd/hub',options=options)
-    driver.set_window_size(950, 20000)
+    s = Service('/root/driver/chromedriver')
+    driver = webdriver.Remote(service=s,options=options)
     return driver
     
 @app.post("/ranking/")