import pyotp
import time
import urllib.parse
import cv2  # OpenCV库，包含QRCodeDetector

def decode_qrcode(image_path):
    #使用OpenCV解析二维码#
    # 读取图片
    img = cv2.imread(image_path)
    if img is None:
        raise FileNotFoundError(f"无法读取图片: {image_path}")

    # 创建QR码检测器
    detector = cv2.QRCodeDetector()

    # 检测并解码
    data, _, _ = detector.detectAndDecode(img)
    if not data:
        raise ValueError("无法识别二维码，请检查图片是否清晰或为有效的QR码")

    print(f"二维码内容: {data}")
    return data


def extract_secret_from_qr(qr_content):
    #从二维码内容中提取secret密钥#
    parsed = urllib.parse.urlparse(qr_content)
    params = urllib.parse.parse_qs(parsed.query)

    if 'secret' not in params:
        raise ValueError("二维码中未找到有效的secret密钥")

    return params['secret'][0]


def generate_totp_from_qrcode(image_path):
    #从二维码生成TOTP令牌#
    qr_content = decode_qrcode(image_path)
    secret_key = extract_secret_from_qr(qr_content)

    totp = pyotp.TOTP(secret_key)
    current_token = totp.now()
    remaining_seconds = 30 - (int(time.time()) % 30)

    return current_token, remaining_seconds, secret_key


if __name__ == "__main__":
    QR_CODE_IMAGE_PATH = "./cloud_desktop/s.jpeg"  # 例如："~/Downloads/2fa_qr.png"

    try:
        token, remaining, secret = generate_totp_from_qrcode(QR_CODE_IMAGE_PATH)
        print(f"\n解析到的共享密钥: {secret}")
        print(f"当前动态令牌: {token}")
        print(f"令牌剩余有效时间: {remaining}秒")
    except Exception as e:
        print(f"发生错误: {str(e)}")

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.action_chains import ActionChains
import pyotp
import time


# 配置参数 - 请根据实际情况修改
LOGIN_URL = "https://xxxxx.com.cn/#/"  # 登录页面URL
USERNAME = "xxxx"  # 账号
PASSWORD = "xxxx"  # 密码
QR_CODE_PATH = "./cloud_desktop/s.jpeg"  # 二维码图片路径（首次运行需要，之后可注释）
SAVED_SECRET = "xxxxx"  # 保存解析后的密钥，如："JBSWY3DPEHPK3PXP"（首次运行后填写）

def generate_totp_token(secret):
    """生成当前有效的TOTP令牌"""
    totp = pyotp.TOTP(secret)
    return totp.now()


# 自定义条件：检查元素是否不包含某个属性
class attribute_not_contains(object):
    def __init__(self, locator, attribute, value):
        self.locator = locator
        self.attribute = attribute
        self.value = value

    def __call__(self, driver):
        element = driver.find_element(*self.locator)
        return element.get_attribute(self.attribute) != self.value


def login():
    # 初始化浏览器
    driver = webdriver.Safari()
    driver.maximize_window()
    driver.get(LOGIN_URL)

    try:
        # 1. 生成TOTP令牌
        totp_token = generate_totp_token(SAVED_SECRET)
        print(f"生成当前令牌: {totp_token}")

        # 2. 输入账号
        username_field = WebDriverWait(driver, 15).until(
            EC.element_to_be_clickable((By.ID, "username"))
        )
        ActionChains(driver).move_to_element(username_field).perform()
        driver.execute_script("arguments[0].focus();", username_field)
        username_field.clear()
        username_field.send_keys(USERNAME)
        time.sleep(1)

        # 3. 输入密码
        password_field = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.ID, "password"))  # 替换为实际密码框ID
        )
        ActionChains(driver).move_to_element(password_field).perform()
        driver.execute_script("arguments[0].focus();", password_field)
        password_field.clear()
        password_field.send_keys(PASSWORD)
        time.sleep(1)

        # 4. 输入令牌
        token_field = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable((By.CSS_SELECTOR, "input.el-input__inner[type='password'][placeholder='令牌']"))
        )
        ActionChains(driver).move_to_element(token_field).perform()
        driver.execute_script("arguments[0].focus();", token_field)
        token_field.clear()
        token_field.send_keys(totp_token)
        time.sleep(1)  # 等待表单验证

        # 5. 等待登录按钮启用并点击（使用自定义条件）
        login_button_locator = (By.ID, "login")
        login_button = WebDriverWait(driver, 15).until(
            attribute_not_contains(login_button_locator, "disabled", "disabled")
        )

        # 再次确认按钮可点击
        login_button = WebDriverWait(driver, 10).until(
            EC.element_to_be_clickable(login_button_locator)
        )

        # 执行点击
        driver.execute_script("arguments[0].click();", login_button)

        print("登录操作已完成")
        time.sleep(5)  # 等待登录后页面加载

    except Exception as e:
        print(f"发生错误: {str(e)}")

    finally:
        input("按Enter键关闭浏览器...")
        driver.quit()


if __name__ == "__main__":
    login()