如何用Python解决图像验证码

pip install capsolver

import capsolver
import base64
import requests
from io import BytesIO

# -------------------请修改这些值-------------------
# 从 CapSolver 仪表板获取的 API 密钥
CAPSOLVER_API_KEY = "Your_API_KEY"
# ----------------------------------------------------

def setup_capsolver():
    """设置 CapSolver API 密钥。"""
    capsolver.api_key = CAPSOLVER_API_KEY

def get_base64_from_image(source, is_url=False):
    """
    将图像文件或 URL 转换为 Base64 编码的字符串。
    :param source: 图像的路径（本地文件）或 URL。
    :param is_url: 如果 source 是 URL，则为 True。
    :return: Base64 编码的字符串，或在失败时返回 None。
    """
    try:
        if is_url:
            response = requests.get(source)
            response.raise_for_status()  # 对于错误的状态码抛出异常
            image_bytes = BytesIO(response.content)
        else:
            with open(source, 'rb') as image_file:
                image_bytes = BytesIO(image_file.read())
        
        base64_string = base64.b64encode(image_bytes.read()).decode('utf-8')
        return base64_string
    except requests.exceptions.RequestException as e:
        print(f"无法从 URL 获取图像: {e}")
        return None
    except FileNotFoundError:
        print(f"错误：在 {source} 找不到文件")
        return None
    except Exception as e:
        print(f"转换图像为 Base64 时发生错误: {e}")
        return None

def solve_image_captcha(base64_image, module_name="common"):
    """
    使用 CapSolver 解决图像验证码。
    :param base64_image: Base64 编码的图像字符串。
    :param module_name: 识别模块，'common' 用于通用目的。
    :return: 解决方案对象，或在失败时返回 None。
    """
    print("正在将图像验证码任务提交给 CapSolver...")
    try:
        solution = capsolver.solve({
            "type": "ImageToTextTask",
            "module": module_name,
            "body": base64_image
        })
        return solution
    except Exception as e:
        print(f"解决验证码时发生错误: {e}")
        return None

def main():
    """主函数以执行整个流程。"""
    setup_capsolver()

    # --- 示例 1：使用本地图像文件 ---
    print("--- 示例 1：使用本地图像文件 ---")
    local_image_path = "your/captcha/image/path.jpg"  # 将此替换为您的图像路径
    base64_from_file = get_base64_from_image(local_image_path)
    
    if base64_from_file:
        solution = solve_image_captcha(base64_from_file)
        if solution:
            print("解决方案文本:", solution.get("text"))

    print("\n" + "="*30 + "\n")

    # --- 示例 2：使用图像 URL ---
    print("--- 示例 2：使用图像 URL ---")
    image_url = "https://i.postimg.cc/B6hK2V19/captcha-example.png" # 示例图像 URL
    base64_from_url = get_base64_from_image(image_url, is_url=True)

    if base64_from_url:
        solution = solve_image_captcha(base64_from_url)
        if solution:
            print("解决方案文本:", solution.get("text"))


if __name__ == "__main__":
    main()