|
|
|
@ -5,7 +5,7 @@ import zipfile
|
|
|
|
|
from lxml import etree
|
|
|
|
|
import os
|
|
|
|
|
import uploader
|
|
|
|
|
|
|
|
|
|
from iso639 import languages as isoLanguages
|
|
|
|
|
|
|
|
|
|
def extractCover(zip, coverFile, coverpath, tmp_file_name):
|
|
|
|
|
if coverFile is None:
|
|
|
|
@ -41,14 +41,34 @@ def get_epub_info(tmp_file_path, original_file_name, original_file_extension):
|
|
|
|
|
p = tree.xpath('/pkg:package/pkg:metadata', namespaces=ns)[0]
|
|
|
|
|
|
|
|
|
|
epub_metadata = {}
|
|
|
|
|
try:#maybe description isn't present
|
|
|
|
|
comments = tree.xpath("//*[local-name() = 'description']/text()")[0]
|
|
|
|
|
epub_metadata['comments'] = comments
|
|
|
|
|
except:
|
|
|
|
|
epub_metadata['comments'] = ""
|
|
|
|
|
|
|
|
|
|
for s in ['title', 'description', 'creator']:
|
|
|
|
|
tmp = p.xpath('dc:%s/text()' % s, namespaces=ns)
|
|
|
|
|
if len(tmp) > 0:
|
|
|
|
|
epub_metadata[s] = p.xpath('dc:%s/text()' % s, namespaces=ns)[0]
|
|
|
|
|
else:
|
|
|
|
|
epub_metadata[s] = "Unknown"
|
|
|
|
|
#detect lang need futher modification in web.py /upload
|
|
|
|
|
try:#maybe dc:language isn't present, less possible but possible
|
|
|
|
|
lang = p.xpath('dc:language/text()', namespaces=ns)[0]
|
|
|
|
|
lang = lang.split('-', 1)[0]
|
|
|
|
|
lang.lower()
|
|
|
|
|
if len(lang) == 2:
|
|
|
|
|
epub_metadata['languages'] = isoLanguages.get(part1=lang).name
|
|
|
|
|
elif len(lang) == 3:
|
|
|
|
|
epub_metadata['languages'] = isoLanguages.get(part3=lang).name
|
|
|
|
|
else:
|
|
|
|
|
epub_metadata['languages'] = ""
|
|
|
|
|
except:
|
|
|
|
|
epub_metadata['languages'] = ""
|
|
|
|
|
|
|
|
|
|
coversection = tree.xpath("/pkg:package/pkg:manifest/pkg:item[@id='cover-image']/@href", namespaces=ns)
|
|
|
|
|
coverfile = None
|
|
|
|
|
if len(coversection) > 0:
|
|
|
|
|
coverfile = extractCover(zip, coversection[0], coverpath, tmp_file_path)
|
|
|
|
|
else:
|
|
|
|
@ -59,20 +79,18 @@ def get_epub_info(tmp_file_path, original_file_name, original_file_extension):
|
|
|
|
|
markup = zip.read(os.path.join(coverpath,coversection[0]))
|
|
|
|
|
markupTree = etree.fromstring(markup)
|
|
|
|
|
#no matter xhtml or html with no namespace
|
|
|
|
|
imgsrc = markupTree.xpath( "//*[local-name() = 'img']/@src")
|
|
|
|
|
imgsrc = markupTree.xpath("//*[local-name() = 'img']/@src")
|
|
|
|
|
#imgsrc maybe startwith "../"" so fullpath join then relpath to cwd
|
|
|
|
|
filename = os.path.relpath(os.path.join(os.path.dirname(os.path.join(coverpath,coversection[0])),imgsrc[0]))
|
|
|
|
|
filename = os.path.relpath(os.path.join(os.path.dirname(os.path.join(coverpath, coversection[0])), imgsrc[0]))
|
|
|
|
|
coverfile = extractCover(zip, filename, "", tmp_file_path)
|
|
|
|
|
else:
|
|
|
|
|
coverfile = extractCover(zip, coversection[0], coverpath, tmp_file_path)
|
|
|
|
|
else:
|
|
|
|
|
meta_cover = tree.xpath("/pkg:package/pkg:metadata/pkg:meta[@name='cover']/@content", namespaces=ns)
|
|
|
|
|
if len(meta_cover) > 0:
|
|
|
|
|
meta_cover_content = tree.xpath("/pkg:package/pkg:manifest/pkg:item[@id='"+meta_cover[0]+"']/@href",namespaces=ns)
|
|
|
|
|
meta_cover_content = tree.xpath("/pkg:package/pkg:manifest/pkg:item[@id='"+meta_cover[0]+"']/@href", namespaces=ns)
|
|
|
|
|
if len(meta_cover_content) > 0:
|
|
|
|
|
coverfile = extractCover(zip, meta_cover_content[0], coverpath, tmp_file_path)
|
|
|
|
|
else:
|
|
|
|
|
coverfile = None
|
|
|
|
|
|
|
|
|
|
if epub_metadata['title'] is None:
|
|
|
|
|
title = original_file_name
|
|
|
|
@ -88,4 +106,6 @@ def get_epub_info(tmp_file_path, original_file_name, original_file_extension):
|
|
|
|
|
description=epub_metadata['description'],
|
|
|
|
|
tags="",
|
|
|
|
|
series="",
|
|
|
|
|
series_id="")
|
|
|
|
|
series_id="",
|
|
|
|
|
comments=epub_metadata['comments'],
|
|
|
|
|
languages=epub_metadata['languages'])
|
|
|
|
|