def mangatraders_test_bad_url(self):
dc = DownloadController()
# test chapter request exception
mod = MangatradersModel("http://mangatraders.org", dc.guiInfoFcn)
r = mod.downloadChapter(["bad_chapter", "http://mangatraders.org/manga/asdfgh"], './')
self.assertFalse(r)
def mangatraders_test(self):
dc = DownloadController()
# test chapter list request
mod = MangatradersModel(self.mangatraders_series_url, dc.guiInfoFcn)
chapters = mod.getChaptersList()
# test chapter download
r = mod.downloadChapter(chapters[self.mangatraders_chapter_index], './')
# test if download was sucessfull
self.assertTrue(r)
def setSeriesUrl(self, url):
logger.debug('Set series url: '+url)
try:
if ("bato.to" in url) or ("batoto.com" in url):
self.guiInfoFcn('Detected batoto url')
self.webpage_model = BatotoModel(url, self.guiInfoFcn)
elif "mangafox.me" in url:
self.guiInfoFcn('Detected mangafox.me url')
self.webpage_model = MangafoxModel(url, self.guiInfoFcn)
elif "mangatraders.biz" in url:
self.guiInfoFcn('Detected mangatraders.biz url')
self.webpage_model = MangatradersModel(url, self.guiInfoFcn)
else:
logger.debug('Unsupported url!')
self.guiInfoFcn('Unsupported url!')
self.webpage_model = None
self.chapters = []
return False
except Exception as e:
logger.exception(e)
self.guiInfoFcn('Error when setting series url!', exception=True)
self.webpage_model = None
self.chapters = []
return False
try:
self.chapters = self.webpage_model.getChaptersList()
except Exception as e:
logger.exception(e)
self.guiInfoFcn('Error when downloading list of chapters! wrong url?', exception=True)
self.webpage_model = None
self.chapters = []
return False
return True
class DownloadController():
def __init__(self, gui_info_fcn=None):
self.gui_info_fcn = gui_info_fcn
self.webpage_model = None
self.chapters = []
self.results = []
self.downloadPath = None
self.setDownloadPath('.')
def guiInfoFcn(self, s='Testing printing...', exception=False, downloadProgress=False, trace=[]):
"""
Used to add ability to use keywords to pyqt signals
"""
if type(s) != type("") and type(s) != type(b""):
s = str(s)
if self.gui_info_fcn is None:
if downloadProgress:
s = "Downloading progress: %s " % (s,)
if exception:
logger.exception(s)
for t in trace:
logger.exception(str(t))
else:
print(s)
else:
self.gui_info_fcn(s, exception, downloadProgress, trace)
def setGuiInfoFcn(self, fcn):
self.gui_info_fcn = fcn
def setDownloadPath(self, path):
if os.path.isdir(path):
self.downloadPath = os.path.abspath(path)
return True
else:
return False
def getDownloadPath(self):
return self.downloadPath
def setSeriesUrl(self, url):
logger.debug('Set series url: '+url)
try:
if ("bato.to" in url) or ("batoto.com" in url):
self.guiInfoFcn('Detected batoto url')
self.webpage_model = BatotoModel(url, self.guiInfoFcn)
elif "mangafox.me" in url:
self.guiInfoFcn('Detected mangafox.me url')
self.webpage_model = MangafoxModel(url, self.guiInfoFcn)
elif "mangatraders.biz" in url:
self.guiInfoFcn('Detected mangatraders.biz url')
self.webpage_model = MangatradersModel(url, self.guiInfoFcn)
else:
logger.debug('Unsupported url!')
self.guiInfoFcn('Unsupported url!')
self.webpage_model = None
self.chapters = []
return False
except Exception as e:
logger.exception(e)
self.guiInfoFcn('Error when setting series url!', exception=True)
self.webpage_model = None
self.chapters = []
return False
try:
self.chapters = self.webpage_model.getChaptersList()
except Exception as e:
logger.exception(e)
self.guiInfoFcn('Error when downloading list of chapters! wrong url?', exception=True)
self.webpage_model = None
self.chapters = []
return False
return True
def getChaptersList(self):
return self.chapters
def downloadChapter(self, chapter_list_number):
# list [name, url]
chapter = self.chapters[chapter_list_number]
self.guiInfoFcn('Downloading: '+chapter[0])
logger.debug('Starting download of chapter:'+str(chapter)+', '+self.downloadPath)
try:
r = self.webpage_model.downloadChapter(chapter, self.downloadPath)
except Exception as e:
logger.debug('Failed download of chapter:'+str(chapter)+', '+self.downloadPath)
exc_type, exc_value, exc_traceback = sys.exc_info()
trace = traceback.format_exception(exc_type, exc_value, exc_traceback)
self.guiInfoFcn(e, exception=True, trace=trace)
r = False
if not r:
self.guiInfoFcn('Download finished with errors')
return r
def downloadChapterRange(self, ch_from, ch_to):
results = [False] * len(range(ch_from, ch_to+1))
for i, c_id in enumerate(range(ch_from, ch_to+1)):
# Downloading progress: x/y
self.guiInfoFcn(str(c_id+1-ch_from)+'/'+str(ch_to+1-ch_from), downloadProgress=True)
r = self.downloadChapter(c_id)
results[i] = r
self.results = results
return self.results
