#coding=UTF-8
import time
import os
from selenium import webdriver
from selenium.common.exceptions import NoAlertPresentException
from pytesser import *
from PIL import Image
from PIL import ImageEnhance  
from PIL import ImageFilter  
import traceback

threshold = 140  
table = []  
for i in range(256):  
    if i < threshold:  
        table.append(0)  
    else:  
        table.append(1)  
rep={'O':'0',  
    'I':'1','L':'1',  
    'Z':'2',  
    'S':'8'  
    };  

## 灰度化照片后得到驗(yàn)證碼
def  getverify1(name):        
    im = Image.open(name)  
    # 轉(zhuǎn)化到灰度圖
    imgry = im.convert('L')
    # imgry.save('g'+name)  
    # 二值化，采用閾值分割法，threshold為分割點(diǎn)
    out = imgry.point(table,'1')  
    # out.save('b'+name)  
    # 識(shí)別
    text = image_to_string(out)  
    # 校正 
    text = text.strip()  
    text = text.upper();    
    for r in rep:  
        text = text.replace(r,rep[r])   
    # out.save(text+'.jpg')  
    print text  
    return text  

# 獲取瀏覽器當(dāng)前的驗(yàn)證碼圖片并調(diào)用返回驗(yàn)證碼
def getVCode(driver):  
    # 保存瀏覽器當(dāng)前頁(yè)面
    driver.save_screenshot("page.png")
    # 從頁(yè)面中截取驗(yàn)證碼（XPATH定位）
    vcode = driver.find_element_by_xpath("http://*[@id='randImage']")
    # 獲取驗(yàn)證碼上下左右邊界坐標(biāo)（手動(dòng)加減像素以更精確）
    loc = vcode.location    
    size = vcode.size
    left = loc['x']+5
    top = loc['y']
    right = (loc['x'] +size['width']-5)
    button = (loc['y']+size['height'])
    # 截取頁(yè)面中的驗(yàn)證碼（進(jìn)行截圖：參數(shù)時(shí)一個(gè)元組（left,top,right,button）并保存
    page_pic = Image.open('page.png')
    v_code_pic = page_pic.crop((left,top,right,button))   
    v_code_pic.save('yzm.png')   
    return getverify1('yzm.png')
    # return getverify1(v_code_pic)

#自動(dòng)登錄操作(參數(shù)為登路賬號(hào)，密碼，webdriver驅(qū)動(dòng)對(duì)象)
def login(username,password,driver):
    v_code = getVCode(driver)
    driver.find_element_by_id('user_name').click() # 點(diǎn)擊用戶名輸入框
    driver.find_element_by_id('user_name').clear() # 清空輸入框
    driver.find_element_by_id('user_name').send_keys(username) # 自動(dòng)敲入用戶名
    
    driver.find_element_by_id('user_password').click() # 點(diǎn)擊密碼輸入框
    driver.find_element_by_id('user_password').clear() # 清空輸入框
    driver.find_element_by_id('user_password').send_keys(password) # 自動(dòng)敲入密碼

    driver.find_element_by_id('v_code').click() # 點(diǎn)擊驗(yàn)證碼輸入框
    driver.find_element_by_id('v_code').clear() # 清空輸入框
    driver.find_element_by_id('v_code').send_keys(v_code) # 自動(dòng)敲入驗(yàn)證碼

    driver.find_element_by_xpath('//*[@id="SubmitButton"]').click() 

#coding=UTF-8
from selenium import webdriver
from batchlogin import *
driver = webdriver.Chrome()
urls=[
    'http://www.test.com:6086/QX_SPD_B2B/',
	'http://www.test.com:6086/QX_SPD_B2B/'
]

for i in range(0, len(urls)):
    url = urls[i]
    windows_open = "window.open('" + url + "')"
    driver.execute_script(windows_open)
    time.sleep(1)
    #獲取當(dāng)前頁(yè)面句柄
    windows = driver.window_handles
    driver.switch_to_window(windows[i+1])
    login("USERNAME","PASSWORD",driver)

    #cookies=driver.get_cookies()
    #print(cookies)
    ##解決頁(yè)面加載不正確的問題
    time.sleep(2)


time.sleep(100)
driver.close()