Before diving into the script, make sure you have the required libraries installed. You can do this by running the following commands.<\/p>\n\n\n\n
<\/p>\n\n\n\n
pip install PyPDF2\npip install reportlab<\/code><\/pre>\n\n\n\n<\/p>\n\n\n\n
Understanding the Script<\/strong><\/p>\n\n\n\nThe entire script can be seen at the end. Let’s break down the key components of the script.<\/p>\n\n\n\n
1. create_watermark() Function<\/p>\n\n\n\n
– This function uses the ReportLab library to generate a PDF containing a customizable watermark.<\/p>\n\n\n\n
– You can specify the watermark text, color, transparency, font, and rotation angle.<\/p>\n\n\n\n
2. add_watermark() Function<\/p>\n\n\n\n
– The core function that adds the watermark to each page of the input PDF.<\/p>\n\n\n\n
– It uses PyPDF2 to merge the original PDF with the watermark PDF on each page.<\/p>\n\n\n\n
3. delete_watermark_file() Function<\/p>\n\n\n\n
– A utility function to delete the temporary watermark PDF file after it has been merged with the input files.<\/p>\n\n\n\n
4. Command Line Arguments<\/p>\n\n\n\n
– The script accepts two command line arguments:<\/p>\n\n\n\n
– `–path`: The path to the directory containing the PDF files to watermark.<\/p>\n\n\n\n
– `–watermark_text`: The text to be used as the watermark.<\/p>\n\n\n\n
Here is the entire script.<\/p>\n\n\n\n
import PyPDF2\nimport argparse\nfrom reportlab.pdfgen import canvas\nfrom reportlab.lib.units import inch\nfrom reportlab.lib import colors\nimport os\n\ndef create_watermark(watermark_text, output_pdf):\n pdf = canvas.Canvas(output_pdf)\n pdf.translate(inch, inch) # move the current origin point(0,0) of the canvas by the current given horizontal and vertical distances\n pdf.setFillColor(colors.red, alpha=0.3) # set the font color with alpha value to adjust the transparency of watermark text\n pdf.setFont(\"Helvetica\", 50) # set font and font size\n pdf.rotate(45) # rotate the canvas by 45 degrees\n pdf.drawCentredString(400, 100, watermark_text) # center the watermark text\n pdf.save()\n\ndef add_watermark(input_pdf, output_directory, watermark_pdf):\n base_filename = os.path.splitext(os.path.basename(input_pdf))[0]\n output_pdf = os.path.join(output_directory, f'{base_filename}.pdf')\n with open(input_pdf, 'rb') as file:\n pdf_reader = PyPDF2.PdfReader(file)\n pdf_writer = PyPDF2.PdfWriter()\n\n for page_num in range(len(pdf_reader.pages)):\n page = pdf_reader.pages[page_num]\n watermark_reader = PyPDF2.PdfReader(watermark_pdf)\n watermark_page = watermark_reader.pages[0]\n page.merge_page(watermark_page)\n pdf_writer.add_page(page)\n\n with open(output_pdf, 'wb') as output_file:\n pdf_writer.write(output_file)\n\ndef delete_watermark_file(watermark_pdf):\n if os.path.exists(watermark_pdf):\n os.remove(watermark_pdf)\n \nif __name__ == '__main__':\n parser = argparse.ArgumentParser()\n parser.add_argument(\"--path\", required=True, type=str, help=\"Path of the directory of file lists\")\n parser.add_argument(\"--watermark_text\", required=True, type=str, help=\"Text to be watermark\")\n \n args = parser.parse_args()\n\n path = args.path\n watermark_text = args.watermark_text\n \n print('Processing.....')\n file_list = [file_name for file_name in os.listdir(path) if os.path.isfile(os.path.join(path, file_name))]\n\n output_directory = os.path.join(path, 'output_directory')\n if not os.path.exists(output_directory):\n os.makedirs(output_directory)\n \n watermark_pdf_file = os.path.join(path, 'watermark.pdf')\n create_watermark(watermark_text, watermark_pdf_file)\n for file_name in file_list:\n if not file_name == 'watermark.pdf':\n input_pdf_file = os.path.join(path, file_name)\n add_watermark(input_pdf_file, output_directory, watermark_pdf_file)\n \n delete_watermark_file(watermark_pdf_file)\n print('Done!')<\/code><\/pre>\n\n\n\n<\/p>\n\n\n\n
Running the Script<\/strong><\/p>\n\n\n\nTo run the script, execute the following command.<\/p>\n\n\n\n
<\/p>\n\n\n\n
python script_name.py --path \/path\/to\/pdf\/files --watermark_text \"Your Watermark Text\"<\/code><\/pre>\n\n\n\n<\/p>\n\n\n\n
The script will process each PDF file in the specified directory, add the watermark, and save the watermarked files in a newly created ‘output_directory.’<\/p>\n\n\n\n
Conclusion<\/strong><\/p>\n\n\n\nWith this Python script, you can easily add watermarks to your PDF documents, making them visually distinctive and secure. Whether you’re protecting sensitive information or branding your documents, this solution provides a quick and efficient way to enhance your PDF files. I would recommend to look into the used libraries in details and feel free to customize the script further to suit your specific requirements, such as adjusting colors, fonts, or rotation angles for the watermark.<\/p>\n\n\n\n
Ref: https:\/\/pypdf2.readthedocs.io\/en\/3.0.0\/index.html<\/u><\/a><\/p>\n\n\n\n
![\"\u3053\u306e\u30a8\u30f3\u30c8\u30ea\u30fc\u3092\u306f\u3066\u306a\u30d6\u30c3\u30af\u30de\u30fc\u30af\u306b\u8ffd\u52a0\"](\"\/\/b.hatena.ne.jp\/images\/entry-button\/button-only@2x.png\")
<\/a>