From 0d583ea5f9873a5b2a22a89bbb979bf08dd05a90 Mon Sep 17 00:00:00 2001
From: Thibaut Horel <thibaut.horel@gmail.com>
Date: Thu, 27 Feb 2014 12:27:42 -0500
Subject: Add the possibility to specify list of pages to parse_book

---
 parsedjvutext.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'parsedjvutext.py')

diff --git a/parsedjvutext.py b/parsedjvutext.py
index 44340cc..9855786 100644
--- a/parsedjvutext.py
+++ b/parsedjvutext.py
@@ -4,6 +4,7 @@ import subprocess
 import djvu
 from djvu.decode import Context
 from itertools import chain
+import collections
 
 
 def parse_book_xml(djvubook):
@@ -64,8 +65,10 @@ def parse_book(djvubook, page=None, html=False):
     c = Context()
     document = c.new_document(djvu.decode.FileURI(djvubook))
     document.decoding_job.wait()
-    if page:
+    if type(page) is int:
         toparse = [document.pages[page - 1]]
+    elif isinstance(page, collections.Iterable):
+        toparse = page
     else:
         toparse = document.pages
 
-- 
cgit v1.2.3-70-g09d2