from login import login
from selenium import webdriver
from selenium.webdriver.common.by import By
import time  # 导入时间模块中的 sleep 函数
def main():
    username = 'sun.jian'
    password = '123456'

    # 调用login函数并获取浏览器实例（如果登录成功）
    browser = login(username, password)
    time.sleep(10)
    if browser:
        #点击设备
        step1 = browser.get('http://test.ihemodialysis.com/#/device/index')
        time.sleep(5)
        #点击第一台设备
        step2 = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/div[2]/div/div[1]/div/div").click()
        time.sleep(2)
        #点击故障
        step3 = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/form/div/div[9]/div/div/div/label[2]/span[1]/span").click()
        time.sleep(2)
        #点击确定
        step4 = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/form/div/div[14]/div/div/button[2]").click()
        time.sleep(2)
        print("设备状态被置为故障。")
        #点击筛选
        step5 = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/div[1]/div/div[2]/div[2]/div/div/input").click()
        time.sleep(2)
        #点选故障
        step6 = browser.find_element(By.XPATH,"/html/body/div[8]/div[1]/div/div[1]/ul/li[3]").click()
        time.sleep(2)
        #点击查询
        step7 = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/div[1]/div/div[2]/button[1]/span").click()
        time.sleep(2)
        browser.save_screenshot("logs/screenshot.png")
        #点击患者排班
        step8 = browser.get('http://test.ihemodialysis.com/#/scheduling/PatientsWithScheduling')
        time.sleep(10)
        #选中页面底部
        element = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div[4]/div/table/tbody/tr[6]/td[1]")
        browser.execute_script("arguments[0].scrollIntoView({block: 'start'});", element)
        browser.save_screenshot("logs/screenshot1.png")
        print("截图故障后患者排班。")
        time.sleep(5)
        if browser:
            # 点击设备
            step9 = browser.get('http://test.ihemodialysis.com/#/device/index')
            time.sleep(5)
            # 点击第一台设备
            step10 = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/div[2]/div/div[1]/div/div").click()
            time.sleep(2)
            # 点击报废
            step11 = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/form/div/div[9]/div/div/div/label[3]/span[1]/span").click()
            time.sleep(2)
            print("设备状态被置为报废。")
            # 点击确定
            step12 = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/form/div/div[14]/div/div/button[2]").click()
            time.sleep(2)
            # 点击患者排班
            step13 = browser.get('http://test.ihemodialysis.com/#/scheduling/PatientsWithScheduling')
            time.sleep(5)
            print("点击患者排班")
            # 选中页面底部
            element = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div[4]/div/table/tbody/tr[7]/td[1]")
            browser.execute_script("arguments[0].scrollIntoView({block: 'start'});", element)
            #截图
            browser.save_screenshot("logs/screenshot2.png")
            print("截图报废后患者排班 ")
            time.sleep(5)
            if browser:
                # 点击设备
                step14 = browser.get('http://test.ihemodialysis.com/#/device/index')
                time.sleep(5)
                # 点击筛选
                step15 = browser.find_element(By.XPATH,
                                              "/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/div[1]/div/div[2]/div[2]/div/div/input").click()
                time.sleep(3)
                # 点选报废
                step16 = browser.find_element(By.XPATH, "/html/body/div[8]/div[1]/div/div[1]/ul/li[4]").click()
                time.sleep(3)
                # 点击查询
                step17 = browser.find_element(By.XPATH,
                                              "/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/div[1]/div/div[2]/button[1]/span").click()

                time.sleep(5)
                # 点击第一台设备
                step18 = browser.find_element(By.XPATH,
                                              "/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/div[2]/div/div[1]/div/div").click()
                time.sleep(2)
                # 点击正常
                step19 = browser.find_element(By.XPATH,
                                              "/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/form/div/div[9]/div/div/div/label[3]/span[1]/span").click()
                time.sleep(2)
                print("设备状态被置为正常。")
                # 点击确定
                step20 = browser.find_element(By.XPATH,
                                              "/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div/div/form/div/div[14]/div/div/button[2]").click()
                time.sleep(2)
                # 点击患者排班
                step21 = browser.get('http://test.ihemodialysis.com/#/scheduling/PatientsWithScheduling')
                time.sleep(5)
                print("点击患者排班")
                # 选中页面底部
                element = browser.find_element(By.XPATH,"/html/body/div[1]/section/div[2]/section/div/div[1]/div/main/div/div[1]/div/div/div/div[4]/div/table/tbody/tr[7]/td[1]")
                browser.execute_script("arguments[0].scrollIntoView({block: 'start'});", element)
                # 截图
                browser.save_screenshot("logs/screenshot2.png")
                print("截图正常后患者排班 ")
                time.sleep(5)
            else:
                print("登录失败，浏览器可能未正确打开或已关闭（取决于异常处理）。")
        else:
            print("登录失败，浏览器可能未正确打开或已关闭（取决于异常处理）。")
    else:
        print("登录失败，浏览器可能未正确打开或已关闭（取决于异常处理）。")


if __name__ == "__main__":
    main()