Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

PDF writer cleanups

Browse Source
This commit is contained in:
John Schember 2009-02-12 17:19:38 -05:00
parent 32d9faff6a
commit 2f3562ca20
1 changed files with 23 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
src/calibre/ebooks/pdf/writer.py
View File

@ -7,8 +7,8 @@ __license__ = 'GPL v3'
__copyright__ = '2009, John Schember <john@nachtimwald.com>' __copyright__ = '2009, John Schember <john@nachtimwald.com>'
import os, logging, shutil, sys import os, logging, shutil, sys
from calibre.ebooks.oeb.base import Logger, OEBBook
from calibre.ebooks.oeb.profile import Context from calibre import LoggingInterface
from calibre.ebooks.epub.iterator import SpineItem from calibre.ebooks.epub.iterator import SpineItem
from calibre.ebooks.metadata.opf2 import OPF from calibre.ebooks.metadata.opf2 import OPF
from calibre.ptempfile import PersistentTemporaryDirectory from calibre.ptempfile import PersistentTemporaryDirectory
@ -17,8 +17,7 @@ from calibre.utils.config import Config, StringConfig
from PyQt4 import QtCore from PyQt4 import QtCore
from PyQt4.Qt import QUrl, QEventLoop, SIGNAL, QObject, QApplication, QPrinter, \ from PyQt4.Qt import QUrl, QEventLoop, SIGNAL, QObject, QApplication, QPrinter, \
QMetaObject QMetaObject, Qt
from PyQt4.Qt import *
from PyQt4.QtWebKit import QWebView from PyQt4.QtWebKit import QWebView
from pyPdf import PdfFileWriter, PdfFileReader from pyPdf import PdfFileWriter, PdfFileReader
@ -29,6 +28,8 @@ class PDFWriter(QObject):
QApplication([]) QApplication([])
QObject.__init__(self) QObject.__init__(self)
self.logger = logging.getLogger('oeb2pdf')
self.loop = QEventLoop() self.loop = QEventLoop()
self.view = QWebView() self.view = QWebView()
self.connect(self.view, SIGNAL('loadFinished(bool)'), self._render_html) self.connect(self.view, SIGNAL('loadFinished(bool)'), self._render_html)
@ -36,12 +37,12 @@ class PDFWriter(QObject):
self.combine_queue = [] self.combine_queue = []
self.tmp_path = PersistentTemporaryDirectory('_any2pdf_parts') self.tmp_path = PersistentTemporaryDirectory('_any2pdf_parts')
def dump(self, oeb, oebpath, path): def dump(self, oebpath, path):
self._reset() self._delete_tmpdir()
opf = OPF(oebpath, os.path.dirname(oebpath)) opf = OPF(oebpath, os.path.dirname(oebpath))
self.render_queue = [SpineItem(i.path) for i in opf.spine] self.render_queue = [SpineItem(i.path) for i in opf.spine]
self.combine_queue = []
self.path = path self.path = path
QMetaObject.invokeMethod(self, "_render_book", Qt.QueuedConnection) QMetaObject.invokeMethod(self, "_render_book", Qt.QueuedConnection)
@ -57,28 +58,32 @@ class PDFWriter(QObject):
def _render_next(self): def _render_next(self):
item = str(self.render_queue.pop(0)) item = str(self.render_queue.pop(0))
self.combine_queue.append(os.path.join(self.tmp_path, '%s.pdf' % os.path.basename(item))) self.combine_queue.append(os.path.join(self.tmp_path, '%s.pdf' % os.path.basename(item)))
self.logger.info('Processing %s...' % item)
self.view.load(QUrl(item)) self.view.load(QUrl(item))
def _render_html(self, ok): def _render_html(self, ok):
if ok: if ok:
item_path = os.path.join(self.tmp_path, '%s.pdf' % os.path.basename(str(self.view.url().toLocalFile())))
self.logger.debug('\tRendering item as %s' % item_path)
printer = QPrinter(QPrinter.HighResolution) printer = QPrinter(QPrinter.HighResolution)
printer.setPageMargins(1, 1, 1, 1, QPrinter.Inch) printer.setPageMargins(1, 1, 1, 1, QPrinter.Inch)
printer.setOutputFormat(QPrinter.PdfFormat) printer.setOutputFormat(QPrinter.PdfFormat)
printer.setOutputFileName(os.path.join(self.tmp_path, '%s.pdf' % os.path.basename(str(self.view.url().toLocalFile())))) printer.setOutputFileName(item_path)
self.view.print_(printer) self.view.print_(printer)
self._render_book() self._render_book()
def _reset(self): def _delete_tmpdir(self):
self.render_queue = []
self.combine_queue = []
self.path = ''
if os.path.exists(self.tmp_path): if os.path.exists(self.tmp_path):
shutil.rmtree(self.tmp_path, True) shutil.rmtree(self.tmp_path, True)
self.tmp_path = PersistentTemporaryDirectory('_any2pdf_parts') self.tmp_path = PersistentTemporaryDirectory('_any2pdf_parts')
def _write(self): def _write(self):
print self.path self.logger.info('Combining individual PDF parts...')
try: try:
outPDF = PdfFileWriter() outPDF = PdfFileWriter()
for item in self.combine_queue: for item in self.combine_queue:
@ -89,7 +94,7 @@ class PDFWriter(QObject):
outPDF.write(outputStream) outPDF.write(outputStream)
outputStream.close() outputStream.close()
finally: finally:
self._reset() self._delete_tmpdir()
self.loop.exit(0) self.loop.exit(0)
@ -117,26 +122,18 @@ def option_parser():
return parser return parser
def oeb2pdf(opts, inpath): def oeb2pdf(opts, inpath):
logger = Logger(logging.getLogger('oeb2pdf')) logger = LoggingInterface(logging.getLogger('oeb2pdf'))
logger.setup_cli_handler(opts.verbose) logger.setup_cli_handler(opts.verbose)
outpath = opts.output outpath = opts.output
if outpath is None: if outpath is None:
outpath = os.path.basename(inpath) outpath = os.path.basename(inpath)
outpath = os.path.splitext(outpath)[0] + '.pdf' outpath = os.path.splitext(outpath)[0] + '.pdf'
# source = opts.source_profile
# if source not in Context.PROFILES:
# logger.error(_('Unknown source profile %r') % source)
# return 1
# dest = opts.dest_profile
# if dest not in Context.PROFILES:
# logger.error(_('Unknown destination profile %r') % dest)
# return 1
oeb = OEBBook(inpath, logger=logger, encoding=opts.encoding)
writer = PDFWriter() writer = PDFWriter()
writer.dump(oeb, inpath, outpath) writer.dump(inpath, outpath)
run_plugins_on_postprocess(outpath, 'pdf') run_plugins_on_postprocess(outpath, 'pdf')
logger.info(_('Output written to ') + outpath) logger.log_info(_('Output written to ') + outpath)
def main(argv=sys.argv): def main(argv=sys.argv):
parser = option_parser() parser = option_parser()