Skip to content

Commit d4a0beb

Browse files
authored
[Refactor] Update PyPDF2 deprecated usage (#28)
1 parent 3c3c707 commit d4a0beb

File tree

1 file changed

+17
-17
lines changed

1 file changed

+17
-17
lines changed

QR_TestMaker/make_all.py

+17-17
Original file line numberDiff line numberDiff line change
@@ -25,37 +25,37 @@
2525
def make_custom_pdf(blank_test_file,cribsheet_file,overlay_file,custom_file):
2626
print (custom_file)
2727
# 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)
3030
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)
3333
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)
3737
except IndexError:
3838
print("Failed to get cover page for exam")
3939
return False
4040
# 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)
4242
try:
4343
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)
4747
except IndexError:
4848
print("Failed to get page",i,"from exam")
4949
return False
5050
# and attach the cribsheet
5151
if cribsheet_file != "":
5252
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)
5959
except IndexError:
6060
print("Failed to get a page of cribsheet file (should be pages 3 and 4 of)",cribsheet_file)
6161
return False

0 commit comments

Comments
 (0)