X-Git-Url: https://plomlompom.com/repos/?a=blobdiff_plain;f=bookmaker.py;h=e6c8d2aa7a0375452bcd305973f43e9e30ad408f;hb=62bedf672e025c1e8cca6dd04ce75bcbe25fbe26;hp=9e66775f4ae9aaefa9a5d3b2783f865fc51faeff;hpb=be153cb8c016b195168a1dbb736f601e24fe2989;p=misc

diff --git a/bookmaker.py b/bookmaker.py
index 9e66775..e6c8d2a 100755
--- a/bookmaker.py
+++ b/bookmaker.py
@@ -13,6 +13,8 @@ parser.add_argument("-c", "--crop", dest="crop_range", help="crops left, bottom,
 parser.add_argument("-n", "--nup4", dest="nup4", action='store_true', help="puts 4 input pages onto 1 output page")
 parser.add_argument("-a", "--analyze", dest="analyze", action="store_true", help="print lines identifying spine, page borders")
 parser.add_argument("-t", "--symmetry", dest="symmetry", action="store_true", help="alternate horizontal crops between odd and even pages")
+parser.add_argument("-s", "--second", dest="second", help="append second file as input to append")
+parser.add_argument("-r", "--rotate", dest="rotate", help="rotate page of number by 90°")
 args = parser.parse_args()
 
 with open(args.input_file, 'rb') as file:
@@ -33,9 +35,27 @@ with open(args.input_file, 'rb') as file:
         pages_to_add += [page]
         print("read in page number", page_num+1)
 
-    # normalize all pages to A4
-    for page_num in range(start_page, end_page):
-        page = reader.pages[page_num]
+    # add pages of second PDF
+    if args.second:
+        file2 = open(args.second, 'rb')
+        reader2 = pypdf.PdfReader(file2)
+        page_num = 1
+        for page in reader2.pages:
+            pages_to_add += [page]
+            print("read second PDF's page number", page_num)
+            page_num += 1
+
+    # rotate page canvas
+    if args.rotate:
+        page = pages_to_add[int(args.rotate) - 1]
+        page.add_transformation(pypdf.Transformation().translate(tx=-a4_width/2, ty=-a4_height/2))
+        page.add_transformation(pypdf.Transformation().rotate(-90))
+        page.add_transformation(pypdf.Transformation().translate(tx=a4_width/2, ty=a4_height/2))
+
+    # normalize all pages to portrait A4
+    for page in pages_to_add:
+        if "/Rotate" in page:
+            page.rotate(360 - page["/Rotate"])
         page.mediabox.left = 0
         page.mediabox.bottom = 0
         page.mediabox.top = a4_height 
@@ -215,3 +235,5 @@ with open(args.input_file, 'rb') as file:
 
     with open(args.output_file, 'wb') as output_file:
         writer.write(output_file)
+    if args.second:
+        file2.close()