34 lines
1.2 KiB
Python
34 lines
1.2 KiB
Python
|
from selenium.webdriver.common.by import By
|
||
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
||
|
|
from selenium.webdriver.support import expected_conditions as EC
|
||
|
|
from selenium.common.exceptions import NoSuchElementException, UnexpectedAlertPresentException
|
||
|
|
|
||
|
|
|
||
|
|
def login(driver, url, username, password):
|
||
|
|
try:
|
||
|
|
driver.get(url)
|
||
|
|
|
||
|
|
# Fill in the login form
|
||
|
|
user_input = WebDriverWait(driver, 10).until(
|
||
|
|
EC.element_to_be_clickable((By.ID, "user_name"))
|
||
|
|
)
|
||
|
|
user_input.send_keys(username)
|
||
|
|
|
||
|
|
password_input = WebDriverWait(driver, 10).until(
|
||
|
|
EC.element_to_be_clickable((By.ID, "user_password"))
|
||
|
|
)
|
||
|
|
password_input.send_keys(password)
|
||
|
|
|
||
|
|
# Click the login button
|
||
|
|
login_button = WebDriverWait(driver, 10).until(
|
||
|
|
EC.element_to_be_clickable((By.ID, "login_btn"))
|
||
|
|
)
|
||
|
|
login_button.click()
|
||
|
|
|
||
|
|
# Wait for the admin_tab to become clickable
|
||
|
|
admin_tab = WebDriverWait(driver, 10).until(
|
||
|
|
EC.element_to_be_clickable((By.ID, "admin_tab"))
|
||
|
|
)
|
||
|
|
except (NoSuchElementException, UnexpectedAlertPresentException) as e:
|
||
|
|
return f"Error: {str(e)}"
|