python爬虫之初始Selenium

 1 import os
 2 import time
 3 from selenium import webdriver
 4 from  selenium.webdriver.common.by import By
 5 from selenium.webdriver.support.ui import WebDriverWait
 6 from selenium.webdriver.support import expected_conditions as EC
 7 
 8 USER_NAME = 'super'
 9 USER_PWD = 'gaosiedu.com'
10 PATH = os.getcwd()
11 browser = webdriver.Chrome()
12 web_wait = WebDriverWait(browser,10)
13 
14 #加载ip列表
15 def GET_Ip():
16     path = PATH+'\\ip_list.txt'
17     with open(path,'r') as file:
18         for ip in file.readlines():
19             ip = ip.strip()
20             reboot_spiders(ip)
21 
22 #重启方法
23 def reboot_spiders(ip_arg):
24     if len(ip_arg) > 0:
25         try:
26             browser.get("https://{IP}/login.asp".format(IP=ip_arg))
27 
28             # 获取用户名输入框
29             login_name = web_wait.until(EC.presence_of_element_located((By.XPATH,'//*[@id="login-username"]')))
30             # 获取输入密码的输入框
31             login_pwd = web_wait.until(EC.presence_of_element_located((By.XPATH,'//*[@id="password"]')))
32             #获取登录的按钮
33             button = web_wait.until(EC.element_to_be_clickable((By.XPATH,'//*[@id="loginform"]/input')))
34             #在输入用户名的位置输入用户名
35             login_name.send_keys(USER_NAME)
36             #在输入密码的输入框输入密码
37             login_pwd.send_keys(USER_PWD)
38             #点击登录按钮
39             button.click()
40             #分析已经登录的页面
41             #登录成功，开始分析页面!
42             #找到第一个frame标签（左侧栏）
43             browser.switch_to.frame("navframe")
44             #切换进去，执行Reboot Now这个按钮，
45             reboot_bo = web_wait.until(EC.presence_of_element_located((By.XPATH,'/html/body/dl[3]/dd/ul/li[2]/a')))
46             #执行操作，得到右边Reboot Now的重启按钮
47             reboot_bo.click()
48             #跳出当前的frame标签
49             browser.switch_to.default_content()
50             #找到执行完后右边的frame标签
51             browser.switch_to.frame('mainframe')
52             #进入标签后找到重启的按钮
53             reboot_boot = web_wait.until(EC.presence_of_element_located((By.XPATH,'//*[@id="adminform"]/table/tbody/tr[1]/td/input[2]')))
54             #执行重启的操作
55             reboot_boot.click()
56             #关闭浏览器
57             #完成后关闭
58             timee = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
59             s_log = "时间 %s  ip地址[%s]重启成功" %(timee,ip_arg)
60             Success_log(s_log)
61         except Exception as e:
62             timeee = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
63             r_log = "时间 %s  ip地址[%s]重启失败" %(timeee,ip_arg)
64             Error_log(r_log)
65     else:
66         pass
67 
68 def Error_log(args):
69     path = PATH+'\\error.txt'
70     with open(path, 'a+', encoding='utf-8') as file:
71         file.write('\n%s'%args)
72         file.close()
73 
74 def Success_log(args):
75     path = PATH+'\\success.txt'
76     with open(path,'a+',encoding='utf-8') as file:
77         file.write('\n%s'%args)
78         file.close()
79 
80 if __name__ == '__main__':
81     GET_Ip()
82     browser.quit()

1、建立ip_list.txt文件，把ip地址如下面写进去。
172.16.5.2
172.16.5.1
172.16.5.5
172.16.5.50
172.16.5.51
2、本脚本提供了成功日志和错误日志
3、如果上线windows任务计划还需要有多地方需要注意，如有需要可以发邮件给我464774082@qq.com。