Produits Entr'ouvert » Calebasse

# -*- coding: utf-8 -*-

import os

import tempfile

import cairo

import pango

import pangocairo

from ..pdftk import PdfTk

class TransportTemplate(object):

    fields = {

            'NOM_BENEFICIAIRE': {

            'PRENOM_BENEFICIAIRE': {

            'DATE': {

                'size': 7,

            'LIEU': {

                'size': 7,

            'IDENTIFICATION_ETABLISSEMENT': {

                'type': 'multiline',

            'NIR_ASSURE': {

            },

            'NIR_KEY_ASSURE': {

            },

            'NOM_ASSURE': {

            'CODE_ORGANISME_1': {

            },

            'CODE_ORGANISME_2': {

            },

            'CODE_ORGANISME_3': {

            },

            'ADRESSE_BENEFICIAIRE': {

            },

            'SITUATION_CHOICE_2': {

            },

            'SITUATION_CHOICE_3': {

            },

            'SITUATION_CHOICE_4': {

            },

            'SITUATION_DATE': {

            },

            'TRAJET_TEXT': {

            },

            'TRAJET_CHOICE_1': {

            },

            'TRAJET_CHOICE_2': {

            },

            'TRAJET_CHOICE_3': {

            },

            'TRAJET_NUMBER': {

            },

            'PC_CHOICE_1': {

            },

            'PC_CHOICE_2': {

            },

            'MODE_CHOICE_1': {

            },

            'MODE_CHOICE_2': {

            },

            'MODE_CHOICE_3': {

            },

            'MODE_CHOICE_4': {

            },

            'MODE_CHOICE_5': {

            },

            'MODE_CHOICE_6': {

            },

            'CDTS_CHOICE_1': {

            },

            'CDTS_CHOICE_2': {

            },

    def __init__(self, template_path=None, prefix='tmp', suffix=''):

        self.prefix = prefix

        self.suffix = suffix

        self.template_path = template_path

    def draw_field(self, ctx, field, value, size=9):

        _type = field.get('type', 'text')

        size = field.get('size', size)

        if _type == 'bool':

            x, y = field['pos']

            if value:

                ctx.move_to(x, y - 10)

                pangocairo_context = pangocairo.CairoContext(ctx)

                pangocairo_context.set_antialias(cairo.ANTIALIAS_SUBPIXEL)

                layout = pangocairo_context.create_layout()

                font = pango.FontDescription("Georgia %s" % size)

                layout.set_font_description(font)

                layout.set_text(u'\u2714')

                pangocairo_context.update_layout(layout)

                pangocairo_context.show_layout(layout)

        if _type in ('text', 'multiline'):

            x, y = field['pos']

            ctx.move_to(x, y)

            pangocairo_context = pangocairo.CairoContext(ctx)

            pangocairo_context.set_antialias(cairo.ANTIALIAS_SUBPIXEL)

            layout = pangocairo_context.create_layout()

            font = pango.FontDescription("Georgia Bold %s" % size)

            layout.set_font_description(font)

            layout.set_text(unicode(value))

            pangocairo_context.update_layout(layout)

            if field.get('border'):

                a, b, width, height = layout.get_pixel_extents()[1]

                ctx.save()

                ctx.set_source_rgb(1, 1, 1)

                ctx.rectangle(x - 2, y - 2, width + 4, height + 4)

                ctx.fill()

                ctx.set_source_rgb(0, 0, 0)

                ctx.rectangle(x - 2, y - 2, width + 4, height + 4)

                ctx.set_line_width(0.1)

                ctx.stroke()

                ctx.restore()

            ctx.move_to(x, y)

            pangocairo_context.show_layout(layout)

        if _type == 'array':

            field = field.copy()

            y = field['y']

            offset = field.get('x', 0)

            for row in value:

                for x, v in zip(field['cols'], row):

                    sub = {

                            'size': size,

                    }

                    self.draw_field(ctx, sub, v, 7)

                y += field['lineheight']

        ctx.set_source_rgb(0, 0, 0)

        for i in range(0, 827, 100):

            ctx.move_to(i, 5)

            ctx.show_text(str(i // 100));

            ctx.move_to(i, 0)

            ctx.line_to(i, 1169)

            ctx.stroke()

        for i in range(0, 1169, 100):

            ctx.move_to(0, i-5)

            ctx.show_text(str(i // 100))

            ctx.move_to(0, i)

            ctx.line_to(827, i)

            ctx.stroke()

        width, height = 827, 1169

        with tempfile.NamedTemporaryFile(prefix=self.prefix,

                suffix=self.suffix, delete=False) as temp_out_pdf:

            try:

                overlay = tempfile.NamedTemporaryFile(prefix='overlay',

                        suffix='.pdf')

                surface = cairo.PDFSurface(overlay.name, width, height)

                ctx = cairo.Context(surface)

                ctx.set_source_rgb(0, 0, 0)

                for key, value in content.iteritems():

                    field = self.fields[key]

                    self.draw_field(ctx, field, value)

                ctx.show_page()

                surface.finish()

                pdftk = PdfTk()

                pdftk.background(overlay.name, self.template_path,

                    temp_out_pdf.name)

                overlay.close()

                return temp_out_pdf.name

            except:

                if delete:

                    try:

                        os.unlink(temp_out_pdf.name)

                    except:

                        pass

                raise