From 51fe3faf6963e30e69076d0d5f290035a1481491 Mon Sep 17 00:00:00 2001
From: Mark Riedesel <mark@klowner.com>
Date: Wed, 16 Oct 2013 12:45:02 -0500
Subject: [PATCH] Added page clearing function

---
 controller.py | 27 +++++++++++++++++++++++++--
 models.py     | 21 ++++++++++++++++-----
 2 files changed, 41 insertions(+), 7 deletions(-)

diff --git a/controller.py b/controller.py
index 7c3a544..a985f8d 100644
--- a/controller.py
+++ b/controller.py
@@ -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,
diff --git a/models.py b/models.py
index cb1ea8d..b388609 100644
--- a/models.py
+++ b/models.py
@@ -9,6 +9,7 @@ from pdfformfiller.controller import PopplerController
 FIELD_TYPE_CHOICES = (
     ('t', 'Text'),
     ('c', 'Checkbox'),
+    ('b', 'Button'),
     )
 
 
@@ -52,9 +53,14 @@ class Page(models.Model):
     def __unicode__(self):
         return self.name
 
-    def process_fields(self):
-        poppler_con = PopplerController(self.document.pdf.path)
-        fields = poppler_con.get_page_fields(self.page_num)
+    @property
+    def poppler(self):
+        if not hasattr(self, '_poppler'):
+            self._poppler = PopplerController(self.document.pdf.path)
+        return self._poppler
+
+    def extract_fields(self):
+        fields = self.poppler.get_page_fields(self.page_num)
 
         for fdata in fields:
             try:
@@ -63,12 +69,12 @@ class Page(models.Model):
                 field = FormField(page=self,
                                   name=fdata['name'])
 
+            field.fieldtype = fdata['type']
             area = fdata['area']
             field.pos_x = area['pos_x']
             field.pos_y = area['pos_y']
             field.width = area['width']
             field.height = area['height']
-            field.fieldtype = 't'
             field.save()
 
 
@@ -76,7 +82,7 @@ class Page(models.Model):
 def page_post_save(sender, **kwargs):
     instance = kwargs.get('instance', None)
     if instance:
-        instance.process_fields()
+        instance.extract_fields()
 
 models.signals.post_save.connect(page_post_save, sender=Page)
 
@@ -95,3 +101,8 @@ class FormField(models.Model):
 
     def page_num(self):
         return self.page.page_num
+
+    @classmethod
+    def create_from_poppler_field(cls, field):
+        return cls()
+