source: sicp/apps/simulacion/encuesta/views.py @ efbe2a8

# coding=utf-8
"""
Simulador Integral de Cadenas Productivas (SICP)

Copyleft (@) 2015 CENDITEL nodo Mérida - https://miv.cenditel.gob.ve/simulacion/
"""
## @package apps.simulacion.encuesta.views
#
# Funciones a implementar en las vistas del módulo de la Encuesta Industrial
# @author Ing. Erwin Paredes (eparedes at cenditel.gob.ve)
# @author <a href='http://www.cenditel.gob.ve'>Centro Nacional de Desarrollo e Investigación en Tecnologías Libres
# (CENDITEL) nodo Mérida - Venezuela</a>
# @copyright <a href='http://www.gnu.org/licenses/gpl-2.0.html'>GNU Public License versión 2 (GPLv2)</a>
# @date 21-08-2015
# @version 3.0.0

## Muestra el autor del script cuando es invocado
__author__ = "Ing. Erwin Paredes (eparedes at cenditel.gob.ve)"
## Muestra documentación breve sobre el script
__doc__    = "Funciones a implementar en las vistas del módulo de la Encuesta Industrial"

from django.conf import settings
from django.contrib.auth.decorators import login_required
from django.shortcuts import render_to_response
from django.template import RequestContext
from django.core.context_processors import csrf
from django.contrib.messages.views import SuccessMessageMixin
from django.views.generic import FormView
from django.utils.encoding import smart_unicode, force_unicode
from apps.comun.functions import render_cadena
from django.http import HttpResponse
from apps.simulacion.encuesta.models import Encuesta
from datetime import date

import os
import logging
import subprocess
from PIL import Image

def graficar(request):
    """!
    Función que ejecuta la instrucción necesaria para mostrar el correspondiente gráfico de la Encuesta Industrial 
    asociada al escenario en estudio

    @author Ing. Erwin Paredes (eparedes at cenditel.gob.ve)
    @copyright <a href='http://www.gnu.org/licenses/gpl-2.0.html'>GNU Public License versión 2 (GPLv2)</a>
    @date 21-08-2015
    @param request <b>{object}</b> Objeto que contiene la petición
    @return Devuelve el correspondiente response con los datos de la consulta correspondiente al gráfico de la encuesta industrial a mostrar
    """
    rif = request.GET['rif']
    ahora = request.GET.get('ahora')

    existe_encuesta = None

    try:
    
        existe_encuesta = Encuesta.objects.using('encuesta_2006').filter(rif=rif).count()
        salida = os.path.join(settings.BASE_DIR, "tmp/%s_%s.svg" % (str(request.user),ahora))

        narriba = '4'
        nabajo = '4'
        os.system("cd " + os.path.join(settings.BASE_DIR, "tmp/"))
        os.system( os.path.join(settings.BASE_DIR, "apis/chains/print_prod_chain") + " --host=" + settings.DATABASES['encuesta_2006']['HOST'] + " --port=" + settings.DATABASES['encuesta_2006']['PORT'] + " --username=" + settings.DATABASES['encuesta_2006']['USER'] + " --password=" +settings.DATABASES['encuesta_2006']['PASSWORD'] + " --dbname=" + settings.DATABASES['encuesta_2006']['NAME'] + " --levelsup=" + narriba + " --levelsdown=" + nabajo + " --output=" + str(request.user) + "_" + ahora + " RIF " + rif)
        archivo = str(request.user) + "_" + ahora + ".svg "
        os.system("mv -f " + os.path.join(settings.BASE_DIR, archivo) + os.path.join(settings.BASE_DIR, "tmp/")+";"+"chmod -R 755 "+ os.path.join(settings.BASE_DIR, "tmp/"))
        
        cadena = render_cadena(str(request.user))
    except Exception, e:
        print e
    
    return render_to_response("grafico.html",{"username":request.user, "existe_encuesta":existe_encuesta,"rif":rif,'ahora':ahora}, context_instance=RequestContext(request))


def image_as_png_pdf(request):
    """!
    Función que ejecuta la instrucción para convertir un archivo .svg a un formato .png y posteriormente ser mostrado en un archivo .pdf

    @author Ing. Erwin Paredes (eparedes at cenditel.gob.ve)
    @copyright <a href='http://www.gnu.org/licenses/gpl-2.0.html'>GNU Public License versión 2 (GPLv2)</a>
    @date 21-08-2015
    @pre Se requiere tener instalado el comando rsvg-convert
    @param request <b>{object}</b> Objeto que contiene la petición
    @return Devuelve el correspondiente response con el archivo pdf a descargar
    """
    output_format = request.GET.get('format')
    usuario = smart_unicode(request.GET["usuario"], encoding='utf-8', strings_only=False, errors='strict')

    comando = "cd %s;%s %s%s.svg %s %s%s.png"

    # Excepciones que permiten verificar si el comando que convierte archivos svg a png o pdf existe,
    # en caso contrario asigna el comando rsvg-convert
    # Agregado por: Ing. Roldan D. Vargas G.
     # fecha: 31/07/2014
    try:
        command_exists = subprocess.check_output(["rsvg", "--help"], stderr=subprocess.STDOUT)
        convert = "rsvg"
        flag_output = ""
    except subprocess.CalledProcessError:
        convert = "rsvg-convert"
        flag_output = "-o"
    except OSError:
        convert = "rsvg-convert"
        flag_output = "-o"

    os.system(comando % (
        os.path.join(settings.BASE_DIR, "tmp/"), convert, os.path.join(settings.BASE_DIR, "tmp/"), usuario, flag_output,
        os.path.join(settings.BASE_DIR, "tmp/"), usuario
    ))

    archivo = os.path.join(settings.BASE_DIR, "tmp/") + usuario + ".png"
    
    im = Image.open(archivo)  # any Image object should work
    filename = str(request.user)
    if output_format == 'png':
        response = HttpResponse(content_type='image/png')
        response['Content-Disposition'] = 'attachment; filename=%s.png' % filename
        im.save(response, 'png')  # will call response.write()
    else:
        # Temporary disk space, server process needs write access
        tmp_path = '/tmp/'
        # Full path to ImageMagick convert binary
        convert_bin = '/usr/bin/convert'
        im.save(tmp_path + filename + '.png', 'png')
        response = HttpResponse(content_type='application/pdf')
        response['Content-Disposition'] = 'attachment; filename=%s.pdf' % filename
        ret = subprocess.Popen([convert_bin, "%s%s.png" % (tmp_path, filename), "pdf:-"], stdout=subprocess.PIPE)
        response.write(ret.stdout.read())
    return response