django-pdfformfiller/controller.py

57 lines
1.7 KiB
Python

import poppler
import pdfformfiller.models
import pdb
class PopplerController(object):
def __init__(self, pdf=""):
self.doc = poppler.document_new_from_file('file:///home/mark/Bedraga.pdf', password=None)
def load_poppler_doc(self, filename):
from django.conf import settings
pdf_path = os.path.abspath(os.path.join(settings.STATIC_FORMS_ROOT, filename))
if os.path.isfile(pdf_path):
return poppler.document_new_from_file('file://' + pdf_path, password=None)
def generate_page_images(self, pages=[]):
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)
context.scale(2, 2)
for page_num in pages or xrange(self.doc.get_n_pages()):
page = self.doc.get_page(page_num)
page.render(context)
fd = StringIO.StringIO()
surface.write_to_png(fd)
fd.seek(0)
yield fd
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
yield {
'name': data.get_name(),
'value': data.text_get_text(),
'area': {
'pos_x': area.x1,
'pos_y': area.y1,
'width': area.x2 - area.x1,
'height': area.y2 - area.y1,
}
}
class PDFFormFillerPostSave(PopplerController):
def __init__(self, document):
pass