65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import re
|
|
import collections
|
|
import ebooklib.epub
|
|
|
|
|
|
def get_book_essentials(filename):
|
|
book = ebooklib.epub.read_epub(filename)
|
|
|
|
# Get book title
|
|
title = book.title.strip()
|
|
|
|
# Get cover image
|
|
# This seems hack-ish, but that's never stopped me before
|
|
image_path = None
|
|
try:
|
|
cover = book.metadata['http://www.idpf.org/2007/opf']['cover'][0][1]['content']
|
|
cover_item = book.get_item_with_id(cover)
|
|
|
|
# In case no cover_item is returned, we search the items
|
|
# in the book and get the first referenced image
|
|
if not cover_item:
|
|
for j in book.guide:
|
|
try:
|
|
if (j['title'].lower in ['cover', 'cover-image', 'coverimage'] or j['type'] == 'coverimagestandard'):
|
|
image_path = j['href']
|
|
break
|
|
except KeyError:
|
|
pass
|
|
|
|
if not image_path:
|
|
for j in book.items:
|
|
if j.media_type == 'application/xhtml+xml':
|
|
_regex = re.search(r"src=\"(.*)\"\/", j.content.decode('utf-8'))
|
|
if _regex:
|
|
image_path = _regex[1]
|
|
break
|
|
|
|
for k in book.get_items_of_type(ebooklib.ITEM_IMAGE):
|
|
if os.path.basename(k.file_name) == os.path.basename(image_path):
|
|
image_content = k.get_content()
|
|
|
|
else:
|
|
image_content = cover_item.get_content()
|
|
|
|
except KeyError:
|
|
print('Cannot parse ' + filename)
|
|
|
|
# Get ISBN ID
|
|
isbn_id = None
|
|
try:
|
|
identifier = book.metadata['http://purl.org/dc/elements/1.1/']['identifier']
|
|
for i in identifier:
|
|
identifier_provider = i[1]['{http://www.idpf.org/2007/opf}scheme']
|
|
if identifier_provider.lower() == 'isbn':
|
|
isbn_id = i[0]
|
|
break
|
|
except KeyError:
|
|
pass
|
|
|
|
with open('/home/akhil/aa.jpg', 'bw') as myimg:
|
|
myimg.write(image_content)
|