-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdriver.py
47 lines (38 loc) · 1.6 KB
/
driver.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.remote.webelement import WebElement
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.common.by import By
from time import sleep
class Driver:
def __init__(self):
self.options = Options()
self.options.page_load_strategy = 'eager'
self.driver = None
def initialize_driver(self) -> None:
# En caso que no cuenten con el driver de Chrome, deben descomentar esta linea
# self.driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()))
self.driver = webdriver.Chrome(options=self.options)
self.driver.delete_all_cookies()
self.driver.maximize_window()
def load_page(self, page: str) -> None:
self.driver.get(page)
sleep(5)
def click_element(self, by: str, xpath: str) -> None:
sleep(1)
element = self.driver.find_element(by=by, value=xpath)
element.click()
def find_element(self, by: str, xpath: str) -> WebElement:
return self.driver.find_element(by=by, value=xpath)
def scroll_to_element(self, by: str, xpath: str) -> None:
element = self.find_element(by, xpath)
self.driver.execute_script("arguments[0].scrollIntoView(true);", element)
sleep(5)
def close(self) -> None:
self.driver.quit()
self.driver = None
if __name__ == '__main__':
chrome = Driver()
chrome.initialize_driver()
chrome.load_page('https://www.google.com')