image_upload.py

import requests
from dotenv import dotenv_values
from pathlib import Path


class UploadImageError(RuntimeError):
    pass


def upload_image(image_path, token, url, extra_data=None, content_type="application/octet-stream"):
    """
    Upload a file as multipart/form-data with fields:
      - file (file upload)
      - filename (text)
      - auth_token (text)
    """
    image_path = Path(image_path)
    if not image_path.exists():
        raise FileNotFoundError(f"{image_path} not found")

    filename = image_path.name
    file_size = image_path.stat().st_size

    with image_path.open("rb") as f:
        files = {"file": (filename, f, content_type)}
        data = {"filename": filename, "auth_token": token}
        if extra_data:
            data.update(extra_data)

        resp = requests.post(url, files=files, data=data, timeout=30)

    if resp.status_code == 200:
        print("✓ HTTP 200")
        try:
            print(resp.json())
        except Exception:
            print(resp.text)
    else:
        print(f"upload failed for {image_path} ({file_size} bytes, {content_type})")
        print(f"✗ HTTP {resp.status_code}: {resp.text}")
        raise UploadImageError(
            f"Upload failed for {image_path.name} ({file_size} bytes, {content_type}): "
            f"HTTP {resp.status_code} {resp.text}"
        )

    return resp


def upload_live_photo(live_photo_result, token, url):
    still_path = live_photo_result.still_path
    motion_path = live_photo_result.motion_path

    if still_path is None or motion_path is None:
        raise ValueError("Live photo bundle is incomplete")

    bundle_id = live_photo_result.bundle_id
    still_response = upload_image(
        image_path=still_path,
        token=token,
        url=url,
        extra_data={
            "bundle_id": bundle_id,
            "asset_id": live_photo_result.asset_id,
            "asset_kind": "live_photo_still",
            "apple_metadata_ready": str(live_photo_result.apple_metadata_ready).lower(),
        },
        content_type="image/jpeg",
    )
    print(f"uploaded live photo still {still_path} ({bundle_id})")
    motion_response = upload_image(
        image_path=motion_path,
        token=token,
        url=url,
        extra_data={
            "bundle_id": bundle_id,
            "asset_id": live_photo_result.asset_id,
            "asset_kind": "live_photo_motion",
            "apple_metadata_ready": str(live_photo_result.apple_metadata_ready).lower(),
        },
        content_type="application/octet-stream",
    )
    print(f"uploaded live photo motion {motion_path} ({bundle_id})")
    return still_response, motion_response

if __name__ == "__main__":
    env = dotenv_values(".env")
    upload_image_token = env.get("UPLOAD_IMAGE_TOKEN")
    upload_image_url = env.get("UPLOAD_IMAGE_URL")

    # Ensure URL ends exactly with /api/upload_image
    if not upload_image_url or not upload_image_url.rstrip("/").endswith("/api/upload_image"):
        raise SystemExit("Set UPLOAD_IMAGE_URL to the server endpoint ending with `/api/upload_image`")

    upload_image("test.png", token=upload_image_token, url=upload_image_url)