# Attributes:
#     driver (WebDriver): The Selenium WebDriver instance used to control the browser.
#     By (By): Selenium By class used to locate elements on a web page.
#     WebDriverWait (WebDriverWait): Selenium WebDriverWait class used for implementing explicit waits.
#     expected_conditions (expected_conditions): Module in Selenium used to set expected conditions for explicit waits.
#     queryselector (str): CSS selector used to query and return elements from the DOM. Defaults to '*' which selects all elements.
#     repeat_until_element_in_columns (optional): Specific element to be checked for its presence in the dataframe columns before stopping the query. Useful for waiting on AJAX or dynamically loaded content.
#     max_repeats (int): Maximum number of iterations to perform when checking for the presence of 'repeat_until_element_in_columns'. Defaults to 1.
#     with_methods (bool): Flag to determine if JavaScript methods should be attached to the elements in the resulting dataframe. Defaults to True.

# Methods:
#     __call__(queryselector=None, with_methods=None, repeat_until_element_in_columns=None, max_repeats=None, driver=None, By=None, WebDriverWait=None, expected_conditions=None):
#         Generates a dataframe of web elements based on the specified query selector. The dataframe can include methods attached to these elements if 'with_methods' is True. This method allows for overriding class attributes during its call for flexibility in querying different elements without needing to create multiple instances of the class.

from cythonselenium import SeleniumFrame
from seleniumbase import Driver
from selenium.webdriver.support import expected_conditions
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.common.by import By
import subprocess
from time import sleep

# for SeleniumBase zombie processes
subprocess.run("taskkill /IM uc_driver.exe /F", shell=True)
sleep(5)

max_col_width = 30


if __name__ == "__main__":
    driver = Driver(uc=True)
    getframe = SeleniumFrame(
        driver=driver,
        By=By,
        WebDriverWait=WebDriverWait,
        expected_conditions=expected_conditions,
        queryselector="*",
        repeat_until_element_in_columns=None,
        max_repeats=1,
        with_methods=True,
    )
    driver.get(r"https://www.whitehouse.gov/")

    df = getframe("*")
    # Use https://github.com/directvt/vtm for a line break free experience
    print(df)

    # get only links - query selector "a"
    df2 = getframe("a")

    # Selector
    df3=df2.loc[df2.aa_text=='News'] 

    # Click on the item, without worrying about switching to the correct frame
    # JS click - always works
    df3.iloc[0].js_click()
    # Selenium Click
    df3.iloc[0].se_click()