jyapayne/safaribooks
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Bug fix for link_replace empty arg

Browse source
This commit is contained in:
Lorenzo Di Fuccia 2018-01-15 11:26:30 +01:00
commit e75ce42e12
1 changed files with 11 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

20
safaribooks.py
View file

@ -475,17 +475,19 @@ class SafariBooks:
return root
def link_replace(self, link):
if link[0] == "/" and ("cover" in link or "images" in link or "graphics" in link
or link[-3:] in ["jpg", "peg", "png", "gif"]):
if link not in self.images:
self.images.append(link)
self.display.log("Crawler: found a new image at %s" % link)
if link:
image = link.split("/")[-1]
return "Images/" + image
if link[0] == "/" and ("cover" in link or "images" in link or "graphics" in link
or link[-3:] in ["jpg", "peg", "png", "gif"]):
if link not in self.images:
self.images.append(link)
self.display.log("Crawler: found a new image at %s" % link)
elif link[0] not in ["/", "h"]:
return link.replace(".html", ".xhtml")
image = link.split("/")[-1]
return "Images/" + image
elif link[0] not in ["/", "h"]:
return link.replace(".html", ".xhtml")
return link