from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.support.ui import Select
import time

periodo = "2025-07"

# Configurar opciones para headless Chromium
options = Options()
options.add_argument("--headless")
options.add_argument("--no-sandbox")
options.add_argument("--disable-dev-shm-usage")

options.add_argument("user-agent=Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36")

# Ruta al chromedriver (ajusta si está en otro lugar)
service = Service(executable_path="/usr/bin/chromedriver")

# Iniciar navegador Chromium
driver = webdriver.Chrome(service=service, options=options)

# Acceder a la página
driver.get("https://www.sbs.gob.pe/app/spp/empleadores/comisiones_spp/Paginas/comision_prima.aspx")
with open("debug.html", "w", encoding="utf-8") as f:
    f.write(driver.page_source)
time.sleep(15)  # Esperar que cargue

with open("debug2.html", "w", encoding="utf-8") as f:
    f.write(driver.page_source)

# Seleccionar periodo
select = Select(driver.find_element("id", "cboPeriodo"))
select.select_by_value(periodo)

# Hacer clic en "Consultar"
driver.find_element("id", "btnConsultar").click()
time.sleep(5)  # Esperar resultados

# Obtener HTML completo
html = driver.page_source
with open("resultado_afp.html", "w", encoding="utf-8") as f:
    f.write(html)

print("✅ HTML guardado en resultado_afp.html")

# Cerrar navegador
driver.quit()