Add option to specify end in page range

pull/2/head
Vinayak Mehta 2017-08-16 14:53:15 +05:30
parent 861ed0b64e
commit 9753889ea2
1 changed files with 3 additions and 1 deletions

View File

@ -399,13 +399,15 @@ if __name__ == '__main__':
if args['--pages'] == '1':
p.append({'start': 1, 'end': 1})
else:
infile = PdfFileReader(open(filename, 'rb'), strict=False)
if args['--pages'] == 'all':
infile = PdfFileReader(open(filename, 'rb'), strict=False)
p.append({'start': 1, 'end': infile.getNumPages()})
else:
for r in args['--pages'].split(','):
if '-' in r:
a, b = r.split('-')
if b == 'end':
b = infile.getNumPages()
p.append({'start': int(a), 'end': int(b)})
else:
p.append({'start': int(r), 'end': int(r)})