Spaces:
Running
Running
import subprocess | |
try: | |
import requests | |
except: | |
subprocess.run(["pip3", "install", "requests"]) | |
import requests | |
import os | |
import zipfile | |
import io | |
url = os.getenv("URL") | |
def process(url): | |
respuesta = requests.get(url) | |
if respuesta.status_code != 200: | |
raise Exception(f"Error al descargar el archivo. Código de estado: {respuesta.status_code}") | |
if 'zip' in respuesta.headers.get('Content-Type', '').lower(): | |
with zipfile.ZipFile(io.BytesIO(respuesta.content)) as archivo_zip: | |
archivo_zip.extractall() | |
print(f"Archivos extraídos en: {os.getcwd()}") | |
else: | |
nombre_archivo = url.split('/')[-1] | |
with open(nombre_archivo, 'wb') as archivo: | |
archivo.write(respuesta.content) | |
print(f"Archivo descargado: {nombre_archivo}") | |
process(url) |