Added field extraction and model creation

models.py

@@ -1,3 +1,97 @@
-from django.db import models
+import os
 
-# Create your models here.
+from django.db import models
+from django.core.files import File
+
+from pdfformfiller.controller import PopplerController
+
+
+FIELD_TYPE_CHOICES = (
+    ('t', 'Text'),
+    ('c', 'Checkbox'),
+    )
+
+
+class Document(models.Model):
+    name = models.CharField(max_length=64)
+    pdf = models.FileField(upload_to='pdfformfiller_pdf', blank=True)
+
+    def __unicode__(self):
+        return self.name
+
+    def process_pages(self):
+        poppler_con = PopplerController(self.pdf.path)
+        for page_num, image in enumerate(poppler_con.generate_page_images()):
+            try:
+                page = self.page_set.get(page_num=page_num)
+            except Page.DoesNotExist:
+                page = Page(document=self,
+                            page_num=page_num,
+                            name=self.name + (' page %d' % page_num))
+            page.image.save(
+                    os.path.basename(self.pdf.path) + ('_page%03d.png' % page_num),
+                    File(image)
+                    )
+
+
+def document_post_save(sender, **kwargs):
+    instance = kwargs.get('instance', None)
+
+    if instance:
+        instance.process_pages()
+
+models.signals.post_save.connect(document_post_save, sender=Document)
+
+
+class Page(models.Model):
+    document = models.ForeignKey('Document')
+    name = models.CharField(max_length=64, blank=True)
+    image = models.ImageField(upload_to='pdfformfiller_page')
+    page_num = models.SmallIntegerField()
+
+    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)
+
+        for fdata in fields:
+            try:
+                field = self.formfield_set.get(name=fdata['name'])
+            except FormField.DoesNotExist:
+                field = FormField(page=self,
+                                  name=fdata['name'])
+
+            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()
+
+
+
+def page_post_save(sender, **kwargs):
+    instance = kwargs.get('instance', None)
+    if instance:
+        instance.process_fields()
+
+models.signals.post_save.connect(page_post_save, sender=Page)
+
+
+class FormField(models.Model):
+    page = models.ForeignKey('Page')
+    name = models.CharField(max_length=255)
+    fieldtype = models.CharField(choices=FIELD_TYPE_CHOICES, max_length=1)
+    pos_x = models.FloatField()
+    pos_y = models.FloatField()
+    width = models.FloatField()
+    height = models.FloatField()
+
+    def __unicode__(self):
+        return self.name
+
+    def page_num(self):
+        return self.page.page_num