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

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

app_login(driver)

# 进入直播间


def start_class(driver):
    sleep(3)

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

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

    return start_class


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