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

FIX-IMAGE make it work with api v1 and v2

Browse source
This commit is contained in:
gudata 2021-09-27 13:32:38 +03:00
commit 2260ffc674
2 changed files with 13 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
.gitignore vendored
View file

@ -11,3 +11,4 @@ Books/
cookies.json
*.log
*.txt
*.code-workspace

15
safaribooks.py
View file

@ -812,10 +812,19 @@ class SafariBooks:
self.chapter_title = next_chapter["title"]
self.filename = next_chapter["filename"]
# Images
asset_base_url = next_chapter['asset_base_url']
api_v2_detected = False
if 'v2' in next_chapter['content']:
asset_base_url = SAFARI_BASE_URL + "/api/v2/epubs/urn:orm:book:{}/files".format(self.book_id)
api_v2_detected = True
if "images" in next_chapter and len(next_chapter["images"]):
self.images.extend(urljoin(next_chapter['content'], img_url)
for img_url in next_chapter['images'])
for img_url in next_chapter['images']:
if api_v2_detected:
self.images.extend([asset_base_url + '/' + img_url])
else:
self.images.extend(urljoin(next_chapter['asset_base_url'], img_url)
# Stylesheets
self.chapter_stylesheets = []