add --skip-pages argument

This commit is contained in:
Evgeny Sorokin 2024-06-26 01:31:28 +03:00
parent 2db780e200
commit 42295b7b0a

View File

@ -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