使用 Appium 自动化测试 Gmail OTP 验证

from googleapiclient.discovery import build
from google.oauth2 import credentials

# 替换为您的凭据文件路径
CREDENTIALS_FILE = 'path/to/your/credentials.json'

def get_gmail_service():
    creds = None
    # 从 token.json 文件加载用户凭据。
    # 如果没有可用的有效凭据，请让用户登录。
    try:
        creds = credentials.Credentials.from_authorized_user_file('token.json', ['https://www.googleapis.com/auth/gmail.readonly'])
    except FileNotFoundError:
        pass
    # 如果没有可用的（有效）凭据，请让用户登录。
    if not creds or not creds.valid:
        flow = google_auth_oauthlib.flow.InstalledAppFlow.from_client_secrets_file(
            CREDENTIALS_FILE, ['https://www.googleapis.com/auth/gmail.readonly'])
        creds = flow.run_local_server(port=0)
        # 保存凭据以便下次运行
        with open('token.json', 'w') as token:
            token.write(creds.to_json())

    service = build('gmail', 'v1', credentials=creds)
    return service

def get_latest_otp(service, sender_email, subject_keyword):
    """
    获取来自特定发件人且主题包含关键字的最新邮件中的 OTP。
    """
    query = f"from:{sender_email} subject:{subject_keyword}"
    results = service.users().messages().list(userId='me', q=query).execute()
    messages = results.get('messages', [])

    if not messages:
        print("No messages found.")
        return None

    message = service.users().messages().get(userId='me', id=messages[0]['id'], format='full').execute()
    payload = message['payload']
    headers = payload['headers']
    for header in headers:
        if header['name'] == 'Subject':
            subject = header['value']
            break

    # 解析邮件内容，提取 OTP (这里需要根据邮件格式进行调整)
    # 假设 OTP 在邮件正文中，并且是数字
    if 'parts' in payload:
        for part in payload['parts']:
            if part['mimeType'] == 'text/plain':
                data = part['body']['data']
                text = base64.urlsafe_b64decode(data).decode('utf-8')
                # 使用正则表达式提取 OTP
                otp = re.search(r'\d{6}', text) # 假设 OTP 是 6 位数字
                if otp:
                    return otp.group(0)
                else:
                    print("OTP not found in email body.")
                    return None
    else:
        print("No parts found in email payload.")
        return None

# 使用示例
service = get_gmail_service()
sender_email = "your_sender_email@example.com"  # 发送 OTP 的邮箱地址
subject_keyword = "Your OTP"  # 邮件主题中包含的关键字
otp = get_latest_otp(service, sender_email, subject_keyword)

if otp:
    print(f"OTP found: {otp}")
else:
    print("Failed to retrieve OTP.")

import io.appium.java_client.AppiumDriver;
import org.openqa.selenium.By;
import org.openqa.selenium.WebElement;

public class OTPVerification {

    private AppiumDriver driver;

    public OTPVerification(AppiumDriver driver) {
        this.driver = driver;
    }

    public void enterOTP(String otp) {
        // 使用 Appium Inspector 找到 OTP 输入框的定位器
        By otpInputLocator = By.id("otp_input_field"); // 替换为实际的定位器

        // 找到 OTP 输入框元素
        WebElement otpInput = driver.findElement(otpInputLocator);

        // 输入 OTP
        otpInput.sendKeys(otp);

        // (可选) 点击 "验证" 按钮
        // By verifyButtonLocator = By.id("verify_button"); // 替换为实际的定位器
        // WebElement verifyButton = driver.findElement(verifyButtonLocator);
        // verifyButton.click();
    }
}

from appium import webdriver
from selenium.webdriver.common.by import By

def enter_otp(driver, otp):
    # 使用 Appium Inspector 找到 OTP 输入框的定位器
    otp_input_locator = (By.ID, "otp_input_field")  # 替换为实际的定位器

    # 找到 OTP 输入框元素
    otp_input = driver.find_element(*otp_input_locator)

    # 输入 OTP
    otp_input.send_keys(otp)

    # (可选) 点击 "验证" 按钮
    # verify_button_locator = (By.ID, "verify_button")  # 替换为实际的定位器
    # verify_button = driver.find_element(*verify_button_locator)
    # verify_button.click()