【Python】使用Python实现调用API获取图片存储到本地

news2025/5/23 21:34:29

使用Python实现调用API获取图片存储到本地

在这里插入图片描述

1、项目概述

开发一个自动化工具，用于从JSON数据源中提取图像ID，通过调用指定API获取未经压缩的原始图像文件，并确保下载结果与Postman等工具直接调用API获取的文件完全一致。

2、核心功能

自动解析含嵌套结构的JSON输入文件

API请求模拟引擎

精确复现Postman请求行为

支持自动重定向跟踪

禁用自动解压缩（保留原始二进制流）

3、环境准备

# 安装依赖
pip install requests python-magic-bin==0.4.24  # Windows
pip install requests python-magic           # Linux/macOS

编辑input.json文件格式：

[
  {
    "postimage": "{...}",
    "preimage": "{...}",
    "original_image_id": "6cd6187a-a20e-f011-998a-000d3ac8927d"
  },
  // 更多数据条目...
]

4、代码实现

import json
import requests
import base64
import os
import mimetypes

# 配置参数
INPUT_JSON = "input.json"  # 输入的JSON文件路径
OUTPUT_DIR = "raw_images"   # 图片输出目录（修改目录以示区别）
API_URL = "https://prod-53.southeastasia.logic.azure.com:443/workflows/8de920a489c7490a9e2207bd538ce964/triggers/manual/paths/invoke?api-version=2016-06-01&sp=%2Ftriggers%2Fmanual%2Frun&sv=1.0&sig=qO01WQl5tzDZeGpeVid43SpndovXbq4G7QO0NWcz7B8"

def determine_image_type(base64_str):
    """通过数据URI或魔术数字检测图片类型"""
    try:
        # 处理数据URI格式
        if base64_str.startswith("data:image/"):
            mime_type = base64_str.split(";")[0].split(":")[1]
            return mimetypes.guess_extension(mime_type) or ".png"
        
        # 通过魔术数字检测原始二进制格式
        decoded = base64.b64decode(base64_str)
        if decoded.startswith(b'\xff\xd8\xff'):
            return ".jpg"
        elif decoded.startswith(b'\x89PNG\r\n\x1a\n'):
            return ".png"
        elif decoded.startswith(b'GIF87a') or decoded.startswith(b'GIF89a'):
            return ".gif"
        elif decoded.startswith(b'RIFF') and decoded[8:12] == b'WEBP':
            return ".webp"
    except Exception:
        pass
    return ".bin"  # 未知类型

def process_images():
    # 创建输出目录
    os.makedirs(OUTPUT_DIR, exist_ok=True)

    # 读取JSON文件
    try:
        with open(INPUT_JSON, 'r', encoding='utf-8') as f:
            data = json.load(f)
    except Exception as e:
        print(f"读取JSON文件失败: {e}")
        return

    # 遍历处理每个条目
    for idx, item in enumerate(data, 1):
        image_id = item.get("original_image_id")
        if not image_id:
            print(f"条目 {idx} 缺少original_image_id，已跳过")
            continue

        print(f"正在处理 ({idx}/{len(data)}): {image_id}")

        # 调用API
        try:
            response = requests.post(
                API_URL,
                json={"EntityGuid": image_id},
                headers={"Content-Type": "application/json"},
                timeout=60  # 延长超时时间
            )
            print(f"响应状态码: {response.status_code}")
            response.raise_for_status()
        except requests.exceptions.RequestException as e:
            print(f"API调用失败: {str(e)[:200]}")  # 显示部分错误信息
            continue

        # 处理base64数据
        try:
            base64_str = response.text
            
            # 检测并分割数据URI
            if "," in base64_str:
                header, payload = base64_str.split(",", 1)
                file_ext = determine_image_type(header)
                base64_str = payload
            else:
                decoded = base64.b64decode(base64_str)
                file_ext = determine_image_type(base64_str)
            
            # 解码base64
            image_data = base64.b64decode(base64_str)
            
            # 生成文件名（保留原始格式）
            output_path = os.path.join(OUTPUT_DIR, f"{image_id}{file_ext}")
            
            # 保存原始二进制数据
            with open(output_path, "wb") as f:
                f.write(image_data)
            
            print(f"原始图片已保存至: {output_path} ({len(image_data):,} bytes)")
            
        except Exception as e:
            print(f"处理失败: {str(e)[:200]}")

if __name__ == "__main__":
    process_images()