サイトコピー

# ==========================================================
# 案件ごとのサイト作成：案件一覧（CSV）の1行ごとに、テンプレートのフォルダをサイトコピーAPIで
# 複製し、フォルダ名を「案件コード 案件名」にします。作業区分のようなマスタはレコードごと、
# 作業明細は空で複製します。複製したフォルダとサイトのIDは、結果のCSVに出力します。
# ==========================================================
import csv
import json
from pathlib import Path

import requests

# ===== 設定 =====
# APIキーは、テナント管理者のものを使ってください（サイトコピーAPIの前提）
BASE_URL = "【URL】"
API_KEY = "【APIキー】"
TEMPLATE_FOLDER_ID = "【テンプレートのフォルダのサイトID】"
# テンプレートのフォルダの中で、複製するサイト。IncludeData が True のサイトはレコードも複製する
TEMPLATE_SITES = [
    {"SiteId": "【作業区分のサイトID】", "IncludeData": True},
    {"SiteId": "【作業明細のサイトID】", "IncludeData": False},
]
TARGET_FOLDER_ID = "【複製先のフォルダのサイトID】"
INPUT_CSV = Path("./projects.csv")           # 1行目は見出し「案件コード,案件名」
OUTPUT_CSV = Path("./project_sites.csv")
# True のときは複製せず、作るフォルダの名前だけを表示する
DRY_RUN = False

# 権限・通知・リマインダーを複製に含めるか。全案件で同じ設定にする
COPY_OPTIONS = {
    "IncludeSitePermission": False,
    "IncludeRecordPermission": False,
    "IncludeColumnPermission": False,
    "IncludeNotifications": True,
    "IncludeReminders": True,
}


def api_post(path, payload=None):
    """Pleasanter APIへPOSTする。レスポンスのJSONを返す（HTTPエラーでも本文を返す）"""
    response = requests.post(
        f"{BASE_URL}{path}",
        json={"ApiVersion": 1.1, "ApiKey": API_KEY, **(payload or {})},
        headers={"Content-Type": "application/json"},
        timeout=60,
    )
    return response.status_code, response.json()


def check_templates():
    """複製するサイトが、テンプレートのフォルダの中の期限付き・記録テーブルかを確かめ、タイトルを返す。
    存在しないサイトIDを指定すると、エラーにならず空のフォルダが作られるため、複製の前に確かめる"""
    titles = []
    for site in TEMPLATE_SITES:
        status, body = api_post(f"/api/items/{int(site['SiteId'])}/getsite")
        if status != 200:
            raise SystemExit(f"サイトID {site['SiteId']} を取得できません: {body.get('Message')}")
        data = body["Response"]["Data"]
        if data["ParentId"] != int(TEMPLATE_FOLDER_ID) or data["ReferenceType"] not in ("Issues", "Results"):
            raise SystemExit(f"サイトID {site['SiteId']} は、テンプレートのフォルダの中の期限付き・記録テーブルではありません")
        titles.append(data["Title"])
    return titles


def copy_project(folder_title):
    """テンプレートのフォルダを複製し、フォルダ名を folder_title にする。複製元→複製先のIDの対応を返す"""
    status, body = api_post(f"/api/items/{int(TEMPLATE_FOLDER_ID)}/copysitepackage", {
        "TargetSiteId": int(TARGET_FOLDER_ID),
        # SiteTitle は、URLに指定したサイト（テンプレートのフォルダ）の複製先のタイトルになる
        "SiteTitle": folder_title,
        "SelectedSites": [{"SiteId": int(TEMPLATE_FOLDER_ID)}]
        + [{"SiteId": int(site["SiteId"]), "IncludeData": site["IncludeData"]} for site in TEMPLATE_SITES],
        **COPY_OPTIONS,
    })
    # 指定の誤りで複製できなかったときも HTTP 200 で返り、Data に画面向けのメッセージが入る
    rows = json.loads(body["Response"]["Data"]) if status == 200 else []
    if not rows or any("NewSiteId" not in row for row in rows):
        raise RuntimeError(f"「{folder_title}」を複製できませんでした: {body}")
    return {row["OldSiteId"]: row["NewSiteId"] for row in rows}


def read_projects(path):
    with open(path, encoding="utf-8-sig", newline="") as f:
        return [(row["案件コード"].strip(), row["案件名"].strip()) for row in csv.DictReader(f) if row["案件コード"].strip()]


def main():
    projects = read_projects(INPUT_CSV)
    if DRY_RUN:
        for code, name in projects:
            print(f"[DRY_RUN] 作るフォルダ: {code} {name}")
        return
    titles = check_templates()
    with open(OUTPUT_CSV, "w", encoding="utf-8-sig", newline="") as f:
        writer = csv.writer(f)
        writer.writerow(["案件コード", "案件名", "フォルダのサイトID"] + [f"{title}のサイトID" for title in titles])
        for code, name in projects:
            # 失敗したら、そこで止める（それまでに複製した案件は結果のCSVに残る）
            new_ids = copy_project(f"{code} {name}")
            writer.writerow([code, name, new_ids[int(TEMPLATE_FOLDER_ID)]]
                            + [new_ids[int(site["SiteId"])] for site in TEMPLATE_SITES])
            f.flush()
            print(f"複製しました: {code} {name}（フォルダ {new_ids[int(TEMPLATE_FOLDER_ID)]}）")


if __name__ == "__main__":
    main()