deal with NoSuchElementException

wuyue92tree · Nov 8, 2018 · f61190b · f61190b
1 parent e90e2ba
commit f61190b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/crwy/utils/scrapy_plugs/middlewares.py b/crwy/utils/scrapy_plugs/middlewares.py
@@ -17,7 +17,7 @@
 
 """
 
-import json
+
 import random
 from scrapy.downloadermiddlewares.retry import RetryMiddleware
 from scrapy_redis.connection import get_redis_from_settings

diff --git a/crwy/utils/selenium_api.py b/crwy/utils/selenium_api.py
@@ -25,7 +25,7 @@
     raise CrwyImportException(
         "You should install selenium first! suggestion: pip install "
         "selenium==3.6.0")
-from selenium.common.exceptions import TimeoutException
+from selenium.common.exceptions import TimeoutException, NoSuchElementException
 from selenium.webdriver import DesiredCapabilities, ActionChains
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.webdriver.support.wait import WebDriverWait
@@ -233,7 +233,7 @@ def is_element_visible(self, element):
             the_element = EC.visibility_of_element_located(element)
             assert the_element(driver)
             flag = True
-        except AssertionError:
+        except (AssertionError, NoSuchElementException):
             self.logger.warning('the element is not visible.')
             flag = False
         except Exception as e: