# 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庫 其他的會(huì)在下面用到
# u 帳號(hào),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（這個(gè)sleep時(shí)間因網(wǎng)速而異,部分的錯(cuò)誤就是因?yàn)榫W(wǎng)站還沒打開你就開始獲取網(wǎng)頁的標(biāo)簽進(jìn)行操作,當(dāng)然就獲取不到然后報(bào)錯(cuò)了~）
 time.sleep(2)

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

 # 點(diǎn)擊選擇帳號(hào)密碼登錄
 browser.find_element_by_id("switcher_plogin").click()

 # 休息1s
 time.sleep(1)

 # 輸入帳號(hào) 將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)

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

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

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

# 代碼2
# 圖片下載到本地,返回一個(gè)本地鏈接。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語句可以不用自己手動(dòng)關(guān)閉已經(jīng)打開的文件流
 with open(path, "wb") as f: # 開始寫文件，wb代表寫二進(jìn)制文件
 f.write(r.content)
 print(f.name)
 print("下載完成")
 return f.name

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

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

 # 計(jì)算拼圖還原距離
 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 做測(cè)試
 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)了滑動(dòng)驗(yàn)證碼
try:
 # 先切換frame回到默認(rèn)
 browser.switch_to.default_content()
 
 # 將frame切換到 login_frame(也就是之前的登錄frame)
 browser.switch_to.frame("login_frame")
 
 # 根據(jù)xpath獲取到含有安全提示的標(biāo)簽然后將其中文本獲取到打印出來 如果異常就進(jìn)except塊 說明沒有驗(yàn)證碼
 code = browser.find_element_by_xpath('//*[@id="newVcodeArea"]/div[1]/div/div[2]').text
 print(code)
 
 # 如果后面拖動(dòng)失敗 我們就再次循環(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'))
 
 # 點(diǎn)擊刷新 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')
 
 # 下載圖片并計(jì)算拼圖還原的距離
 y = get_distance(small_url, big_url)
 
 # 獲取當(dāng)前網(wǎng)頁鏈接，用于判斷拖動(dòng)驗(yàn)證碼后是否成功,如果拖動(dòng)后地址沒變則為失敗
 url1 = browser.current_url
 
 # 獲取藍(lán)色拖動(dòng)按鈕對(duì)象
 element = browser.find_element_by_id('tcaptcha_drag_button')
 
 # 計(jì)算distance
 distance = y * (280 / 680) - 21
 print('distance:', distance)
except :
 print('無安全驗(yàn)證碼!')

 # 省略上面的代碼 1.1 和1.2
 # 代碼1.3
 # 接著上面的 distance = y * (280 / 680) - 21 繼續(xù)
 # 模擬人為拖動(dòng)按鈕
 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:
  # 開始階段移動(dòng)較慢
  span = random.randint(5, 8)
  elif ratio > 0.8:
  # 結(jié)束階段移動(dòng)較慢
  span = random.randint(5, 8)
  else:
  # 中間部分移動(dòng)快
  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()
 
 # 判斷拖動(dòng)按鈕后網(wǎng)頁地址是否有改變,如果變了則說明登錄成功（失敗則停留在該頁面）
 if url1 == url2:
 try :
 print(browser.find_element_by_class_name('tcaptcha-title').text)
 print('滑動(dòng)失敗!')
 except : 
 print('帳號(hào)密碼有誤!')
 else :
 print('登錄成功!')