Add option to specify end in page range

atlanhq · Aug 16, 2017 · 9753889 · 9753889
1 parent 861ed0b
commit 9753889
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/camelot b/tools/camelot
@@ -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)})