class AddJsTestCase(unittest.TestCase):
def setUp(self):
ipdf = PdfFileReader(os.path.join(RESOURCE_ROOT, 'crazyones.pdf'))
self.pdf_file_writer = PdfFileWriter()
self.pdf_file_writer.appendPagesFromReader(ipdf)
def test_add(self):
self.pdf_file_writer.addJS(
"this.print({bUI:true,bSilent:false,bShrinkToFit:true});")
self.assertIn('/Names', self.pdf_file_writer._root_object,
"addJS should add a name catalog in the root object.")
self.assertIn(
'/JavaScript', self.pdf_file_writer._root_object['/Names'],
"addJS should add a JavaScript name tree under the name catalog.")
self.assertIn('/OpenAction', self.pdf_file_writer._root_object,
"addJS should add an OpenAction to the catalog.")
def test_overwrite(self):
self.pdf_file_writer.addJS(
"this.print({bUI:true,bSilent:false,bShrinkToFit:true});")
first_js = self.get_javascript_name()
self.pdf_file_writer.addJS(
"this.print({bUI:true,bSilent:false,bShrinkToFit:true});")
second_js = self.get_javascript_name()
self.assertNotEqual(
first_js, second_js,
"addJS should overwrite the previous script in the catalog.")
def get_javascript_name(self):
self.assertIn('/Names', self.pdf_file_writer._root_object)
self.assertIn('/JavaScript',
self.pdf_file_writer._root_object['/Names'])
self.assertIn(
'/Names',
self.pdf_file_writer._root_object['/Names']['/JavaScript'])
return self.pdf_file_writer._root_object['/Names']['/JavaScript'][
'/Names'][0]
output.addPage(input1.getPage(1).rotateClockwise(90))
# add page 3 from input1, rotated the other way:
output.addPage(input1.getPage(2).rotateCounterClockwise(90))
# alt: output.addPage(input1.getPage(2).rotateClockwise(270))
# add page 4 from input1, but first add a watermark from another PDF:
page4 = input1.getPage(3)
watermark = PdfFileReader(open("watermark.pdf", "rb"))
page4.mergePage(watermark.getPage(0))
output.addPage(page4)
# add page 5 from input1, but crop it to half size:
page5 = input1.getPage(4)
page5.mediaBox.upperRight = (page5.mediaBox.getUpperRight_x() / 2,
page5.mediaBox.getUpperRight_y() / 2)
output.addPage(page5)
# 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
output.addJS("this.print({bUI:true,bSilent:false,bShrinkToFit:true});")
# encrypt your new PDF and add a password
password = "******"
output.encrypt(password)
# finally, write "output" to document-output.pdf
outputStream = file("PyPDF3-output.pdf", "wb")
output.write(outputStream)