Use parsed arguments from console to split PDF

Author: netotz

pdf-splitter/main.py

@@ -1,5 +1,7 @@
 import argparse
 
+from editor import SplitFile, split_pdf
+
 parser = argparse.ArgumentParser()
 parser.add_argument(
     'filename',
@@ -14,7 +16,15 @@
 parser.add_argument(
     '-p', '--page-numbers',
     action='append',
+    type=int,
     nargs=2,
     help='first and last page numbers to split from the input file'
 )
+
 args = parser.parse_args()
+print(args)
+if len(args.page_numbers) != len(args.split_name):
+    raise argparse.ArgumentTypeError('specify the same number of page ranges as the number of splits.')
+
+splits = [SplitFile(name, pages) for name, pages in zip(args.split_name, args.page_numbers)]
+split_pdf(args.filename, splits)