From 36014a9bbea65b0d685b9268e9ef3a87bfba9f7a Mon Sep 17 00:00:00 2001 From: Klaus-Uwe Mitterer Date: Mon, 17 Jul 2017 21:11:07 +0200 Subject: [PATCH] Some simplification --- readit.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/readit.py b/readit.py index f102026..38d9d6e 100755 --- a/readit.py +++ b/readit.py @@ -45,11 +45,11 @@ def login(driver): return True def pageHandler(driver, id, page): - if not loadPage("https://app.myreadit.com/reader/%s#%i" % (id, int(page)), driver): - if "/ %i" % (int(page) - 1) in driver.page_source or "/ %i" % (int(page) - 2) in driver.page_source: - return False - while not loadPage("https://app.myreadit.com/reader/%s#%i" % (id, int(page)), driver): - time.sleep(5) + if "/ %i" % (int(page) - 1) in driver.page_source or "/ %i" % (int(page) - 2) in driver.page_source: + return False + + while not loadPage("https://app.myreadit.com/reader/%s#%i" % (id, int(page)), driver): + pass img = driver.find_element_by_id("page1").get_attribute("src") helpers.downloadPage(img, id, page, helpers.JPG)