from django.views.decorators.csrf import csrf_exempt
from django.http import JsonResponse
from rest_framework.decorators import action
from django.db import connection
import tempfile
import threading
import time
import shutil
import logging
import zipfile
import os
import csv


@csrf_exempt
@action(detail=False, methods=['post'])
def upload_zip(request):

    if request.method == 'POST':
        filezip = request.FILES['filezip']

        if filezip.closed == False and hasattr(filezip,'temporary_file_path'):
            hilo = threading.Thread(target=procesa_zip, args=(filezip,))
            hilo.start()
            return JsonResponse({
                'ok': True,
                'message': 'Procesando archivo zip'
            })
        
    return JsonResponse({
        'ok': False,
        'message': 'Error en archivo zip enviado'
    })

def procesa_zip(file_upload):
    carpeta_destino = tempfile.gettempdir()
    # shutil.copy(filezip.temporary_file_path(), carpeta_destino)

    # descomprimir archivo zip
    archivo_zip = zipfile.ZipFile(file_upload.temporary_file_path(), "r")
    password=''
    try:
        # Extraer todos los archivos del archivo ZIP
        archivo_zip.extractall(pwd=password, path=carpeta_destino)
        print("Archivos extraídos con éxito", flush=True)
    except Exception as e:
        print("Error al extraer archivos:", e, flush=True)
    finally:
        archivo_zip.close()

    # Lista de archivos GTFS para cargar en la base de datos
    gtfs_files = [
        "agency.txt",
        "calendar.txt",
        "feed_info.txt",
        "routes.txt",
        "shapes.txt",
        "stops.txt",
        "stop_times.txt",
        "trips.txt"
    ]

    # Crear las tablas en el esquema y cargar los datos
    cur = connection.cursor()
    for file in gtfs_files:
        filepath = os.path.join(carpeta_destino, file)
        if os.path.exists(filepath):
            create_and_load_table(filepath, file, cur)
            os.remove(filepath)

    cur.close()
    connection.commit()

    # for i in range(10):
    #     time.sleep(1)
    #     print('.', end='', flush=True)

    print('==============', flush=True)
    print(f'fin proceso archivo: {archivo_zip}', flush=True)



# Función para crear la definición de la tabla como texto y cargar los datos
def create_and_load_table(filepath, file_name, cursor):
    with open(filepath, 'r', encoding='utf-8-sig') as file:
        reader = csv.reader(file)
        columns = next(reader)

        table_name = "z_"+file_name.replace('.txt', '')
        column_definitions = ',\n'.join([f"{column_name} TEXT" for column_name in columns])

        create_table = f"CREATE TABLE IF NOT EXISTS {table_name} (\n{column_definitions}\n);"
        cursor.execute(create_table)
        print(f'SQL> {create_table}', flush=True)
        print('', flush=True)

        truncate_table = f"truncate table {table_name}; commit;"
        cursor.execute(truncate_table)
        print(f'SQL> {truncate_table}', flush=True)
        print('', flush=True)

        file.seek(0)
        next(reader)

        insert_query = f"COPY {table_name} ({', '.join(columns)}) FROM STDIN WITH CSV"
        cursor.copy_expert(sql=insert_query, file=file)
        print(f'SQL> {insert_query}', flush=True)
        print('', flush=True)