add --skip-pages argument

1 files changed, 5 insertions, 0 deletions

diff --git a/main.py b/main.py
index db0c80e..0a2662b 100755
--- a/main.py
+++ b/main.py
@@ -76,11 +76,16 @@ if __name__ == '__main__':
     parser = ArgumentParser()
     parser.add_argument('--uuid', type=str, required=True)
     parser.add_argument('--output-dir', type=str, required=True)
+    parser.add_argument('--skip-pages', type=int, default=0,
+                        help='number of pages to skip')
     args = parser.parse_args()
 
     m = fetch_manifest(args.uuid)
     page = 0
     for c in m['sequences'][0]['canvases']:
+        if args.skip_pages != 0 and page < args.skip_pages:
+            page += 1
+            continue
         download_and_merge_image(c['width'], c['height'], extract_uuid(c['@id']), os.path.join(args.output_dir, f'{page}.jpg'))
         print(f'{page} done')
         page += 1