from playwright.sync_api import sync_playwright
from PIL import Image
import numpy as np
import cv2 as cv
import requests
import random
import re
 
 
class QQZonSlide:
    def __init__(self):
        self.login_url = "https://i.qq.com/"
        self.username = "你的賬號"
        self.password = "你的密碼"
        self.page = None
 
    def start(self):
        with sync_playwright() as p:
            self.init_page(p)
            self.login()
 
            while True:
                self.get_slide_bg_img()
                start_x = self.get_slide_block_img_and_start_x()
                distance1, distance2 = self.get_slide_distance(start_x)
                slide_result = self.move_to_notch(distance1, distance2)
                if not slide_result:
                    self.refresh_captcha()
                else:
                    break
 
    def init_page(self, p):
        """初始化瀏覽器，獲取page對象"""
        browser = p.chromium.launch(headless=False)
        self.page = browser.new_page()
 
    def login(self):
        """通過賬號密碼登錄"""
        print("開始登錄")
 
        # 訪問頁面
        self.page.goto(self.login_url)
 
        # 定位到登錄框元素并點擊密碼登錄
        login_frame = self.page.frame_locator("#login_frame")
        login_frame.get_by_role("link", name="密碼登錄").click()
 
        # 清空賬號框然后輸入賬號
        login_frame.locator("#u").clear()
        login_frame.locator("#u").fill(self.username)
 
        # 清空密碼框然后輸入密碼
        login_frame.locator("#p").clear()
        login_frame.locator("#p").fill(self.password)
 
        # 點擊登錄按鈕
        self.page.wait_for_timeout(1000)
        login_frame.locator("#login_button").click()
 
    def get_slide_bg_img(self):
        """截取滑動驗證碼背景圖片"""
        self.page.wait_for_timeout(2000)
        print("正在獲取滑動驗證碼背景圖片")
 
        # 獲取滑動驗證碼所在的iframe
        captcha_iframe = self.page.frame_locator("#login_frame").frame_locator("#tcaptcha_iframe_dy")
 
        # 獲取滑動驗證碼的背景圖
        slide_bg_style = captcha_iframe.locator("#slideBg").get_attribute("style")
        slide_bg_url = re.search(r'url\("(.+)"\)', slide_bg_style).groups()[0]
        r = requests.get(slide_bg_url)
        with open("./slide_bg.png", "wb") as f:
            f.write(r.content)
 
        # 調整圖片大小，根據(jù)style內容將寬度調整為280，高度等比例調整
        img = Image.open("./slide_bg.png")
        ratio = img.width / 280
        img = img.resize(size=(280, int(img.height/ratio)))
        img.save("./slide_bg.png")
 
    def get_slide_block_img_and_start_x(self):
        """獲取滑塊圖片以及初始x坐標"""
        print("正在獲取滑塊圖片")
 
        # 首先保存整個登錄背景截圖
        self.page.screenshot(path="bg.png")
 
        # 獲取滑動驗證碼所在的iframe
        captcha_iframe = self.page.frame_locator("#login_frame").frame_locator("#tcaptcha_iframe_dy")
 
        # 獲取滑塊圖片
        # .tc-fg-item對應的有三個元素，一個是目標滑塊，一個是滑軌，還有一個是滑軌上的按鈕
        for i in range(3):
            slide_block_ele = captcha_iframe.locator(".tc-fg-item").nth(i)
            slide_block_style = slide_block_ele.get_attribute("style")
 
            # 滑軌按鈕元素的style值中不包含url字符串
            if "url" not in slide_block_style:
                continue
 
            # 從元素的style值中分析得出只有目標滑塊的top值小于150
            top_value = re.search(r'top: (.+)px;', slide_block_style).groups()[0]
            if float(top_value) > 150:
                continue
 
            # 獲取x坐標
            slide_block_x = float(re.search(r'left: (.+)px; top: ', slide_block_style).groups()[0])
 
            # 通過滑塊位置，從背景圖中截取滑塊圖片
            slide_block_rect = slide_block_ele.bounding_box()
            bg = Image.open("./bg.png")
            offset = slide_block_rect["width"] // 4  # 從背景圖上截取會混入滑塊周圍的一些像素點，所以加一個偏移值，截取到滑塊內部的圖片。
            slide_block_img = bg.crop((slide_block_rect["x"] + offset, slide_block_rect["y"] + offset,
                                       slide_block_rect["x"] + slide_block_rect["width"] - offset,
                                       slide_block_rect["y"] + slide_block_rect["height"] - offset))
            slide_block_img.save("slide_block.png")
 
            return slide_block_x + slide_block_rect["width"] // 4
 
    @staticmethod
    def set_contrast_brightness(frame, contrast_value, brightness_value):
        if not contrast_value:
            contrast_value = 0.0
 
        if not brightness_value:
            brightness_value = 0
 
        blank = np.zeros(frame.shape, frame.dtype)
        frame = cv.addWeighted(frame, contrast_value, blank, 1 - contrast_value, brightness_value)
 
        return frame
 
    def get_slide_distance(self, start_x):
        """獲取滑動距離"""
        print("正在獲取滑動距離")
        # 通過opencv比較圖片，獲取缺口位置
        slide_bg_img = cv.imread("./slide_bg.png")
        slide_block_img = cv.imread("./slide_block.png")
        slide_block_img = self.set_contrast_brightness(slide_block_img, 0.4, 0)
        result = cv.matchTemplate(slide_block_img, slide_bg_img, cv.TM_CCOEFF_NORMED)
        minVal, maxVal, minLoc, maxLoc = cv.minMaxLoc(result)
 
        # 缺口的x坐標
        notch_x1 = minLoc[0]
        notch_x2 = maxLoc[0]
 
        # 距離
        distance1 = notch_x1 - start_x
        distance2 = notch_x2 - start_x
        return distance1, distance2
 
    @staticmethod
    def get_tracks(distance):
        """獲取移動軌跡"""
        tracks = []                             # 移動軌跡
        current = 0                             # 當前位移
        mid = distance * 4 / 5                  # 減速閾值
        t = 0.2                                 # 計算間隔
        v = 0                                   # 初始速度
 
        while current < distance:
            if current < mid:
                a = random.randint(3, 5)        # 加速度為正5
            else:
                a = random.randint(-5, -3)      # 加速度為負3
 
            v0 = v                              # 初速度 v0
            v = v0 + a * t                      # 當前速度
            move = v0 * t + 1 / 2 * a * t * t   # 移動距離
            current += move
            tracks.append(round(current))
 
        return tracks
 
    def move_to_notch(self, distance1, distance2):
        """移動滑軌按鈕到缺口處"""
        # 獲取滑動驗證碼所在的iframe
        captcha_iframe = self.page.frame_locator("#login_frame").frame_locator("#tcaptcha_iframe_dy")
 
        for i in range(2):
            # 獲取按鈕位置，將鼠標移到上方并按下
            slider_btn_rect = captcha_iframe.get_by_alt_text("slider").bounding_box()
            self.page.mouse.move(slider_btn_rect['x'], slider_btn_rect['y'])
            self.page.mouse.down()
 
            distance = [distance1, distance2][i]
            if distance <= 0:   # 距離不可能小于等于0
                continue
 
            print(f"正在進行第{i+1}次滑動")
            tracks = self.get_tracks(distance)
            for x in tracks:
                self.page.mouse.move(slider_btn_rect['x']+x, random.randint(-5, 5)+slider_btn_rect['y'])
            self.page.mouse.move(slider_btn_rect['x'] + tracks[-1] + 5, random.randint(-5, 5) + slider_btn_rect['y'])
            self.page.mouse.move(slider_btn_rect['x'] + tracks[-1] - 5, random.randint(-5, 5) + slider_btn_rect['y'])
            self.page.mouse.up()
 
            # 滑動結束后等待一段時間
            self.page.wait_for_timeout(2000)
 
            # 尋找按鈕是否還存在，不存在的話表明已通過滑動驗證碼，存在的話嘗試下一個距離
            try:
                captcha_iframe.get_by_alt_text("slider").wait_for(timeout=2000)
            except Exception as e:
                print("已通過滑動驗證碼")
                return True
            else:
                print(f"第{i+1}次滑動失敗")
 
        return False
 
    def refresh_captcha(self):
        """刷新驗證碼"""
        # 獲取滑動驗證碼所在的iframe
        print("刷新驗證碼")
        captcha_iframe = self.page.frame_locator("#login_frame").frame_locator("#tcaptcha_iframe_dy")
        captcha_iframe.locator("#e_reload").click()
        self.page.wait_for_timeout(2000)
 
 
if __name__ == "__main__":
    slide = QQZonSlide()
    slide.start()