Handle invalid issue IDs.
This commit is contained in:
parent
bebd8e01ba
commit
712bdd262f
1 changed files with 13 additions and 8 deletions
21
readit.py
21
readit.py
|
@ -3,7 +3,7 @@
|
||||||
from selenium import webdriver
|
from selenium import webdriver
|
||||||
from selenium.webdriver.common.keys import Keys
|
from selenium.webdriver.common.keys import Keys
|
||||||
from selenium.common.exceptions import NoSuchElementException
|
from selenium.common.exceptions import NoSuchElementException
|
||||||
import argparse, multiprocessing, urllib.request, urllib.error, urllib.parse, time, os
|
import argparse, multiprocessing, urllib.request, urllib.error, urllib.parse, shutil, time, os
|
||||||
import helpers, helpers.scripts, setuptools
|
import helpers, helpers.scripts, setuptools
|
||||||
|
|
||||||
def status(driver):
|
def status(driver):
|
||||||
|
@ -74,11 +74,11 @@ def canvasHandler(driver, id):
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(e)
|
print(e)
|
||||||
|
|
||||||
print("Sorry, this type of magazine is not supported yet. Please send the output above to fread@kumi.email.")
|
raise Exception("Sorry, this type of magazine is not supported yet. Please send the output above to fread@kumi.email.")
|
||||||
exit(1)
|
|
||||||
|
|
||||||
def hasCanvas(driver, id):
|
def hasCanvas(driver, id):
|
||||||
loadPage("https://app.myreadit.com/reader/%s#1" % id, driver)
|
if not loadPage("https://app.myreadit.com/reader/%s#1" % id, driver):
|
||||||
|
raise ValueError("Issue %s does not exist." % id)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
assert len(driver.find_elements_by_tag_name("canvas")) == 1
|
assert len(driver.find_elements_by_tag_name("canvas")) == 1
|
||||||
|
@ -88,11 +88,16 @@ def hasCanvas(driver, id):
|
||||||
|
|
||||||
def magazineHandler(driver, id, makepdf = True):
|
def magazineHandler(driver, id, makepdf = True):
|
||||||
if helpers.makeDir(id):
|
if helpers.makeDir(id):
|
||||||
loginHandler(driver)
|
try:
|
||||||
page = canvasHandler(driver, id) if hasCanvas(driver, id) else imgHandler(driver, id)
|
loginHandler(driver)
|
||||||
|
page = canvasHandler(driver, id) if hasCanvas(driver, id) else imgHandler(driver, id)
|
||||||
|
|
||||||
if makepdf:
|
if makepdf:
|
||||||
helpers.makePDF(id, page, helpers.JPG)
|
helpers.makePDF(id, page, helpers.JPG)
|
||||||
|
|
||||||
|
except Exception:
|
||||||
|
shutil.rmtree(id)
|
||||||
|
raise
|
||||||
|
|
||||||
else:
|
else:
|
||||||
print("[NOTICE] Skipping issue %s - already exists." % id)
|
print("[NOTICE] Skipping issue %s - already exists." % id)
|
||||||
|
|
Loading…
Reference in a new issue