組織更新

import csv
from pathlib import Path

import requests

# ==============================
# Pleasanter 接続設定
# ==============================

BASE_URL = "【URL】"
API_KEY = "【APIキー】"

API_VERSION = 1.1

# ==============================
# ファイル設定(環境にあわせ適宜変更してください)
# ==============================
INPUT_DIR = Path("./input")
DEPTS_CSV = INPUT_DIR / "depts.csv"

# True の場合、実際の更新は行わず対象だけを出力する
DRY_RUN = False


# ==============================
# CSV入力設定(環境にあわせ適宜変更してください)
# ==============================
def read_csv(path):
    with open(path, "r", encoding="utf-8-sig", newline="") as f:
        return [
            {k.strip(): (v or "").strip() for k, v in row.items()}
            for row in csv.DictReader(f)
        ]


# ==============================
# Pleasanter API操作
# ==============================
def post_json(url, payload):
    r = requests.post(
        url,
        json=payload,
        headers={"Content-Type": "application/json"},
        timeout=60,
    )
    r.raise_for_status()
    return r.json()


# ==============================
# 組織取得
# ==============================
def get_depts():
    data = post_json(
        f"{BASE_URL}/api/depts/get",
        {"ApiVersion": API_VERSION, "ApiKey": API_KEY},
    )
    return {
        row["DeptCode"]: row
        for row in data.get("Response", {}).get("Data", [])
        if row.get("DeptCode")
    }


# ==============================
# 組織の作成
# ==============================
def create_dept(dept_row):
    payload = {
        "ApiVersion": API_VERSION,
        "ApiKey": API_KEY,
        "DeptCode": dept_row["dept_code"],
        "DeptName": dept_row["dept_name"],
        "Body": dept_row.get("remark", ""),
    }
    if DRY_RUN:
        print(f"[CREATE(dry-run)] {dept_row['dept_code']} {dept_row['dept_name']}")
        return
    post_json(f"{BASE_URL}/api/depts/create", payload)
    print(f"[CREATE] {dept_row['dept_code']} {dept_row['dept_name']}")


# ==============================
# 組織名・説明の更新
# ==============================
def update_dept(dept_id, dept_row):
    payload = {
        "ApiVersion": API_VERSION,
        "ApiKey": API_KEY,
        "DeptName": dept_row["dept_name"],
        "Body": dept_row.get("remark", ""),
    }
    if DRY_RUN:
        print(f"[UPDATE(dry-run)] {dept_row['dept_code']} {dept_row['dept_name']}")
        return
    post_json(f"{BASE_URL}/api/depts/{dept_id}/update", payload)
    print(f"[UPDATE] {dept_row['dept_code']} {dept_row['dept_name']}")


# ==============================
# 組織の無効化（CSVから消えた組織）
# ==============================
def disable_dept(dept_code, dept_id):
    payload = {
        "ApiVersion": API_VERSION,
        "ApiKey": API_KEY,
        "Disabled": True,
    }
    if DRY_RUN:
        print(f"[DISABLE(dry-run)] {dept_code}")
        return
    post_json(f"{BASE_URL}/api/depts/{dept_id}/update", payload)
    print(f"[DISABLE] {dept_code}")


# ==============================
# 更新要否の判定
# ==============================
def needs_update(existing_dept, dept_row):
    if existing_dept.get("DeptName") != dept_row["dept_name"]:
        return True
    if (existing_dept.get("Body") or "") != dept_row.get("remark", ""):
        return True
    return False


# ==============================
# メイン処理
# ==============================
def main():
    dept_rows = read_csv(DEPTS_CSV)
    depts_by_code = {row["dept_code"]: row for row in dept_rows}

    existing_depts = get_depts()

    counts = {"created": 0, "updated": 0, "disabled": 0, "unchanged": 0}

    # CSVにある組織を作成/更新
    for dept_row in dept_rows:
        dept_code = dept_row["dept_code"]
        existing = existing_depts.get(dept_code)

        if existing is None:
            create_dept(dept_row)
            counts["created"] += 1
        elif needs_update(existing, dept_row):
            update_dept(existing["DeptId"], dept_row)
            counts["updated"] += 1
        else:
            counts["unchanged"] += 1

    # CSVから消えた組織を無効化（既に無効化済みのものは対象外）
    for dept_code, existing in existing_depts.items():
        if dept_code in depts_by_code:
            continue
        if existing.get("Disabled"):
            continue
        disable_dept(dept_code, existing["DeptId"])
        counts["disabled"] += 1

    print("=== 結果 ===")
    print(f"作成: {counts['created']}件")
    print(f"更新: {counts['updated']}件")
    print(f"無効化: {counts['disabled']}件")
    print(f"変更なし: {counts['unchanged']}件")


if __name__ == "__main__":
    main()