close mongo connection

2021-12-01 23:44:25 +08:00
parent 481c2f8b6b
commit d61800b755
12 changed files with 21 additions and 3 deletions
--- a/scripts/webdriver.py
+++ b/scripts/webdriver.py
@ -15,7 +15,7 @@ from store.mongo import MongoStore
 def go(url):
    chrome_options = Options()
    chrome_options.add_argument('--proxy-server=%s' % config()['webdriver']['proxy'])
-    # chrome_options.add_argument('--headless')
+    chrome_options.add_argument('--headless')

    proxy = Proxy()
    proxy.proxy_type = ProxyType.MANUAL
@ -52,6 +52,7 @@ def go(url):
            'admin_user_ids': roomInfo['room']['admin_user_ids'],
            'owner': roomInfo['room']['owner']
        })
+        store.close()

        wait.until(presence_of_element_located((By.CLASS_NAME, "oSu9Aw19")))