Update extract_m3u8.py

extract_m3u8.py

@@ -1,11 +1,14 @@
 import time
 import json
 import re
+import os
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.chrome.service import Service
+from webdriver_manager.chrome import ChromeDriverManager
+from webdriver_manager.core.os_manager import ChromeType
 
-def extract_m3u8_urls(url, wait_time=10, headers=None, chromedriver_path="./chromedriver"):
+def extract_m3u8_urls(url, wait_time=10, headers=None):
     """
     Captura URLs m3u8 de una página web y las devuelve directamente.
     
@@ -13,7 +16,6 @@ def extract_m3u8_urls(url, wait_time=10, headers=None, chromedriver_path="./chro
         url (str): URL de la página a analizar
         wait_time (int): Tiempo de espera en segundos
         headers (dict): Headers personalizados
-        chromedriver_path (str): Ruta al ejecutable de ChromeDriver
         
     Returns:
         list: Lista de URLs m3u8 encontradas
@@ -37,8 +39,12 @@ def extract_m3u8_urls(url, wait_time=10, headers=None, chromedriver_path="./chro
     # Habilitar registro de red
     chrome_options.set_capability("goog:loggingPrefs", {"performance": "ALL"})
     
-    # Iniciar navegador con ruta específica al ChromeDriver
-    service = Service(executable_path=chromedriver_path)
+    # Crear un directorio de caché en el directorio del usuario
+    cache_dir = os.path.join(os.path.expanduser("~"), ".wdm_cache")
+    os.makedirs(cache_dir, exist_ok=True)
+    
+    # Iniciar navegador usando ChromeDriverManager con directorio de caché personalizado
+    service = Service(ChromeDriverManager(path=cache_dir).install())
     driver = webdriver.Chrome(service=service, options=chrome_options)
     
     try: