41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
|
from selenium.webdriver import Remote
|
||
|
|
from selenium.webdriver.remote.webelement import WebElement
|
||
|
|
from selenium.webdriver.support.wait import WebDriverWait
|
||
|
|
from selenium.webdriver.common.by import By
|
||
|
|
|
||
|
|
|
||
|
|
BASE_URL = 'https://aternos.org'
|
||
|
|
|
||
|
|
RM_SCRIPTS = '''
|
||
|
|
const lst = document.querySelectorAll("script")
|
||
|
|
for (let js of lst) {
|
||
|
|
if (
|
||
|
|
js.src.includes('googletagmanager.com') ||
|
||
|
|
js.src.includes('cloudflareinsights.com') ||
|
||
|
|
js.innerText.includes('LANGUAGE_VARIABLES')
|
||
|
|
) {
|
||
|
|
js.remove()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
'''
|
||
|
|
|
||
|
|
|
||
|
|
class SeleniumHelper:
|
||
|
|
|
||
|
|
def __init__(self, driver: Remote) -> None:
|
||
|
|
self.driver = driver
|
||
|
|
self.wait = WebDriverWait(driver, 2.0)
|
||
|
|
|
||
|
|
def load_page(self, path: str) -> None:
|
||
|
|
self.driver.get(f'{BASE_URL}{path}')
|
||
|
|
self.driver.execute_script(RM_SCRIPTS)
|
||
|
|
|
||
|
|
def find_by_id(self, value: str) -> WebElement:
|
||
|
|
return self.driver.find_element(By.ID, value)
|
||
|
|
|
||
|
|
def find_by_class(self, value: str) -> WebElement:
|
||
|
|
return self.driver.find_element(By.CLASS_NAME, value)
|
||
|
|
|
||
|
|
def exec_js(self, script: str) -> None:
|
||
|
|
self.driver.execute_script(script)
|