import subprocess
try:
    import requests
except:
    subprocess.run(["pip3", "install", "requests"])
    import requests
import os
import zipfile
import io

# Obtener todas las variables de entorno como un diccionario
variables = os.environ
url = ""
# Imprimir todas las variables de entorno
for key, value in variables.items():
    print(f"{key}: {value}")
    if key == "URL":
        url = value
        break
    print("[INFO] NO SE ENCONTRÓ URL")

def process(url):
    # Descargar el archivo desde la URL
    respuesta = requests.get(url)
    
    # Verificar si la descarga fue exitosa
    if respuesta.status_code != 200:
        raise Exception(f"Error al descargar el archivo. Código de estado: {respuesta.status_code}")
    
    # Verificar si el archivo es un ZIP
    if 'zip' in respuesta.headers.get('Content-Type', '').lower():
        # Crear un objeto BytesIO para manejar el contenido en memoria
        with zipfile.ZipFile(io.BytesIO(respuesta.content)) as archivo_zip:
            # Extraer todos los archivos del ZIP en el directorio actual
            archivo_zip.extractall()
            print(f"Archivos extraídos en: {os.getcwd()}")
    else:
        # Si no es un ZIP, guardar el archivo directamente
        nombre_archivo = url.split('/')[-1]  # Obtener el nombre del archivo desde la URL
        with open(nombre_archivo, 'wb') as archivo:
            archivo.write(respuesta.content)
        print(f"Archivo descargado: {nombre_archivo}")
process(url)