wqxtPDF.py

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
import shutil

from utils import *
import os
import img2pdf
import time
import fitz # pip3 install pymupdf

class wqxtPDF():
	fileExt = ".pdf"
	imgExt  = ".jpeg"
	PDFFolder = "books/PDF";
	PDFTempFolder = "books/temp";
	lists = [];
	filePath = None;
	tempPath = None;


	def __init__( self, bid, name, bookmarks=[], lists=lists ):
		self.bid = bid;
		self.name = name;
		self.lists = lists;
		self.bookmarks = bookmarks;
		mkdir(self.PDFFolder);
		mkdir(self.PDFTempFolder);
		[tempPath, filePath] = self.getFilePath();
		self.tempPath = tempPath
		self.filePath = filePath;

	def getFilePath( self ):
		if self.filePath:
			return self.filePath;
		else:
			curTime = int(time.time());
			PDFFolder	= self.PDFFolder;
			PDFTempFolder	= self.PDFTempFolder;
			bid			= self.bid;
			name 		= self.name;
			fileExt 	= self.fileExt;
			fileName = "{}_{}_{}{}".format( bid, name, curTime, fileExt );
			tempName = "{}_{}_{}_nomarks{}".format( bid, name, curTime, fileExt );
			tempNamePath = "/".join([ PDFTempFolder, tempName ]);
			fileNamePath = "/".join([ PDFFolder, fileName ]);
			return ([ tempNamePath, fileNamePath ]);

	def addPage( self, filename ):
		self.lists.append( filename );

	def addPages( self, filesLists ):
		self.lists += filesLists;

	def generatePDF( self ):
		tempPath = self.tempPath;
		filePath = self.filePath;
		lists = self.lists;
		logging.info("生成pdf文件中");
		pfn_bytes = img2pdf.convert( lists, with_pdfrw=False );
		with open( tempPath, "wb" ) as f:
			f.write(pfn_bytes);
			f.close();
		logging.info("向pdf写入目录中");
		self.addBookMarks();
		logging.info("【pdf路径】：\n{}".format(  os.path.abspath(filePath) ));
		time.sleep(0.1)
		fold = os.getcwd() + "/books/IMG/"+self.bid
		shutil.rmtree(fold)
		print("已删除下载的图片。。。")



	def addBookMarks( self ):
		tempPath = self.tempPath;
		filePath = self.filePath;
		doc = fitz.open( tempPath );
		toc = doc.getToC();
		self.tocAppend( toc, self.bookmarks );
		doc.setToC(toc);
		doc.save( filePath );
		doc.close();
		os.remove( tempPath )

	def tocAppend( self, toc, lists ):
		for chapter in lists:
			level = int(chapter['level']);
			label = chapter['label'];
			pnum = int(chapter['pnum']);
			toc.append([ level, label, pnum ]);
			if "children" in chapter.keys():
				self.tocAppend( toc, chapter['children'] );