gcsi-tamayo / install.py
TechDev's picture
Upload 2 files
dca16cf verified
raw
history blame
1.49 kB
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)