diff options
Diffstat (limited to 'parsedjvutext.py')
| -rw-r--r-- | parsedjvutext.py | 5 |
1 files changed, 4 insertions, 1 deletions
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 |
