# coding = utf-8
from selenium import webdriver
import time
import random
from utils import DbUtil
import uuid
from selenium.webdriver import ActionChains
from PIL import Image as Im
import os
import cv2
import numpy as np
import requests
from pymongo import MongoClient

# 代碼1.1 目前只用到webdriver和time庫 其他的會在下面用到
# u 帳號,p 密碼
def Email(u, p):
 # 定義QQ郵箱的登錄頁
 start_url = "https://mail.qq.com"

 # 這里我用的是火狐瀏覽器。很多人喜歡定義成driver 我喜歡定義成browser
 browser = webdriver.Firefox()

 # 休息2s
 time.sleep(2)

 # 使用火狐瀏覽器打開QQ郵箱的登錄頁
 browser.get(start_url)

 # 休息2s（這個sleep時間因網(wǎng)速而異,部分的錯誤就是因為網(wǎng)站還沒打開你就開始獲取網(wǎng)頁的標(biāo)簽進行操作,當(dāng)然就獲取不到然后報錯了~）
 time.sleep(2)

 # 切換frame。login_frame是該登錄窗口iframe的id
 browser.switch_to.frame("login_frame")

 # 點擊選擇帳號密碼登錄
 browser.find_element_by_id("switcher_plogin").click()

 # 休息1s
 time.sleep(1)

 # 輸入帳號 將u填入id是u的輸入框
 browser.find_element_by_id("u").send_keys(u)
 time.sleep(1)

 # 輸入密碼 將p填入id是p的輸入框
 browser.find_element_by_id("p").send_keys(p)
 time.sleep(1)

 # 點擊登錄 登錄按鈕的id是login_button
 browser.find_element_by_id("login_button").click()

# main方法
if __name__ == '__main__':
 # 為了實現(xiàn)異地登錄 隨意定義一個QQ號（反正我們的目的是滑動解鎖0.0），如果直接提示帳號密碼錯誤沒有驗證碼的話就再隨意編一個QQ號
 Email(u="123456789", p="abcdefg")

# 代碼1.1省略....↑
# 代碼1.2.1
# 判斷是否出現(xiàn)了滑動驗證碼
try:
 # 先切換frame回到默認(rèn)
 browser.switch_to.default_content()
 
 # 將frame切換到 login_frame(也就是之前的登錄frame)
 browser.switch_to.frame("login_frame")
 
 # 根據(jù)xpath獲取到含有安全提示的標(biāo)簽然后將其中文本獲取到打印出來 如果異常就進except塊 說明沒有驗證碼
 code = browser.find_element_by_xpath('//*[@id="newVcodeArea"]/div[1]/div/div[2]').text
 print(code)
except :
 print('無安全驗證碼!')

# 代碼2
# 圖片下載到本地,返回一個本地鏈接。url 是圖片的鏈接,type區(qū)分左側(cè)小拼圖和大圖,大圖傳big,小圖傳small
def pic_download(url,type):
 url = url
 root = "D:/emils_python/pic_test/"
 # path = root + str(time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime()))+'.png'
 path = root + type + '.png'
 try:
 if not os.path.exists(root):
 os.mkdir(root)
 if os.path.exists(path):
 os.remove(path)
 r = requests.get(url)
 r.raise_for_status()
 # 使用with語句可以不用自己手動關(guān)閉已經(jīng)打開的文件流
 with open(path, "wb") as f: # 開始寫文件，wb代表寫二進制文件
 f.write(r.content)
 print(f.name)
 print("下載完成")
 return f.name

 except Exception as e:
 print("獲取失敗!" + str(e))

# 代碼3
# 獲取缺口位置 small_url是小圖的路徑(本地),big_url是大圖的路徑(本地) 最后return一個計算出的距離
def get_distance(small_url,big_url):
 # 引用上面的圖片下載
 otemp = pic_download(small_url,'small')
 
 time.sleep(2)
 
 # 引用上面的圖片下載
 oblk = pic_download(big_url,'big')

 # 計算拼圖還原距離
 target = cv2.imread(otemp, 0)
 template = cv2.imread(oblk, 0)
 w, h = target.shape[::-1]
 temp = 'temp.jpg'
 targ = 'targ.jpg'
 cv2.imwrite(temp, template)
 cv2.imwrite(targ, target)
 target = cv2.imread(targ)
 target = cv2.cvtColor(target, cv2.COLOR_BGR2GRAY)
 target = abs(255 - target)
 cv2.imwrite(targ, target)
 target = cv2.imread(targ)
 template = cv2.imread(temp)
 result = cv2.matchTemplate(target, template, cv2.TM_CCOEFF_NORMED)
 x, y = np.unravel_index(result.argmax(), result.shape)
 # 缺口位置
 print((y, x, y + w, x + h))

 # 調(diào)用PIL Image 做測試
 image = Im.open(oblk)

 xy = (y + 20, x + 20, y + w - 20, x + h - 20)
 # 切割
 imagecrop = image.crop(xy)
 # 保存切割的缺口
 imagecrop.save("D:/emils_python/pic_test/new_image.jpg")
 return y

