Python Selenium + phantomJS 模拟登陆教务管理系统 “抢课”

# _*_coding:utf-8_*_

from selenium import webdriver
from selenium.webdriver.common.action_chains import ActionChains
import os,urllib.request,sys,importlib,time,winreg
def Login_in(username,password):
    #教务管理系统网址
    PostUrl = ""
    #验证码网址
    SecretCodeUrl = ""
    #通过win注册表获得桌面路径地址
    key = winreg.OpenKey(winreg.HKEY_CURRENT_USER, \
                          r'Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders', )
    cwd= winreg.QueryValueEx(key, "Desktop")[0]
    address = str(cwd) + "/image.jpg"
    driver.get(PostUrl)
    driver.find_element_by_name("txtUserName").send_keys(username)
    #双input关联
    driver.find_element_by_name('Textbox1').click()
    driver.find_element_by_id('TextBox2').send_keys(password)
    #验证码
    #读取验证码网址、打开本地路径、写入、输入验证码
    downpicture = urllib.request.urlopen(SecretCodeUrl).read()
    local = open(address, 'wb')
    local.write(downpicture)
    local.close()
    print(u"请从桌面寻找你的验证码")
    print (u"请输入验证码：")
    SecretCode = input()
    driver.find_element_by_id('txtSecretCode').send_keys(SecretCode)
    driver.find_element_by_xpath("//*[@id='Button1']").click()
def To_Grab_class(corm,username):
    directUrl = "http://127.0.0.1/(mil3m1jzac4e0a55aqz5ky45)/xs_main.aspx?xh="+ username
    driver.get(directUrl)
    #鼠标悬停
    chain = ActionChains(driver)
    implement = driver.find_element_by_link_text(u"网上选课")
    chain.move_to_element(implement).perform()
    if corm == 1 :
        driver.find_element_by_link_text("校选修课").click()
    else :
        driver.find_element_by_link_text("院系选修课").click()
def check_all():
    checkboxs = driver.find_elements_by_xpath("//input[@type='checkbox']")
    i=1
    for click in checkboxs :
        i=i+1
        if i % 2 == 0 :
            click.click()
    driver.find_element_by_id("Button1").click()

if __name__ == "__main__":
    importlib.reload(sys)
    print(u"请输入学号：")
    username =input()
    print(u"请输入密码：")
    password = input()
    print(u"校选修课请按1，院系选修课请按2 ：")
    corm = int(input())
    #使用的 phentomjs浏览器 此地需要自行更改
    driver = webdriver.PhantomJS(executable_path=r"d:\Grab_class\phantomjswindows\bin\phantomjs.exe")
    Login_in(username,password)
    To_Grab_class(corm,username)
    check_all()