修复了book118下载失败问题

rty813 · Jun 27, 2020 · 0bfd6c1 · 0bfd6c1
1 parent 5efe991
commit 0bfd6c1
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/book118.py b/book118.py
@@ -82,6 +82,11 @@ def download(url):
             actions = ActionChains(driver)
             actions.move_to_element(elem).perform()
             img = elem.find_element_by_tag_name('img')
+            count = 0
+            while count < 10 and img.get_attribute('data-src') == None and img.get_attribute('src') == None:
+                count += 1
+                time.sleep(1)
+
             img_url = img.get_attribute('src')
             if img_url is None or not 'http' in img_url:
                 img_url = "http:" + img.get_attribute('data-src')