Merge branch 'main' of git.phc.dm.unipi.it:phc/dm-scripts
commit
aa6bda62d5
GPG Key ID:
2BE4AB7FDAD828A4
@ -0,0 +1,174 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import io
|
||||||
|
import argparse
|
||||||
|
from PyPDF2 import PageObject, PdfReader, PdfWriter, PaperSize, Transformation
|
||||||
|
|
||||||
|
def main():
|
||||||
|
PROGRAM_NAME="bookletify"
|
||||||
|
DESCRIPTION = """Rearranges the pages of a pdf in order to print it as is,
|
||||||
|
so that you can take it, cut it in half and close it like you would
|
||||||
|
with a book. The result will be the original pdf in the correct order,
|
||||||
|
with the size of half a page. Especially useful to print pdfs to A5
|
||||||
|
size on A4 paper"""
|
||||||
|
|
||||||
|
parser = argparse.ArgumentParser(prog=PROGRAM_NAME, description=DESCRIPTION)
|
||||||
|
parser.add_argument('input',
|
||||||
|
help="""name of the input file to bookletify. Use '-'
|
||||||
|
to read file from stdin""")
|
||||||
|
parser.add_argument('output',
|
||||||
|
help="""name of the output file to write to. Use '-' to
|
||||||
|
write to stdout""")
|
||||||
|
parser.add_argument('-q', '--quiet',
|
||||||
|
action='store_true',
|
||||||
|
help="""suppress stdout. Not suppressed if the flag
|
||||||
|
this flag is absent. Automatically suppressed if
|
||||||
|
output is '-' (see output), to avoid broken
|
||||||
|
pdfs""")
|
||||||
|
parser.add_argument('-s', '--size',
|
||||||
|
action='store',
|
||||||
|
choices=['A0', 'A1', 'A2', 'A3', 'A4', 'A5',
|
||||||
|
'A6', 'A7', 'A8', 'C4'],
|
||||||
|
default='A4',
|
||||||
|
metavar='SIZE',
|
||||||
|
help="""set final paper size. Possible values: 'A0',
|
||||||
|
..., 'A8' or 'C4'. Default value is 'A4'""")
|
||||||
|
parser.add_argument('-b', '--binding-margin',
|
||||||
|
default=8.0,
|
||||||
|
type=float,
|
||||||
|
metavar='MARGIN',
|
||||||
|
help="""internal margin for the binding, expressed in
|
||||||
|
millimeters relative to the final paper size (ie:
|
||||||
|
half of the height of the paper size specified by
|
||||||
|
--size). Set to 0 to disable. This setting is
|
||||||
|
independent to the trim setting, ie: increasing or
|
||||||
|
decreasing this margin does not trim any content on
|
||||||
|
the final pdf, the content gets scaled down (or up)
|
||||||
|
to fit to the final width minus the binding margin.
|
||||||
|
Default value is 8""")
|
||||||
|
parser.add_argument('-t', '--trim',
|
||||||
|
default=[0,0,0,0],
|
||||||
|
type=float,
|
||||||
|
nargs=4,
|
||||||
|
metavar='T',
|
||||||
|
help="""margins to trim, expressed in millimeters
|
||||||
|
relative to the original page size, as: top right
|
||||||
|
bottom left. This setting is independent to the
|
||||||
|
binding margin, ie: first the trim is applied, then
|
||||||
|
the trimmed content gets scaled up (or down) to the
|
||||||
|
full height and width, minus the binding margin.
|
||||||
|
Changing the trim does not increase or decrease the
|
||||||
|
final binding margin. A negative trim adds an empty
|
||||||
|
border on that side. Default value is 0 0 0 0""")
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
if args.output == '-':
|
||||||
|
# If output is written to stdout, suppress other stdout output such as
|
||||||
|
# current state of the program and progress bars
|
||||||
|
args.quiet = True
|
||||||
|
|
||||||
|
info = lambda text='', end='\n': print(text, end=end)
|
||||||
|
if args.quiet:
|
||||||
|
info = lambda text='', end='\n': None
|
||||||
|
|
||||||
|
prog = lambda curr, max: info(f"\r{curr} / {max}", '\n' if curr == max else '')
|
||||||
|
|
||||||
|
reader = None
|
||||||
|
if args.input == '-':
|
||||||
|
reader = PdfReader(sys.stdin.buffer)
|
||||||
|
else:
|
||||||
|
if not os.path.isfile(args.input):
|
||||||
|
print(f'file {args.input} does not exist')
|
||||||
|
exit(1)
|
||||||
|
reader = PdfReader(args.input)
|
||||||
|
|
||||||
|
fullwidth = getattr(PaperSize, args.size).width
|
||||||
|
fullheight = getattr(PaperSize, args.size).height
|
||||||
|
smallwidth = fullheight / 2
|
||||||
|
smallheight = fullwidth
|
||||||
|
# Convert mm to pixels at 72 dpi
|
||||||
|
bindingwidth = args.binding_margin * 72.0 / 25.4
|
||||||
|
smallblank = PageObject.create_blank_page(
|
||||||
|
width=smallwidth - bindingwidth,
|
||||||
|
height=smallheight
|
||||||
|
)
|
||||||
|
|
||||||
|
tt = args.trim[0] * 72.0 / 25.4
|
||||||
|
tr = args.trim[1] * 72.0 / 25.4
|
||||||
|
tb = args.trim[2] * 72.0 / 25.4
|
||||||
|
tl = args.trim[3] * 72.0 / 25.4
|
||||||
|
|
||||||
|
# Read the input
|
||||||
|
info("Reading...")
|
||||||
|
n = len(reader.pages)
|
||||||
|
pages = []
|
||||||
|
for i in range(n):
|
||||||
|
prog(i, n)
|
||||||
|
pages.append(reader.pages[i])
|
||||||
|
prog(n,n)
|
||||||
|
|
||||||
|
while(len(pages) % 4 != 0):
|
||||||
|
pages.append(smallblank)
|
||||||
|
|
||||||
|
|
||||||
|
output = PdfWriter()
|
||||||
|
info("\nRearranging...")
|
||||||
|
|
||||||
|
def get_scaling(page):
|
||||||
|
sx = float(smallwidth - bindingwidth) / (float(page.mediabox.width) - tr - tl)
|
||||||
|
sy = float(smallheight) / (float(page.mediabox.height) - tt - tb)
|
||||||
|
return (sx, sy)
|
||||||
|
|
||||||
|
m = len(pages)
|
||||||
|
for i in range(int(m / 2)):
|
||||||
|
prog(i, int(m / 2))
|
||||||
|
new_page = PageObject.create_blank_page(width=fullwidth, height=fullheight)
|
||||||
|
if i % 2 == 0:
|
||||||
|
(sx, sy) = get_scaling(pages[m-1-i])
|
||||||
|
t = Transformation().scale(sx, sy).rotate(-90).translate(-tb * sy, fullheight + tl * sx)
|
||||||
|
pages[m-1-i].add_transformation(t)
|
||||||
|
pages[m-1-i].cropbox.upper_right = (smallheight, fullheight)
|
||||||
|
pages[m-1-i].cropbox.lower_left = (0, smallwidth + bindingwidth)
|
||||||
|
new_page.merge_page(pages[m-1-i])
|
||||||
|
|
||||||
|
(sx, sy) = get_scaling(pages[i])
|
||||||
|
t = Transformation().scale(sx, sy).rotate(-90).translate(-tb * sy, smallwidth - bindingwidth + tl * sx)
|
||||||
|
pages[i].add_transformation(t)
|
||||||
|
pages[i].cropbox.upper_right = (smallheight, smallwidth - bindingwidth)
|
||||||
|
pages[i].cropbox.lower_left = (0, 0)
|
||||||
|
new_page.merge_page(pages[i])
|
||||||
|
else:
|
||||||
|
(sx, sy) = get_scaling(pages[m-1-i])
|
||||||
|
t = Transformation().scale(sx, sy).rotate(90).translate(smallheight + tb * sy, smallwidth + bindingwidth - tl * sx)
|
||||||
|
pages[m-1-i].add_transformation(t)
|
||||||
|
pages[m-1-i].cropbox.upper_right = (smallheight, fullheight)
|
||||||
|
pages[m-1-i].cropbox.lower_left = (0, smallwidth + bindingwidth)
|
||||||
|
new_page.merge_page(pages[m-1-i])
|
||||||
|
|
||||||
|
(sx, sy) = get_scaling(pages[i])
|
||||||
|
t = Transformation().scale(sx, sy).rotate(90).translate(smallheight + tb * sy, 0 - tl * sx)
|
||||||
|
pages[i].add_transformation(t)
|
||||||
|
pages[i].cropbox.upper_right = (smallheight, smallwidth - bindingwidth)
|
||||||
|
pages[i].cropbox.lower_left = (0, 0)
|
||||||
|
new_page.merge_page(pages[i])
|
||||||
|
|
||||||
|
output.add_page(new_page)
|
||||||
|
prog(int(m/2), int(m/2))
|
||||||
|
|
||||||
|
info("\nSaving...")
|
||||||
|
if args.output == '-':
|
||||||
|
data = io.BytesIO()
|
||||||
|
output.write(data)
|
||||||
|
sys.stdout.buffer.write(data.getvalue())
|
||||||
|
else:
|
||||||
|
output.write(args.output)
|
||||||
|
|
||||||
|
info("\nDone!")
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
|
||||||
|
|
||||||
Loading…
Reference in New Issue