Added page clearing function

This commit is contained in:
Mark Riedesel 2013-10-16 12:45:02 -05:00
parent 1a5910f1ef
commit 51fe3faf69
2 changed files with 41 additions and 7 deletions

View file

@ -15,11 +15,12 @@ class PopplerController(object):
return poppler.document_new_from_file('file://' + pdf_path, password=None)
def generate_page_images(self, pages=[]):
def generate_page_images(self, pages=[], clear_page=True):
import cairo
import StringIO
page = self.doc.get_page(pages[0] if pages else 0)
width, height = page.get_size()
surface = cairo.ImageSurface(cairo.FORMAT_RGB24, int(width*2), int(height*2))
context = cairo.Context(surface)
@ -27,6 +28,10 @@ class PopplerController(object):
for page_num in pages or xrange(self.doc.get_n_pages()):
page = self.doc.get_page(page_num)
if clear_page:
self.clear_page(page)
page.render(context)
fd = StringIO.StringIO()
surface.write_to_png(fd)
@ -34,15 +39,33 @@ class PopplerController(object):
yield fd
def clear_page(self, page):
for field_mapping in page.get_form_field_mapping():
field = field_mapping.field
field_type = field.get_field_type().value_nick
if field_type == 'text':
field.text_set_text('')
elif field_type == 'button':
field.button_set_state(0)
def get_page_fields(self, page_num):
page = self.doc.get_page(page_num)
for field in page.get_form_field_mapping():
area = field.area
data = field.field
field_type = data.get_field_type()
if field_type.value_nick == 'text':
value = data.text_get_text()
elif field_type.value_nick == 'button':
value = data.button_get_state()
yield {
'name': data.get_name(),
'value': data.text_get_text(),
'value': value,
'type': field_type.value_nick,
'area': {
'pos_x': area.x1,
'pos_y': area.y1,