|
25 | 25 | def make_custom_pdf(blank_test_file,cribsheet_file,overlay_file,custom_file):
|
26 | 26 | print (custom_file)
|
27 | 27 | # overlay custom infomation on the cover page
|
28 |
| - output = PyPDF2.PdfFileWriter() |
29 |
| - main_handout = PyPDF2.PdfFileReader(blank_test_file) |
| 28 | + output = PyPDF2.PdfWriter() |
| 29 | + main_handout = PyPDF2.PdfReader(blank_test_file) |
30 | 30 | if cribsheet_file != "":
|
31 |
| - cribsheet = PyPDF2.PdfFileReader(cribsheet_file, strict=False) |
32 |
| - overlay = PyPDF2.PdfFileReader(overlay_file) |
| 31 | + cribsheet = PyPDF2.PdfReader(cribsheet_file, strict=False) |
| 32 | + overlay = PyPDF2.PdfReader(overlay_file, strict=False) |
33 | 33 | try:
|
34 |
| - my_cover = main_handout.getPage(0) |
35 |
| - my_cover.mergePage(overlay.getPage(0)) |
36 |
| - output.addPage(my_cover) |
| 34 | + my_cover = main_handout.pages[0] |
| 35 | + my_cover.merge_page(overlay.pages[0]) |
| 36 | + output.add_page(my_cover) |
37 | 37 | except IndexError:
|
38 | 38 | print("Failed to get cover page for exam")
|
39 | 39 | return False
|
40 | 40 | # for every other page, we just put the name in the upper right corner
|
41 |
| - numpages = main_handout.getNumPages() |
| 41 | + numpages = len(main_handout.pages) |
42 | 42 | try:
|
43 | 43 | for i in range(1,numpages):
|
44 |
| - page = main_handout.getPage(i) |
45 |
| - page.mergePage(overlay.getPage(1)) |
46 |
| - output.addPage(page) |
| 44 | + page = main_handout.pages[i] |
| 45 | + page.merge_page(overlay.pages[1]) |
| 46 | + output.add_page(page) |
47 | 47 | except IndexError:
|
48 | 48 | print("Failed to get page",i,"from exam")
|
49 | 49 | return False
|
50 | 50 | # and attach the cribsheet
|
51 | 51 | if cribsheet_file != "":
|
52 | 52 | try:
|
53 |
| - crib1 = cribsheet.getPage(2) |
54 |
| - crib1.mergePage(overlay.getPage(1)) |
55 |
| - crib2 = cribsheet.getPage(3) |
56 |
| - crib2.mergePage(overlay.getPage(1)) |
57 |
| - output.addPage(crib1) |
58 |
| - output.addPage(crib2) |
| 53 | + crib1 = cribsheet.pages[2] |
| 54 | + crib1.merge_page(overlay.pages[1]) |
| 55 | + crib2 = cribsheet.pages[3] |
| 56 | + crib2.merge_page(overlay.pages[1]) |
| 57 | + output.add_page(crib1) |
| 58 | + output.add_page(crib2) |
59 | 59 | except IndexError:
|
60 | 60 | print("Failed to get a page of cribsheet file (should be pages 3 and 4 of)",cribsheet_file)
|
61 | 61 | return False
|
|
0 commit comments