kindle manager

mobimaster/mobi/extract.py

@@ -0,0 +1,87 @@
+# -*- 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