certificado.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

# Generador de certificados pdf usando una plantilla svg a través de inkscape
# Copyright 2016 David Hernández

# certificado.py is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# certificado.py is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with certificado.py. If not, see <http://www.gnu.org/licenses/>.

import csv
import time
import sys
import traceback
import pdfmerge
from subprocess import Popen

def generar(reemplazos,rol,cedula,nombre,contador):
    """Genera el certificado en formato pdf."""
    tiempo = str(int(time.time()))                      #Para el nombre temporal
    nombretmp = '/tmp/' + tiempo + str(contador) + '.certificado.svg'	#Nombre único temporal del svg modificado

    with open('certificado.svg', 'r') as entrada, open(nombretmp, 'w') as salida:
        for line in entrada:                            #Reemplazo de variables en el archivo svg
            for src, target in reemplazos.iteritems():
                line = line.replace(src, target)
            salida.write(line)
    entrada.close()
    salida.close()

    certsalidat = '/tmp/'+cedula+'-'+rol+'.pdf'         #Nombre de pdf temporal
    certsalida = cedula+'-'+rol+'.pdf'					#Nombre del certificado pdf final

    print(str(contador) + " Generando certificado de " + rol + " para " + nombre)
    x = Popen(['/usr/bin/inkscape', nombretmp, '-A', certsalidat])  #Generación del certificado temporal.

    print("Añadiendo programa al certificado ")
    time.sleep(5)
    pdfmerge.merge([certsalidat, 'programa.pdf'], certsalida)   #Se añade el programa al certificado y se genera el certificado final

    print("Removiendo archivos temporales")
    time.sleep(5)
    x = Popen(['rm', nombretmp])                        #Eliminación de archivos temporales
    x = Popen(['rm', certsalidat])                      #Eliminación de archivos temporales

def main():
    """Recolecta los datos y los envía a la función de generación."""
    try:
        contador = 0
        with open('participantes.csv', 'r') as listado: #Lectura de participantes
            datos = csv.reader(listado, delimiter=';')
            for row in datos:
                if row[0].startswith('#'):              #Permite comentar líneas en el archivo csv
                    continue
                nombre = row[0]                         #Columna 1 corresponden a Nombre y Apellido
                cedula = row[1]							#Columna 2 corresponde a la cédula
                if row[3]=='0':							#Columna 4 corresponde a un código de participación
                    rol = 'ponente'
                if row[3]=='1':
                    rol = 'organizador'
                if row[3]=='2':
                    rol = 'asistente'
        # Variables de sustitución: Nombre, cédula, rol, título del evento (1ra línea), título del evento (2da línea) y fecha
                reemplazos = {'Nombre_Participante':nombre, 'cedula':cedula, 'Rol':'Por su participación como <tspan font-weight = "bold" font-style = "italic">' + rol + '</tspan>', 'Evento':'PyTatuy 2016', 'subtitulo':'4to Día Python de Mérida','Fecha':'Mérida, 12 y 13 de febrero de 2016'}

                contador = contador + 1                 #Contador que se agrega al nombre temporal del svg
                generar(reemplazos,rol,cedula,nombre,contador)  #Función de generación de certificados
        listado.close()
        print("\nTotal de certificados generados: " + str(contador))
    except KeyboardInterrupt:
        print "Interrupción por teclado."
    except Exception:
        traceback.print_exc(file=sys.stdout)
    sys.exit(0)

if __name__ == "__main__":
    main()