You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
3.6 KiB
83 lines
3.6 KiB
2 years ago
|
import sys
|
||
|
|
import subprocess
|
||
|
|
import re
|
||
|
|
import os
|
||
|
|
|
||
|
|
source_dir = sys.argv[1]
|
||
|
|
build_dir = sys.argv[2]
|
||
|
|
insert_toc_after_page = 1
|
||
|
|
|
||
|
|
index_pdf_filename = build_dir + '/weasyprint/index.pdf'
|
||
|
|
css_filename = source_dir + '/css/print-theme.css'
|
||
|
|
|
||
|
|
script_dir = os.path.dirname(__file__)
|
||
|
|
|
||
|
|
# Compile PDF
|
||
|
|
assert(subprocess.run(['weasyprint', build_dir + '/weasyprint/index.html', index_pdf_filename, '-s', css_filename]).returncode == 0)
|
||
|
|
|
||
|
|
# Generate table of content (TOC)
|
||
|
|
assert(subprocess.run(['sh', '-c', script_dir + '/../pdftoc-to-latex "' + index_pdf_filename + '" > "' + build_dir + '/weasyprint/toc.tex"']).returncode == 0)
|
||
|
|
assert(subprocess.run(['pdflatex', '-interaction', 'nonstopmode', '-output-directory=' + build_dir + '/weasyprint', build_dir + '/weasyprint/toc.tex']).returncode == 1)
|
||
|
|
|
||
|
|
# Count TOC pages
|
||
|
|
toc_pdfinfo = subprocess.run(['pdfinfo', build_dir + '/weasyprint/toc.pdf'], stdout=subprocess.PIPE)
|
||
|
|
assert(toc_pdfinfo.returncode == 0)
|
||
|
|
toc_pages_match = re.search('\\nPages:\s+([0-9]+)\\n', toc_pdfinfo.stdout.decode())
|
||
|
|
toc_num_pages = int(toc_pages_match.group(1))
|
||
|
|
print('toc.pdf: ' + str(toc_num_pages) + ' page(s)')
|
||
|
|
|
||
|
|
# Count pages in index.pdf
|
||
|
|
pdfinfo = subprocess.run(['pdfinfo', index_pdf_filename], stdout=subprocess.PIPE)
|
||
|
|
assert(pdfinfo.returncode == 0)
|
||
|
|
pages_match = re.search('\\nPages:\s+([0-9]+)\\n', pdfinfo.stdout.decode())
|
||
|
|
num_pages = int(pages_match.group(1))
|
||
|
|
print('index.pdf: ' + str(num_pages) + ' pages')
|
||
|
|
|
||
|
|
num_pages = num_pages + toc_num_pages # account for table of content that will be added later
|
||
|
|
|
||
|
|
# If needed, update the CSS file with the correct number of pages
|
||
|
|
with open(css_filename) as css_file:
|
||
|
|
original_css = css_file.read()
|
||
|
|
|
||
|
|
modified_css = re.sub('content: counter\(page\) "/[0-9]+";', 'content: counter(page) "/'+str(num_pages)+'";', original_css)
|
||
|
|
|
||
|
|
if modified_css != original_css:
|
||
|
|
with open(css_filename, 'w') as css_file:
|
||
|
|
css_file.write(modified_css)
|
||
|
|
|
||
|
|
# We need to compile again with the modified CSS (this won't impact the TOC)
|
||
|
|
print('Number of pages has changed, rebuilding PDF...')
|
||
|
|
assert(subprocess.run(['weasyprint', build_dir + '/weasyprint/index.html', index_pdf_filename, '-s', css_filename]).returncode == 0)
|
||
|
|
|
||
|
|
# Insert TOC in the PDF
|
||
|
|
assert(subprocess.run(['pdftk', 'A='+index_pdf_filename, 'B='+build_dir+'/weasyprint/toc.pdf', 'cat', 'A'+str(insert_toc_after_page), 'B', 'A'+str(insert_toc_after_page+1)+'-end', 'output', build_dir + '/weasyprint/vheliotech-without-bookmarks.pdf']).returncode == 0)
|
||
|
|
|
||
|
|
# Restore bookmarks
|
||
|
|
extract_bookmarks_from = index_pdf_filename
|
||
|
|
source_pdf_filename = build_dir + '/weasyprint/vheliotech-without-bookmarks.pdf'
|
||
|
|
output_filename = build_dir + '/weasyprint/vheliotech.pdf'
|
||
|
|
|
||
|
|
bookmarks_filename = extract_bookmarks_from.replace('.pdf', '.txt')
|
||
|
|
assert(bookmarks_filename != extract_bookmarks_from)
|
||
|
|
|
||
|
|
# extract PDF metadata into a text file
|
||
|
|
assert(subprocess.run(['pdftk', extract_bookmarks_from, 'dump_data', 'output', bookmarks_filename]).returncode == 0)
|
||
|
|
|
||
|
|
with open(bookmarks_filename) as bookmarks_file:
|
||
|
|
metadata = bookmarks_file.read()
|
||
|
|
|
||
|
|
# Offset page numbers of bookmarks
|
||
|
|
def replaceBookmarkPageNumber(match):
|
||
|
|
initial_page = int(match.group(1))
|
||
|
|
final_page = initial_page + toc_num_pages if initial_page > insert_toc_after_page else initial_page
|
||
|
|
return 'BookmarkPageNumber: ' + str(final_page)
|
||
|
|
metadata = re.sub('BookmarkPageNumber:\s+([0-9]+)', replaceBookmarkPageNumber, metadata)
|
||
|
|
|
||
|
|
with open(bookmarks_filename, 'w') as bookmarks_file:
|
||
|
|
bookmarks_file.write(metadata)
|
||
|
|
|
||
|
|
# generate the output PDF
|
||
|
|
assert(subprocess.run(['pdftk', source_pdf_filename, 'update_info', bookmarks_filename, 'output', output_filename]).returncode == 0)
|
||
|
|
|
||
|
|
print('Generated file: ' + output_filename)
|