pdf

#!/usr/bin/env python

import sys
from os import path
from pypdf import PdfReader, PdfWriter


def print_info(pdf):
    pdf = PdfReader(pdf)
    info = pdf.metadata
    print(f'Author: {info.author}\n'
          f'Creator: {info.creator}\n'
          f'Producer: {info.producer}\n'
          f'Subject: {info.subject}\n'
          f'Title: {info.title}\n'
          f'Number of pages: {len(pdf.pages)}')


def rotate(angle, pdfin, pdfout):
    reader = PdfReader(pdfin)
    writer = PdfWriter()
    for page in reader.pages:
        writer.add_page(page.rotate(angle))
    with open(pdfout, 'wb') as f:
        writer.write(f)


def split(pdfin, pdfout):
    reader = PdfReader(pdfin)
    name, ext = path.splitext(pdfout)
    for i, page in enumerate(reader.pages):
        writer = PdfWriter()
        writer.add_page(page)
        with open(f'{name}-{i}{ext}', 'wb') as f:
            writer.write(f)


def merge(pdfin, pdfout):
    writer = PdfWriter()
    for pdf in pdfin:
        reader = PdfReader(pdf)
        for page in reader.pages:
            writer.add_page(page)
    with open(pdfout, 'wb') as f:
        writer.write(f)


if __name__ == '__main__':
    action = sys.argv[1]
    if action == 'info':
        print_info(sys.argv[2])
    elif action == 'rotate':
        rotate(int(sys.argv[2]), sys.argv[3], sys.argv[4])
    elif action == 'split':
        split(sys.argv[2], sys.argv[3])
    elif action == 'merge':
        merge(sys.argv[2:-1], sys.argv[-1])
    else:
        raise ValueError(f'unknown action {action}')