python自動化登錄

'''
Created on 2019年
用戶登錄
@author: Root
'''
import time
import pytesseract
from PIL import Image, ImageEnhance
from selenium import webdriver

# 打開瀏覽器並窗口最大化
driver = webdriver.Chrome()
driver.get(url)
driver.maximize_window()

# 識別驗證碼 
def codeRecognize():
    # 1.截取屏幕內容，保存到本地
    driver.save_screenshot("F://09_0102_zjs//code//01.png")

    # 2.打開截圖
    ran = Image.open("F://09_0102_zjs//code//01.png")
    # 3.獲取驗證碼位置，截取保存驗證碼
    imgElement = driver.find_element_by_id("captchaImage")
    #獲取驗證碼x,y軸坐標
    location = imgElement.location
    #獲取驗證碼的長寬
    size=imgElement.size
    # 定位驗證碼,自動定位或者手動定位     
    # 參數：左，上，右，下
    codeRange=(int(location['x']),int(location['y']),int(location['x']+size['width']+1),int(location['y']+size['height']))
    
    # 截取驗證碼並保存
    ran.crop(codeRange).save("F://09_0102_zjs//code//02.png")
    
    # 4、獲取驗證碼圖片，讀取驗證碼
    imageCode = Image.open("F://09_0102_zjs//code//02.png")  # 圖像增強，二值化
    # imageCode.load()
    sharp_img = ImageEnhance.Contrast(imageCode).enhance(2.0)
    sharp_img.save("F://09_0102_zjs//code//03.png")
    sharp_img.load()  # 對比度增強
    time.sleep(3)
    
    # 5、識別驗證碼
    pytesseract.pytesseract.tesseract_cmd = "C://Program Files//Tesseract-OCR//tesseract.exe"
    image = Image.open("F://09_0102_zjs//code//03.png")
    result = pytesseract.image_to_string(image)
    print(result)
    
    return result

def userLogin():
    user = driver.find_element_by_id("username")
    pwd = driver.find_element_by_id("password")
    user.clear()
    pwd.clear()
    # 輸入用戶名和密碼
    user.send_keys("xxx")
    pwd.send_keys("xxx")
   
    # 輸入驗證碼
    result = codeRecognize()
    driver.find_element_by_id("captcha").send_keys(result)
    
    # 點擊登錄
    time.sleep(1.5)
    driver.find_element_by_id("login").click()
    time.sleep(0.5)
# 檢查是否登錄
isLogin = driver.find_element_by_xpath("//span[text()='掃碼登錄']")

try:
    time.sleep(0.5)
    while isLogin.is_enabled():
        userLogin()
        try:
            exitEle = driver.find_element_by_id("headerHandle")
            if exitEle.is_enabled():
                exit()
        except:
            print ("登錄頁面元素不存在")
        else:
            time.sleep(0.5)
            # 刷新頁面    
            driver.refresh()
            userLogin() 
except:
    print ("登錄頁面元素不存在")
else:
    pass

1. 識別驗證碼，實現自動登錄