python模拟登录东北大学教务处

from selenium import webdriver
from PIL import Image
import pytesseract
import re
import time

driver = webdriver.PhantomJS()

driver.get('https://aao.neu.edu.cn/')

iframe = driver.find_element_by_tag_name("iframe")
driver.switch_to.frame(iframe)

driver.find_element_by_id('WebUserNO').clear()
driver.find_element_by_id('WebUserNO').send_keys('')                   # username
driver.find_element_by_id('Password').clear()
driver.find_element_by_id('Password').send_keys('')                    # password

driver.save_screenshot('login.png')

i = Image.open('login.png')

t = (300,425,350,445)

frame = i.crop(t)
frame.save('agnomen.png')

i = Image.open('agnomen.png')
code = pytesseract.image_to_string(i)
print(len(code))
print code

num1 = int(code[0])
num2 = int(code[len(code) - 2])


print num1,num2

a = re.search(r'\+', code)
if(a):
    agnomen = num1 + num2
else:
    agnomen = num1 * num2

print agnomen

driver.find_element_by_id('Agnomen').clear()
driver.find_element_by_id('Agnomen').send_keys(agnomen)

driver.save_screenshot('a.png')

driver.find_element_by_name('submit').click()

time.sleep(3)

driver.save_screenshot('b.png')