Spaces:
Running
Running
File size: 1,487 Bytes
73641ae 478f39c 73641ae dca16cf 73641ae |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
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) |