from commom.appDriver import get_app_driver
from commom.login import app_login

driver = get_app_driver(r'C:\Users\Administrator\AppData\Local\Programs\xmqxy\小麦企学院.exe')

app_login(driver)

# 进入直播间

def enter_class(driver):
    driver.implicitly_wait(5)

    enter_btn = driver.find_element_by_class_name("ant-btn")
    print(enter_btn)

    driver.find_element_by_class_name("ant-btn").click()

    driver.implicitly_wait(3)
    btn = driver.find_element_by_css_selector("button[class='ant-btn']")
    print(btn.size)
    print(btn.text)


    return enter_btn

if __name__ == '__main__':
    enter_class(driver)

