レコードのインポート

import base64
import json
import os
from pathlib import Path

import requests

# =========================
# 設定
# =========================
BASE_URL = "【URL】"
API_KEY = "【APIキー】"
SITE_ID = 【サイトID】  # インポート先サイトのサイトID

# CSVインポートの設定
CSV_PATH = Path("./sample.csv")  # インポートするCSVファイルのパス
CSV_ENCODING = "UTF-8"  # CSVのエンコーディング（"UTF-8" または "Shift-JIS"）
KEY_COLUMN = "ClassB"  # CSVのキー列（管理番号など）。GetItems で照合に使用する

# 画像フォルダのルートパス
IMAGES_DIR = Path("./images")

# "IssueId" or "ResultId"
RECORD_ID_FIELD = "ResultId"

# =========================
# 共通処理
# =========================
# ファイル名プレフィックスと ImageHash のキー（Pleasanter の項目名）の対応
FILE_PREFIX_MAP = {
    "body": "Body",
    "desc_a": "DescriptionA",
    "desc_b": "DescriptionB",
    "desc_c": "DescriptionC",
    "comments": "Comments",
}

SUPPORTED_EXTENSIONS = {".jpg", ".jpeg", ".png", ".gif"}


def post_json(url: str, payload: dict) -> dict:
    response = requests.post(
        url,
        headers={"Content-Type": "application/json"},
        data=json.dumps(payload, ensure_ascii=False),
        timeout=60,
    )
    response.raise_for_status()
    return response.json()


def post_file(url: str, params: dict, file_path: Path) -> dict:
    with open(file_path, "rb") as f:
        response = requests.post(
            url,
            data={"parameters": json.dumps(params, ensure_ascii=False)},
            files={"file": (file_path.name, f, "text/csv")},
            timeout=120,
        )
    response.raise_for_status()
    return response.json()


def encode_image(file_path: Path) -> str:
    """画像ファイルをBase64エンコードして文字列で返す"""
    return base64.b64encode(file_path.read_bytes()).decode("utf-8")


# =========================
# Step 1: CSVインポート
# =========================
def step1_import_csv() -> None:
    print("=" * 50)
    print("Step 1: CSVインポート")
    print("=" * 50)

    url = f"{BASE_URL}/api/items/{SITE_ID}/import"
    params = {
        "ApiVersion": 1.1,
        "ApiKey": API_KEY,
        "Encoding": CSV_ENCODING,
        "UpdatableImport": True,  # キー一致のレコードは更新
        "Key": KEY_COLUMN,
        "MigrationMode": False,
    }

    result = post_file(url, params, CSV_PATH)
    print(f"  ステータス : {result.get('StatusCode')}")
    print(f"  メッセージ : {result.get('Message')}")


# =========================
# Step 2: レコードID取得
# =========================
def step2_get_record_ids() -> dict:
    """
    インポート済みレコードを全件取得し、キー列の値 → レコードID のマップを返す。
    レコード件数が多い場合はページングして全件取得する。
    """
    print("=" * 50)
    print("Step 2: レコードID取得")
    print("=" * 50)

    url = f"{BASE_URL}/api/items/{SITE_ID}/get"
    key_to_id = {}
    offset = 0
    page_size = 200

    while True:
        payload = {
            "ApiVersion": 1.1,
            "ApiKey": API_KEY,
            "Offset": offset,
            "PageSize": page_size,
        }

        result = post_json(url, payload)
        records = result.get("Response", {}).get("Data", [])

        if not records:
            break

        for rec in records:
            # ClassHash 形式と直接指定の両方に対応
            class_hash = rec.get("ClassHash", {})
            key_val = class_hash.get(KEY_COLUMN) or rec.get(KEY_COLUMN)
            record_id = rec.get(RECORD_ID_FIELD)
            if key_val and record_id:
                key_to_id[str(key_val).strip()] = record_id

        if len(records) < page_size:
            break
        offset += page_size

    print(f"  取得件数 : {len(key_to_id)} 件")
    return key_to_id


# =========================
# Step 3: 画像の埋め込み更新
# =========================
def build_image_hash(key_dir: Path) -> dict:
    """
    キーディレクトリ内の画像ファイルを走査し、ImageHash 用の辞書を組み立てる。
    対象外のファイルは無視する。
    """
    image_hash = {}

    for file in sorted(key_dir.iterdir()):
        if not file.is_file():
            continue
        if file.suffix.lower() not in SUPPORTED_EXTENSIONS:
            continue

        # ファイル名のプレフィックスで登録先項目を特定
        prefix = file.stem.lower()
        target_field = FILE_PREFIX_MAP.get(prefix)
        if target_field is None:
            print(f"    [SKIP] 対象外のファイル名: {file.name}（スキップ）")
            continue

        image_hash[target_field] = {
            "HeadNewLine": True,
            "EndNewLine": True,
            "Extension": file.suffix.lower(),
            "Base64": encode_image(file),
        }
        print(f"    {file.name} → {target_field}")

    return image_hash


def step3_update_images(key_to_id: dict) -> None:
    print("=" * 50)
    print("Step 3: 画像の埋め込み更新")
    print("=" * 50)

    if not IMAGES_DIR.exists():
        print(f"  画像フォルダが見つかりません: {IMAGES_DIR}")
        return

    success = skip = error = 0

    for key_dir in sorted(IMAGES_DIR.iterdir()):
        if not key_dir.is_dir():
            continue

        key_val = key_dir.name
        record_id = key_to_id.get(key_val)

        if record_id is None:
            print(f"  [SKIP] キー '{key_val}' に対応するレコードが見つかりません")
            skip += 1
            continue

        print(f"  キー: {key_val}  →  レコードID: {record_id}")
        image_hash = build_image_hash(key_dir)

        if not image_hash:
            print(f"    [SKIP] 登録対象の画像ファイルがありません")
            skip += 1
            continue

        payload = {
            "ApiVersion": 1.1,
            "ApiKey": API_KEY,
            "ImageHash": image_hash,
        }

        url = f"{BASE_URL}/api/items/{record_id}/update"
        try:
            result = post_json(url, payload)
            if result.get("StatusCode") == 200:
                print(f"    [OK] 更新完了")
                success += 1
            else:
                print(
                    f"    [ERR] StatusCode={result.get('StatusCode')}  {result.get('Message')}"
                )
                error += 1
        except requests.HTTPError as e:
            print(f"    [ERR] HTTPエラー: {e}")
            error += 1

    print()
    print(f"完了  成功={success}  スキップ={skip}  エラー={error}")


# =========================
# メイン
# =========================
def main():
    step1_import_csv()
    print()
    key_to_id = step2_get_record_ids()
    print()
    step3_update_images(key_to_id)


if __name__ == "__main__":
    main()