Fix import error and use correct driver for loadwait

brandonrobertz · Apr 12, 2020 · 31dea53 · 31dea53
1 parent 19e251d
commit 31dea53
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/autoscrape/backends/selenium/browser.py b/autoscrape/backends/selenium/browser.py
@@ -223,7 +223,7 @@ def _wait_check(self, driver):
         """
         logger.debug(" - Waiting for page to load (document.readyState)...")
         script = "return document.readyState"
-        result = self._driver_exec(self.driver.execute_script, script)
+        result = driver.execute_script(script)
         return result == "complete"
 
     def _loadwait(self, fn, *args, **kwargs):

diff --git a/autoscrape/backends/selenium/dom.py b/autoscrape/backends/selenium/dom.py
@@ -1,5 +1,6 @@
 # -*- coding: UTF-8 -*-
 import logging
+import re
 
 try:
     from selenium.common.exceptions import (