nemo
/
pystitcher
mirror of https://github.com/captn3m0/pystitcher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add startPage to file list

This commit is contained in:
Nemo 2021-05-26 20:28:15 +05:30
parent db0cffa1a0
commit 0f4d1678db
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/pystitcher/skeleton.py
View File

@ -74,7 +74,6 @@ def main(args):
""" """
args = parse_args(args) args = parse_args(args)
setup_logging(args.loglevel) setup_logging(args.loglevel)
_logger.info("Script ends here")
stitcher = Stitcher(args.input) stitcher = Stitcher(args.input)
stitcher.generate(args.output, args.cleanup) stitcher.generate(args.output, args.cleanup)

19
src/pystitcher/stitcher.py
View File

@ -51,7 +51,7 @@ class Stitcher:
file = element.attrib.get('href') file = element.attrib.get('href')
b = Bookmark(self.currentPage, element.text, self.currentLevel+1) b = Bookmark(self.currentPage, element.text, self.currentLevel+1)
self.currentPage += self._get_pdf_number_of_pages(file) self.currentPage += self._get_pdf_number_of_pages(file)
self.files.append(file) self.files.append((file, self.currentPage))
if b: if b:
self.bookmarks.append(b) self.bookmarks.append(b)
@ -83,11 +83,9 @@ class Stitcher:
for b in self.bookmarks: for b in self.bookmarks:
self._add_bookmark(target, b.title, b.level, b.page) self._add_bookmark(target, b.title, b.level, b.page)
def _generate_concat_command(self, temp_filename):
return ["pdftk"] + self.files + ['cat', 'output', temp_filename]
def _generate_temp_pdf(self, temp_file): def _generate_temp_pdf(self, temp_file):
self._merge(self.files, temp_file) files = [x[0] for x in self.files]
self._merge(files, temp_file)
self._parse_old_bookmarks(temp_file) self._parse_old_bookmarks(temp_file)
def _get_level_from_page_number(self, page): def _get_level_from_page_number(self, page):
@ -130,14 +128,17 @@ class Stitcher:
output.close() output.close()
def generate(self, outputFilename, cleanup = False): def generate(self, outputFilename, cleanup = False):
METADATA_FILENAME = '/tmp/metadata.txt'
tempPdf = tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) tempPdf = tempfile.NamedTemporaryFile(suffix=".pdf", delete=False)
tempMetadataFile = tempfile.NamedTemporaryFile(suffix=".txt", delete=False)
self._generate_temp_pdf(tempPdf) self._generate_temp_pdf(tempPdf)
self._generate_metadata(METADATA_FILENAME) self._generate_metadata(tempMetadataFile.name)
self._update_metadata(tempPdf.name, METADATA_FILENAME, outputFilename) self._update_metadata(tempPdf.name, tempMetadataFile.name, outputFilename)
if (cleanup): if (cleanup):
_logger.info("Deleting temporary files") _logger.info("Deleting temporary files")
os.remove(METADATA_FILENAME) os.remove(tempMetadataFile)
os.remove(tempPdf.name) os.remove(tempPdf.name)
else:
print("Temporary files saved as ", tempPdf.name, tempMetadataFile.name)