Python Selenium: How to retrieve base64 images after clicking a button-CodePudding

The

CodePudding user response：

Reading through the question and OP's comments, it is clear the crux of the issue is not retrieving/manipulating/decoding the base64 captcha image (which in fact is composed of 2 base64 images with different sizes, so imagine the joy of retrieving, decoding and combining them into one exact replica of what is displayed on the screen), but simply getting the captcha image as seen on the screen. Here is a solution to this problem:

from selenium import webdriver from selenium.webdriver.firefox.service import Service from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC import time as t import undetected_chromedriver as uc options = uc.ChromeOptions() options.add_argument("--no-sandbox") options.add_argument('--disable-notifications') options.add_argument("--window-size=1280,720") options.add_argument('--ignore-certificate-errors') options.add_argument('--allow-running-insecure-content') # options.add_argument('--headless') browser = uc.Chrome(options=options) wait = WebDriverWait(browser, 20) url = 'https://boards.4channel.org/o/' browser.get(url) wait.until(EC.element_to_be_clickable((By.PARTIAL_LINK_TEXT, 'Start a New Thread'))).click() t.sleep(1) wait.until(EC.element_to_be_clickable((By.XPATH, '//button[@id="t-load"]'))).click() captcha_img_background = wait.until(EC.element_to_be_clickable((By.XPATH, '//div[@id="t-bg"]'))) captcha_img_background.screenshot('full_captcha.png') print('got captcha!')

This will save the captcha image as displayed on screen, including both background and foreground, so it can be further used in ML training data etc.

If the issue is saving a generic base64 image to file (not including combining foreground/background of different sizes, as it would clearly be out of scope of original question), let me know and I will update my answer with such example.

I ended up using undetected chromedriver after Firefox & Chrome failed to load the captcha images.

Undetected chromedriver docs: https://github.com/ultrafunkamsterdam/undetected-chromedriver

Selenium docs: https://www.selenium.dev/documentation/