88 lines
2.9 KiB
Python
Executable File
88 lines
2.9 KiB
Python
Executable File
# -*- coding: utf-8 -*-
|
|
import shutil
|
|
|
|
from loguru import logger
|
|
import tempfile
|
|
from os.path import basename, splitext, exists, join
|
|
from mobi.kindleunpack import unpackBook
|
|
|
|
def extract(infile):
|
|
"""Extract mobi file and return path to epub file"""
|
|
|
|
tempdir = tempfile.mkdtemp(prefix="mobiex")
|
|
if hasattr(infile, "fileno"):
|
|
tempname = next(tempfile._get_candidate_names()) + ".mobi"
|
|
pos = infile.tell()
|
|
infile.seek(0)
|
|
|
|
with open(join(tempdir, tempname), "wb") as outfile:
|
|
shutil.copyfileobj(infile, outfile)
|
|
|
|
infile.seek(pos)
|
|
infile = join(tempdir, tempname)
|
|
|
|
logger.debug("file: %s" % infile)
|
|
fname_in = basename(infile)
|
|
base, ext = splitext(fname_in)
|
|
fname_out_epub = base + ".epub"
|
|
fname_out_html = "book.html"
|
|
fname_out_pdf = base + ".001.pdf"
|
|
|
|
unpackBook(infile, tempdir, epubver="A")
|
|
|
|
epub_filepath = join(tempdir, "mobi8", fname_out_epub)
|
|
html_filepath = join(tempdir, "mobi7", fname_out_html)
|
|
pdf_filepath = join(tempdir, fname_out_pdf)
|
|
if exists(epub_filepath):
|
|
return tempdir, epub_filepath
|
|
elif exists(html_filepath):
|
|
return tempdir, html_filepath
|
|
elif exists(pdf_filepath):
|
|
return tempdir, pdf_filepath
|
|
raise ValueError("Coud not extract from %s" % infile)
|
|
|
|
def extracttest(infile):
|
|
"""Extract mobi file and return path to epub file"""
|
|
|
|
tempdir = './t/'
|
|
if hasattr(infile, "fileno"):
|
|
tempname = next(tempfile._get_candidate_names()) + ".mobi"
|
|
pos = infile.tell()
|
|
infile.seek(0)
|
|
with open(join(tempdir, tempname), "wb") as outfile:
|
|
shutil.copyfileobj(infile, outfile)
|
|
infile.seek(pos)
|
|
infile = join(tempdir, tempname)
|
|
# tempname 8x2vf7yv.mobi pos 0 infile ./t/8x2vf7yv.mobi
|
|
print('tempname {} pos {} infile {}'.format(tempname,pos,infile))
|
|
|
|
logger.debug("file: %s" % infile)
|
|
fname_in = basename(infile)
|
|
base, ext = splitext(fname_in)
|
|
fname_out_epub = base + ".epub"
|
|
fname_out_html = "book.html"
|
|
fname_out_pdf = base + ".001.pdf"
|
|
|
|
# infile ./t/8x2vf7yv.mobi
|
|
unpackBook(infile, tempdir, epubver="A")
|
|
|
|
epub_filepath = join(tempdir, "mobi8", fname_out_epub)
|
|
html_filepath = join(tempdir, "mobi7", fname_out_html)
|
|
pdf_filepath = join(tempdir, fname_out_pdf)
|
|
|
|
# CGDBG
|
|
#epub_filepath ./t/mobi8/p302tbwb.epub html_filepath ./t/mobi7/book.html pdf_filepath ./t/p302tbwb.001.pdf
|
|
print('epub_filepath {} html_filepath {} pdf_filepath {}'.format( epub_filepath, html_filepath, pdf_filepath))
|
|
|
|
if exists(epub_filepath):
|
|
return tempdir, epub_filepath
|
|
elif exists(html_filepath):
|
|
return tempdir, html_filepath
|
|
elif exists(pdf_filepath):
|
|
return tempdir, pdf_filepath
|
|
raise ValueError("Coud not extract from %s" % infile)
|
|
|
|
if __name__ == "__main__":
|
|
#print(extracttest("../tests/demo.mobi"))
|
|
pass
|