from selenium import webdriver
from time import sleep
import datetime
url = "http://192.168.1.1/login_smart.html"
url_login = "http://192.168.1.1/login.cgi"
url_longout = "http://192.168.1.1/login.cgi#"
Rstart_time = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
i = 0
def write_record(now_log):
    with open(r"C:\Users\brian\Desktop\RestartLogs\log.txt","r+") as f:
        old = f.read()
        f.seek(0,0)
        f.writelines(now_log)
        f.writelines('\n')
        f.writelines(old)
        f.close()

while True:
    try:
        dr = webdriver.Chrome()
        dr.get(url)
        dr.maximize_window()
        sleep(0.1)
        dr.find_element_by_css_selector("#loid_user").clear()
        sleep(0.1)
        dr.find_element_by_css_selector("#loid_user").send_keys("useradmin")
        sleep(0.1)
        dr.find_element_by_css_selector("#loid_pwd").send_keys("a124689a")
        sleep(0.1)
        dr.find_element_by_css_selector("#save_loid").click()
        sleep(10)
        if dr.current_url == url_login:
            print("登录成功")
        dr.find_element_by_css_selector("[alt='重启']").click()
        sleep(0.1)
        dr.find_element_by_css_selector("#okbutton1").click()
        if dr.current_url == url_longout:
            new_log = "重启成功,重启时间是：%s" %Rstart_time
            print(new_log)
            write_record(new_log)
            dr.quit()
            break
        else:
            dr.quit()
            print("重启失败了，关闭浏览器再来")
            continue
    except:
        dr.quit()
        print("代码运行出错，关闭浏览器再来")
        i = i +1
        if i >= 3:
            break
        else:
            continue