pdf_unspread.py

#!/usr/bin/env python

'''
Original author: pmaupin
https://github.com/pmaupin/pdfrw
with minor modification by Daniel Chin for friendlier command-line calling

usage: py -m pdf_unspread my.pdf

Creates my.unspread.pdf

Chops each page in half, e.g. if a source were
created in booklet form, you could extract individual
pages.
'''

import sys
import os
from pdfrw import PdfReader, PdfWriter, PageMerge
from jdt import Jdt
from interactive import listen
from os import system as cmd

def splitpage(src, ratio = .5):
    ''' Split a page into two (left and right)
    '''
    # Yield a result for each half of the page
    yield PageMerge().add(src, viewrect=(0, 0, ratio, 1)).render()
    yield PageMerge().add(src, viewrect=(ratio, 0, 1-ratio, 1)).render()

if __name__ == '__main__':
    inp = sys.argv[1:]
    if inp == []:
        inpfn = input('path/file.ext = ').strip('"')
    else:
        inpfn, = inp
    basename, ext = os.path.splitext(os.path.basename(inpfn))
    outfn = os.path.join(
        os.path.dirname(inpfn), 
        basename + '.unspread' + ext, 
    )
    writer = PdfWriter(outfn)
    pages = PdfReader(inpfn).pages
    ratio = input('Ratio (default 0.5) = ')
    if ratio == '':
        mySplitpage = splitpage
    else:
        ratio = float(ratio)
        mySplitpage = lambda page : splitpage(page, ratio)
    jdt = Jdt(len(pages))
    for i, page in enumerate(pages):
        writer.addpages(mySplitpage(page))
        jdt.acc()
    writer.write()
    jdt.complete()
    print('start briss? ')
    if listen(['y', 'n']) == b'y':
        cmd('briss.lnk "%s"' % outfn)