DaSE-Computer-Vision-2021
Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.
 
 
 

44 řádky
1.1 KiB

import argparse
import os
import subprocess
try:
from PyPDF2 import PdfFileMerger
MERGE = True
except ImportError:
print("Could not find PyPDF2. Leaving pdf files unmerged.")
MERGE = False
def main(files):
os_args = [
"jupyter",
"nbconvert",
"--log-level",
"CRITICAL",
"--to",
"pdf",
]
for f in files:
os_args.append(f)
subprocess.run(os_args)
print("Created PDF {}.".format(f))
if MERGE:
pdfs = [f.split(".")[0] + ".pdf" for f in files]
merger = PdfFileMerger(strict=True)
for pdf in pdfs:
merger.append(pdf)
merger.write("assignment.pdf")
merger.close()
for pdf in pdfs:
os.remove(pdf)
if __name__ == "__main__":
parser = argparse.ArgumentParser()
# we pass in explicit notebook arg so that we can provide
# an ordered list and produce an ordered pdf
parser.add_argument("--notebooks", type=str, nargs="+", required=True)
args = parser.parse_args()
main(args.notebooks)