87 lines
3.2 KiB
Python
87 lines
3.2 KiB
Python
import logging
|
|
from selenium import webdriver
|
|
from selenium.webdriver.chrome.options import Options
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
from selenium.webdriver.support import expected_conditions as EC
|
|
|
|
from config import HEADLESS, CF_WAIT_TIME, SELENIUM_URL, TARGET_URL
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def get_chrome_options() -> Options:
|
|
options = Options()
|
|
if HEADLESS:
|
|
options.add_argument("--headless=new")
|
|
options.add_argument("--no-sandbox")
|
|
options.add_argument("--disable-dev-shm-usage")
|
|
options.add_argument("--disable-gpu")
|
|
options.add_argument("--disable-software-rasterizer")
|
|
options.add_argument("--disable-extensions")
|
|
options.add_argument("--disable-blink-features=AutomationControlled")
|
|
options.add_argument("--disable-setuid-sandbox")
|
|
options.add_argument("--disable-background-networking")
|
|
options.add_argument("--disable-default-apps")
|
|
options.add_argument("--disable-sync")
|
|
options.add_argument("--disable-translate")
|
|
options.add_argument("--metrics-recording-only")
|
|
options.add_argument("--mute-audio")
|
|
options.add_argument("--no-first-run")
|
|
options.add_argument("--safebrowsing-disable-auto-update")
|
|
options.add_argument("--ignore-certificate-errors")
|
|
options.add_argument("--ignore-ssl-errors")
|
|
options.add_argument("--user-data-dir=/tmp/chrome-data")
|
|
options.add_argument("--disable-features=IsolateOrigins,site-per-process")
|
|
return options
|
|
|
|
|
|
def apply_stealth_scripts(driver: webdriver.Remote) -> None:
|
|
driver.execute_script("""
|
|
Object.defineProperty(navigator, 'webdriver', {get: () => undefined});
|
|
Object.defineProperty(navigator, 'plugins', {get: () => [1, 2, 3, 4, 5]});
|
|
Object.defineProperty(navigator, 'languages', {get: () => ['en-US', 'en']});
|
|
window.chrome = {runtime: {}};
|
|
|
|
const originalQuery = window.navigator.permissions.query;
|
|
window.navigator.permissions.query = (parameters) => (
|
|
parameters.name === 'notifications'
|
|
? Promise.resolve({state: Notification.permission})
|
|
: originalQuery(parameters)
|
|
);
|
|
Object.defineProperty(navigator, 'permissions', {get: () => window.navigator.permissions});
|
|
""")
|
|
|
|
|
|
def fetch_fresh_credentials() -> tuple[str, str] | None:
|
|
log.info("Starting Selenium to fetch fresh credentials...")
|
|
|
|
driver = None
|
|
try:
|
|
driver = webdriver.Remote(
|
|
command_executor=f"{SELENIUM_URL}/wd/hub",
|
|
options=get_chrome_options(),
|
|
)
|
|
apply_stealth_scripts(driver)
|
|
driver.get(TARGET_URL)
|
|
|
|
wait = WebDriverWait(driver, CF_WAIT_TIME)
|
|
wait.until(EC.presence_of_element_located((By.ID, "logo")))
|
|
|
|
user_agent = driver.execute_script("return navigator.userAgent;")
|
|
selenium_cookies = driver.get_cookies()
|
|
cookies = "; ".join(
|
|
f"{c['name']}={c['value']}"
|
|
for c in selenium_cookies
|
|
)
|
|
log.info("Successfully extracted credentials via Selenium")
|
|
return (user_agent, cookies)
|
|
|
|
except Exception as e:
|
|
log.error(f"Failed to extract credentials: {e}")
|
|
return None
|
|
|
|
finally:
|
|
if driver is not None:
|
|
driver.quit()
|