forked from mrlan/EnglishPal
107 lines
4.2 KiB
Python
107 lines
4.2 KiB
Python
import random
|
|
import string
|
|
import time
|
|
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
from selenium.webdriver.support import expected_conditions as EC
|
|
|
|
from helper import signup
|
|
|
|
def has_punctuation(s):
|
|
return any(c in string.punctuation for c in s)
|
|
|
|
def login(driver, home, uname, password):
|
|
driver.get(home)
|
|
WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.LINK_TEXT, '登录'))).click()
|
|
driver.find_element(By.ID, 'username').send_keys(uname)
|
|
driver.find_element(By.ID, 'password').send_keys(password)
|
|
driver.find_element(By.XPATH, '//button[text()="登录"]').click()
|
|
WebDriverWait(driver, 10).until(EC.title_is(f"EnglishPal Study Room for {uname}"))
|
|
|
|
|
|
def select_valid_word(driver):
|
|
elem = driver.find_element(By.ID, 'text-content')
|
|
essay_content = elem.text
|
|
valid_word = random.choice([word for word in essay_content.split() if len(word) >= 6 and not has_punctuation(
|
|
word) and 'font>' not in word and 'br>' not in word and 'p>' not in word])
|
|
driver.find_element(By.ID, 'selected-words').send_keys(valid_word)
|
|
return valid_word
|
|
|
|
|
|
def select_more_words(driver):
|
|
elem = driver.find_element(By.ID, 'text-content')
|
|
essay_content = elem.text
|
|
valid_words = [word for word in essay_content.split() if len(word) >= 6 and not has_punctuation(
|
|
word) and 'font>' not in word and 'br>' not in word and 'p>' not in word]
|
|
words_to_send = ', '.join(valid_words[:7])
|
|
driver.find_element(By.ID, 'selected-words').send_keys(words_to_send)
|
|
|
|
|
|
def test_unfamiliar_button(driver, URL):
|
|
try:
|
|
username, password = signup(URL, driver)
|
|
time.sleep(5)
|
|
|
|
word = select_valid_word(driver)
|
|
|
|
# Wait for the button to be clickable
|
|
time.sleep(8)
|
|
# Add selected word to the list of unfamiliar words
|
|
add_to_unfamiliar_button = driver.find_element(By.XPATH, '//button[text()="把生词加入我的生词库"]')
|
|
add_to_unfamiliar_button.click()
|
|
# Wait for the page to update
|
|
time.sleep(3)
|
|
|
|
add_words_button = driver.find_element(By.XPATH, '//button[text()="加入我的生词簿"]') # Replace with actual ID or selector
|
|
add_words_button.click()
|
|
# Wait for the page to update again
|
|
time.sleep(2)
|
|
|
|
# Check if the added word appears in the word container
|
|
word_container = driver.find_element(By.CLASS_NAME, 'word-container')
|
|
words = word_container.find_elements(By.TAG_NAME, 'p')
|
|
added_word_exists = any(word.text == word for word in words)
|
|
time.sleep(2)
|
|
|
|
# Mark word as unfamiliar
|
|
time.sleep(5)
|
|
for _ in range(16):
|
|
time.sleep(1)
|
|
unfamiliar_btn = driver.find_element(By.XPATH, '//a[contains(@class, "btn-warning") and text()="不熟悉"]')
|
|
unfamiliar_btn.click()
|
|
# Wait for the page to updatec
|
|
time.sleep(5)
|
|
|
|
|
|
|
|
#add new words
|
|
word = select_more_words(driver)
|
|
add_to_unfamiliar_button = driver.find_element(By.XPATH, '//button[text()="把生词加入我的生词库"]')
|
|
add_to_unfamiliar_button.click()
|
|
|
|
time.sleep(5)
|
|
|
|
#add the words to the list of words
|
|
add_words_again = driver.find_element(By.XPATH, '//button[text()="加入我的生词簿"]') # Replace with actual ID or selector
|
|
add_words_again.click()
|
|
# Wait for the page to update again
|
|
time.sleep(3)
|
|
|
|
|
|
#current count after adding new words
|
|
current_unfamiliar_count = driver.find_element(By.XPATH, '//a[contains(@class, "btn-warning") and text()="不熟悉"]')
|
|
current_unfamiliar_count = int(unfamiliar_btn.text.split('(')[1].split(')')[0])
|
|
print(f"Unfamiliar word count: {current_unfamiliar_count}")
|
|
time.sleep(3)
|
|
|
|
|
|
# Log out and close the browser
|
|
driver.find_element(By.XPATH, '//a[contains(@class, "btn-secondary") and text()="退出"]')
|
|
driver.execute_script("window.open('');window.close();")
|
|
WebDriverWait(driver, 2)
|
|
driver.execute_script("window.open('');")
|
|
driver.switch_to.window(driver.window_handles[-1])
|
|
|
|
finally:
|
|
driver.quit() |