# encrypt_pdf.py
import sys
from PyPDF2 import PdfReader, PdfWriter

def encrypt_pdf(input_file, output_file, password):
    reader = PdfReader(input_file)
    writer = PdfWriter()

    for page in range(len(reader.pages)):
        writer.add_page(reader.pages[page])

    writer.encrypt(user_pwd=password, owner_pwd=password, use_128bit=True)

    with open(output_file, 'wb') as f:
        writer.write(f)

if __name__ == "__main__":
    input_file = sys.argv[1]
    output_file = sys.argv[2]
    password = sys.argv[3]

    encrypt_pdf(input_file, output_file, password)