forked from AliSK81/LMS-manager
-
Notifications
You must be signed in to change notification settings - Fork 0
/
driver.py
60 lines (47 loc) · 2.21 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
48
49
50
51
52
53
54
55
56
57
58
59
60
from subprocess import CREATE_NO_WINDOW
from selenium.common.exceptions import WebDriverException
from selenium.webdriver import Chrome
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from webdriver_manager.chrome import ChromeDriverManager
class DriverOptions(Options):
def __init__(self):
super().__init__()
self.add_argument("start-maximized")
self.add_argument("--ignore-certificate-errors")
self.add_argument("--ignore-ssl-errors")
self.add_argument('allow-file-access-from-files')
self.add_argument('use-fake-device-for-media-stream')
self.add_argument('use-fake-ui-for-media-stream')
self.add_experimental_option("detach", True)
self.add_experimental_option("excludeSwitches", ["enable-automation", "enable-logging"])
self.add_experimental_option("useAutomationExtension", False)
self.add_experimental_option("prefs", {"credentials_enable_service": False,
"profile.password_manager_enabled": False})
class Browser(Chrome):
def __init__(self):
o = DriverOptions()
s = Service(ChromeDriverManager(log_level=0).install())
# hide chromedriver console window. (windows only)
s.creationflags = CREATE_NO_WINDOW
super().__init__(options=o, service=s)
class WebDriver:
def __init__(self):
self.browser = Browser()
def load_browser(self):
self.browser.get("http://lms.ui.ac.ir/")
def login_account(self, user):
try:
self.browser.find_element(By.ID, "annonmentc").click()
except WebDriverException:
pass
self.browser.find_element(By.ID, "username").send_keys(user.username)
self.browser.find_element(By.ID, "password").send_keys(user.password)
self.browser.find_element(By.ID, "submit").click()
assert "errors" not in self.browser.page_source
def switch_newtab(self):
currentTab = self.browser.current_window_handle
for tab in self.browser.window_handles:
if tab != currentTab:
self.browser.switch_to.window(tab)