"""
HTTP 上传模块 - 将图片上传到指定服务器
"""
import os
import time
import logging
import requests
from typing import Optional
import uuid

logger = logging.getLogger(__name__)


class ImageUploader:
    """图片上传器"""

    def __init__(self, upload_url: str, timeout: int = 30, max_retries: int = 3):
        self.upload_url = upload_url
        self.timeout = timeout
        self.max_retries = max_retries

    def upload(self, image_path: str, serial_number: str, photo_index: int,
               photo_type: str = "front") -> Optional[str]:
        """
        上传单张图片
        返回: 服务器返回的消息（成功时），失败返回 None
        """
        if not os.path.exists(image_path):
            logger.error(f"图片文件不存在: {image_path}")
            return None

        for attempt in range(self.max_retries):
            try:
                with open(image_path, "rb") as f:
                    files = {"file": (os.path.basename(image_path), f, "image/jpeg")}
                    data = {
                        "serialNumber": serial_number,
                        "index": photo_index
                    }
                    resp = requests.post(
                        self.upload_url,
                        files=files,
                        data=data,
                        timeout=self.timeout
                    )

                if resp.status_code == 200:
                    logger.info(f"图片上传成功: {serialNumber} #{photo_index} ({photo_type})")
                    try:
                        return resp.json().get("msg", "success")
                    except:
                        return resp.text
                else:
                    logger.warning(f"上传失败 [{resp.status_code}]: {resp.text[:100]}")

            except requests.exceptions.RequestException as e:
                logger.warning(f"上传异常 (尝试 {attempt+1}/{self.max_retries}): {e}")
                if attempt < self.max_retries - 1:
                    time.sleep(2)

        logger.error(f"图片上传最终失败: {image_path}")
        return None

    def upload_batch(self, image_paths: list, serial_number: str,
                     start_index: int = 0) -> dict:
        """批量上传图片"""
        results = []
        for i, path in enumerate(image_paths):
            result = self.upload(path, serial_number, start_index + i)
            results.append({
                "index": start_index + i,
                "path": path,
                "success": result is not None,
                "msg": result
            })
        return results