import httpx
import os
import logging
from io import BytesIO

# 配置日志
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

def download_image(url: str, save_path: str):
    """
    使用 httpx 下载图片并保存到本地
    :param url: 图片的 URL
    :param save_path: 保存图片的本地路径
    :return: 保存的文件路径或 None（失败时）
    """
    try:
        # 发起 HTTP GET 请求
        with httpx.Client() as client:
            response = client.get(url)
            if response.status_code == 200:
                # 确保保存路径的目录存在
                os.makedirs(os.path.dirname(save_path), exist_ok=True)
                # 保存图片到本地
                with open(save_path, "wb") as f:
                    f.write(response.content)
                logger.info(f"图片已保存到 {save_path}")
                return save_path
            else:
                logger.warning(f"下载失败，状态码: {response.status_code}")
                return None
    except Exception as e:
        logger.error(f"下载图片时发生错误: {e}")
        return None

if __name__ == "__main__":
    # 示例用法
    image_url = input("请输入图片的 URL: ")
    save_path = input("请输入保存路径（例如 E:/img/test.png）: ")
    
    result = download_image(image_url, save_path)
    if result:
        print(f"图片已成功保存到 {result}")
    else:
        print("下载图片失败，请检查 URL 和路径是否正确。")