# 代碼1.1省略....↑
# 還是代碼1.2
# 判斷是否出現(xiàn)了滑動驗證碼
try:
 # 先切換frame回到默認(rèn)
 browser.switch_to.default_content()
 
 # 將frame切換到 login_frame(也就是之前的登錄frame)
 browser.switch_to.frame("login_frame")
 
 # 根據(jù)xpath獲取到含有安全提示的標(biāo)簽然后將其中文本獲取到打印出來 如果異常就進except塊 說明沒有驗證碼
 code = browser.find_element_by_xpath('//*[@id="newVcodeArea"]/div[1]/div/div[2]').text
 print(code)
 
 # 如果后面拖動失敗 我們就再次循環(huán) 所以用while
 while True:
 # 切換frame
 browser.switch_to.default_content()
 
 # 切換frame
 browser.switch_to.frame('login_frame')
 
 # 切換帶有刷新按鈕的frame
 browser.switch_to.frame(browser.find_element_by_xpath('//*[@id="newVcodeIframe"]/iframe'))
 
 # 點擊刷新 id為e_reload
 browser.find_element_by_id('e_reload').click()

 # 獲取圖片鏈接
 big_url = browser.find_element_by_id('slideBkg').get_attribute('src')
 small_url = browser.find_element_by_id('slideBlock').get_attribute('src')
 
 # 下載圖片并計算拼圖還原的距離
 y = get_distance(small_url, big_url)
 
 # 獲取當(dāng)前網(wǎng)頁鏈接，用于判斷拖動驗證碼后是否成功,如果拖動后地址沒變則為失敗
 url1 = browser.current_url
 
 # 獲取藍色拖動按鈕對象
 element = browser.find_element_by_id('tcaptcha_drag_button')
 
 # 計算distance
 distance = y * (280 / 680) - 21
 print('distance:', distance)
except :
 print('無安全驗證碼!')

 # 省略上面的代碼 1.1 和1.2
 # 代碼1.3
 # 接著上面的 distance = y * (280 / 680) - 21 繼續(xù)
 # 模擬人為拖動按鈕
 has_gone_dist = 0
 remaining_dist = distance
 # distance += randint(-10, 10)
 # 按下鼠標(biāo)左鍵
 ActionChains(browser).click_and_hold(element).perform()
 time.sleep(0.5)
 while remaining_dist > 0:
  ratio = remaining_dist / distance
  if ratio < 0.2:
  # 開始階段移動較慢
  span = random.randint(5, 8)
  elif ratio > 0.8:
  # 結(jié)束階段移動較慢
  span = random.randint(5, 8)
  else:
  # 中間部分移動快
  span = random.randint(10, 16)
  ActionChains(browser).move_by_offset(span, random.randint(-5, 5)).perform()
  remaining_dist -= span
  has_gone_dist += span
  time.sleep(random.randint(5, 20) / 100)

 ActionChains(browser).move_by_offset(remaining_dist, random.randint(-5, 5)).perform()
 ActionChains(browser).release(on_element=element).perform()

 # 省略代碼 1.1, 1.2, 1.3 在1.3下繼續(xù)寫
 # 獲取當(dāng)前的網(wǎng)頁地址
 url2 = browser.current_url
 
 # frame切回到上一層
 browser.switch_to.parent_frame()
 
 # 判斷拖動按鈕后網(wǎng)頁地址是否有改變,如果變了則說明登錄成功（失敗則停留在該頁面）
 if url1 == url2:
 try :
 print(browser.find_element_by_class_name('tcaptcha-title').text)
 print('滑動失敗!')
 except : 
 print('帳號密碼有誤!')
 else :
 print('登錄成功!')