PyCutter/pdf.py at master · artcom-net/PyCutter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
""" PfdFile class of the PyCutter.

This class implements the manipulation of PDF files.

"""


import sys

from PyPDF2 import PdfFileReader, PdfFileWriter


__author__ = 'Artem Kustov'
__email__ = 'artem.kustov@artcom-net.ru'
__version__ = '1.0'


sys.setrecursionlimit(2000)


class PageNotExistError(Exception):

    def __init__(self, page):
        self.message = 'Page %d not exist' % page

    def __str__(self):
        return self.message


class PdfFile(PdfFileReader):

    def cut_pages(self, pages, cut_type):
        if cut_type == 'range':
            writer = PdfFileWriter()
            start, end = pages
            for page in range(start, end + 1):
                try:
                    writer.addPage(self.getPage(page))
                except IndexError:
                    raise PageNotExistError(page + 1)
            yield writer

        elif cut_type in ('multiple', 'each'):
            for page in pages:
                writer = PdfFileWriter()
                try:
                    writer.addPage(self.getPage(page))
                    yield writer
                except IndexError:
                    raise PageNotExistError(page + 1)