def test_encrypt(use_128bit):
reader = PdfReader(os.path.join(RESOURCE_ROOT, "form.pdf"))
writer = PdfWriter()
page = reader.pages[0]
orig_text = page.extract_text()
writer.add_page(page)
writer.encrypt(user_pwd="userpwd",
owner_pwd="ownerpwd",
use_128bit=use_128bit)
# write "output" to PyPDF2-output.pdf
tmp_filename = "dont_commit_encrypted.pdf"
with open(tmp_filename, "wb") as output_stream:
writer.write(output_stream)
with open(tmp_filename, "rb") as input_stream:
data = input_stream.read()
assert b"foo" not in data
reader = PdfReader(tmp_filename, password="******")
new_text = reader.pages[0].extract_text()
assert reader.metadata.get("/Producer") == "PyPDF2"
assert new_text == orig_text
# Cleanup
os.remove(tmp_filename)
def test_basic_features():
pdf_path = os.path.join(RESOURCE_ROOT, "crazyones.pdf")
reader = PdfReader(pdf_path)
writer = PdfWriter()
assert len(reader.pages) == 1
# add page 1 from input1 to output document, unchanged
writer.add_page(reader.pages[0])
# add page 2 from input1, but rotated clockwise 90 degrees
writer.add_page(reader.pages[0].rotate(90))
# add page 3 from input1, but first add a watermark from another PDF:
page3 = reader.pages[0]
watermark_pdf = pdf_path
watermark = PdfReader(watermark_pdf)
page3.merge_page(watermark.pages[0])
writer.add_page(page3)
# add page 4 from input1, but crop it to half size:
page4 = reader.pages[0]
page4.mediabox.upper_right = (
page4.mediabox.right / 2,
page4.mediabox.top / 2,
)
writer.add_page(page4)
# add some Javascript to launch the print window on opening this PDF.
# the password dialog may prevent the print dialog from being shown,
# comment the the encription lines, if that's the case, to try this out
writer.add_js("this.print({bUI:true,bSilent:false,bShrinkToFit:true});")
# encrypt your new PDF and add a password
password = "******"
writer.encrypt(password)
# finally, write "output" to PyPDF2-output.pdf
tmp_path = "PyPDF2-output.pdf"
with open(tmp_path, "wb") as output_stream:
writer.write(output_stream)
# cleanup
os.remove(tmp_path)