小猿口算pk脚本(基础版)

实现原理：安卓adb截图传到电脑，然后用python裁剪获得两张数字图片，使用ddddocr识别数字，比较大小，再用adb命令模拟安卓手势实现

import os
import ddddocr
from time import sleep
from PIL import Image

# 定义一个函数用于从连接的设备截取屏幕截图，并保存到指定的文件路径
def take_screenshot(path):
    # 使用adb命令截屏并保存到指定路径
    os.system(f'adb shell screencap -p > {path}')
    
    # 读取截取的屏幕截图并替换行结束符
    with open(path, 'rb') as f:
        return f.read().replace(b'\r\n', b'\n')

# 定义一个函数用于处理图片，包括裁剪等操作
def process_image(image_path, crop_area):
    with Image.open(image_path) as img:
        # 使用给定的坐标裁剪图片
        return img.crop(crop_area)

# 定义一个函数用于从图片中识别出文本
def extract_text(img):
    with open(img, 'rb') as f:
        img_bytes = f.read()
    # 使用OCR工具识别图片中的文字
    res = ocr.classification(img_bytes)
    return res.replace(' ', '').replace('\n', '')

# 定义一个函数用于比较两个字符串代表的数字大小，并根据结果执行滑动操作
def compare_numbers(x, y):
    try:
        # 将字符串转换为整数进行比较
        x_int, y_int = int(x), int(y)
        if x_int > y_int:
            print(f"{x} > {y}")
            # 执行向下的滑动操作
            os.system("adb shell input swipe 450 1800 850 1900 1")
            os.system("adb shell input swipe 850 1900 450 2000 1")
        else:
            print(f"{x} < {y}")
            # 执行向上的滑动操作
            os.system("adb shell input swipe 850 1800 450 1900 1")
            os.system("adb shell input swipe 450 1900 850 2000 1")
    except ValueError:
        print("数字格式无效。")

# 主函数定义了程序的主要逻辑
def main():
    screenshot_path = 'screenshot.png'
    
    # 截取屏幕并保存
    screenshot = take_screenshot(screenshot_path)
    with open(screenshot_path, 'wb') as f:
        f.write(screenshot)
    
    # 定义裁剪区域（左，上，右，下），分别对应图片中两个数字的位置
    crop_areas = [
        (330, 720, 530, 880),
        (730, 720, 930, 880)
    ]
    
    cropped_images = []
    for i, crop_area in enumerate(crop_areas, start=1):
        # 裁剪图片并保存
        cropped_image = process_image(screenshot_path, crop_area)
        cropped_image_path = f"screenshot{i}.png"
        cropped_image.save(cropped_image_path)
        cropped_images.append(cropped_image_path)
    
    # 从裁剪后的图片中提取文本
    texts = [extract_text(image) for image in cropped_images]
    
    # 比较提取的数字
    compare_numbers(texts[0], texts[1])

# 如果这个脚本是主程序，则初始化OCR对象并开始循环执行主函数
if __name__ == '__main__':
    ocr = ddddocr.DdddOcr(show_ad=False)
    while True:
        main()
        sleep(0.2)