source: ecoalba/apps/carga/views.py @ 934481f

from django.shortcuts import render
from django.contrib.auth.decorators import login_required
from django.template.context import Context, RequestContext
from django.shortcuts import render_to_response
from django.template import Template,loader
from django.http import HttpResponse
from geraldo import Report
from geraldo.generators import PDFGenerator
from datetime import date
from django.utils import translation
from django.utils.translation import ugettext_lazy as _l, ugettext as _
from settings import PATH, DEBUG
from StringIO import StringIO
import csv

from comun.paginador import Paginador
from comun.models import Pais, CodigoArmonizado, Idioma, Moneda, UnidadMedida, Comercio
from notificaciones.views import verificar_errores_carga_masiva
from carga.forms import CargaMasivaForm, cargarPaises
from carga.tasks import cargar_comercio_exterior, cargar_produccion_nacional

@login_required
def flujo(request):
    paises = cargarPaises(Pais.objects.exclude(anyo_inclusion__exact = None))
    
    form = CargaMasivaForm(paises = paises)
    
    errores, pais, mes, anyo, flujo = verificar_errores_carga_masiva(request)
    
    contexto = {'form': form, 'error_carga_masiva': errores, 'pais': pais, 'numero_mes': mes, 'anyo': anyo, 'flujo': flujo}
    
    return render_to_response('carga/carga_flujo_comercial.html', contexto, context_instance = RequestContext(request))
    #return render_to_response('carga/carga_flujo_comercial.html', {'form': form}, context_instance = RequestContext(request))

@login_required
def cargar_archivo(request):
    paises = cargarPaises(Pais.objects.exclude(anyo_inclusion__exact = None))
    
    if request.method == 'POST':
        form = CargaMasivaForm(data = request.POST, files=request.FILES, paises = paises)
        if form.is_valid():
            usuario = request.user
            pais = form.cleaned_data['pais']
            mes = form.cleaned_data['mes']
            anyo = form.cleaned_data['anyo']
            archivo = request.FILES['archivo'].read()
            flujo_comercial = int(form.cleaned_data['flujo_comercial'])
            
            cur_language = translation.get_language()
            
            if flujo_comercial == 1:
                if DEBUG:
                    cargar_produccion_nacional(usuario, pais, anyo, mes, archivo)
                else:
                    cargar_produccion_nacional.delay(usuario, pais, anyo, mes, archivo)
            elif flujo_comercial == 2:
                if DEBUG:
                    cargar_comercio_exterior(cur_language, usuario, pais, anyo, mes, archivo)
                else:
                    cargar_comercio_exterior.delay(cur_language, usuario, pais, anyo, mes, archivo)
    else:
        form = CargaMasivaForm(paises = paises)
        
    errores, pais, mes, anyo, flujo = verificar_errores_carga_masiva(request)
    
    contexto = {'form': form, 'error_carga_masiva': errores, 'pais': pais, 'numero_mes': mes, 'anyo': anyo, 'flujo': flujo}
            
    return render_to_response('carga/carga_flujo_comercial.html', contexto, context_instance = RequestContext(request))

@login_required
def descargar_archivo(request, iso3166_3, mes, anyo, flujo):
    """
    Vista para generar un archivo CSV para la descarga
    """
    
    flujo_comercial = [2]
    
    if int(flujo) == 2:
        flujo_comercial = [0, 1]
    
    flujos_comerciales = Comercio.objects.filter(pais__iso3166_3=iso3166_3, mes=mes, anyo=anyo, flujo_comercial__in=flujo_comercial)
    
    salida = StringIO()
    #Creamos el archivo csv
    flujo_csv = csv.writer(salida, delimiter=';', quotechar='"')
    
    for flujo_comercial in flujos_comerciales:
        fila_csv = [flujo_comercial.flujo_comercial]
        fila_csv.append(flujo_comercial.codigo_armonizado.codigo)
        fila_csv.append(flujo_comercial.pais_relacion.iso3166_3)
        fila_csv.append(flujo_comercial.unidad_medida.simbolo)
        fila_csv.append(flujo_comercial.cantidad)
        fila_csv.append(flujo_comercial.monto)
        
        flujo_csv.writerow(fila_csv)
    
    nombre_archivo = '%s_%i_%i_%i' % (iso3166_3, int(mes), int(anyo), int(flujo))
    
    # Definimos el tipo de documento que vamos a enviar al usuario
    response = HttpResponse(mimetype='text/csv')
    response['Content-Disposition'] = 'attachment; filename = %s' % nombre_archivo
    response.write(salida.getvalue())
    return response