Spaces:
Running
Running
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) |