from pypdf import PdfReader
from pypdf import PdfWriter

pdf_writer=PdfWriter()
pdf_reader=PdfReader('/data/demo/servers.pdf')
direction='right'
pages=[0]
for page in pages:
    if direction=='right':
        page_r=pdf_reader.pages[page].rotate(90)
        pdf_writer.add_page(page_r)
    elif direction=='left':
        page_r=pdf_reader.pages[page].rotateCounterClockwise(90)
        pdf_writer.addPage(page_r)
    else:
        print("Encountered an improper argument! Input right or left.")

with open('./xx_rotate.pdf','wb') as fh:
    pdf_writer.write(fh)

name_of_split='./xx_test_'
pdf_path='/data/demo/servers.pdf'
pdf_reader=PdfReader(pdf_path)
for page in range( len(pdf_reader.pages) ):
    pdf_reader=PdfReader(pdf_path)
    pdf_writer=PdfWriter()
    pdf_writer.add_page(pdf_reader.pages[page])
    output=f'{name_of_split}{page}.pdf'
    with open(output,'wb') as output_pdf:
        pdf_writer.write(output_pdf)  

from pypdf import PdfReader
from pypdf import PdfWriter

paths=['/data/demo/test0.pdf','/data/demo/rotate.pdf']
output='/data/demo/merge.pdf'
pdf_writer=PdfWriter()
for path in paths:
    pdf_reader=PdfReader(path)
    for page in range( len(pdf_reader.pages)):
        pdf_writer.add_page(pdf_reader.pages[page])

with open(output,'wb') as out:
    pdf_writer.write(out)

input='/data/demo/servers.pdf'
output='./xx_pwd.pdf'
secret='hello'
pdf_w=PdfWriter()
pdf_r=PdfReader(input)
for page in range( len(pdf_r.pages)):
    pdf_w.add_page(pdf_r.pages[page])

pdf_w.encrypt(user_password=secret,owner_password=secret,use_128bit=True)
with open(output,'wb') as f:
    pdf_w.write(f)