Retrieve User (All)

import csv
import json
from pathlib import Path

import requests

# ==============================
# Pleasanter connection settings
# ==============================
BASE_URL = "【URL】"
API_KEY = "【API key】"
API_VERSION = 1.1

# ==============================
# File settings (change them as appropriate for your environment)
# ==============================
INPUT_DIR = Path("./input")
GROUPS_CSV = INPUT_DIR / "groups.csv"
USERS_CSV = INPUT_DIR / "users.csv"
MEMBERSHIPS_CSV = INPUT_DIR / "user_group_memberships.csv"
RELATIONS_CSV = INPUT_DIR / "group_relations.csv"


# ==============================
# CSV input settings (change them as appropriate for your environment)
# ==============================
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)
        ]


def is_active(row):
    return str(row.get("is_active", "1")).strip() == "1"


def to_bool_text(value):
    return "True" if str(value).strip() == "1" else "False"


# ==============================
# Pleasanter API operations
# ==============================
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()


# ==============================
# Get groups
# ==============================
def get_groups():
    data = post_json(
        f"{BASE_URL}/api/groups/get",
        {"ApiVersion": API_VERSION, "ApiKey": API_KEY},
    )
    return {
        row["GroupName"]: row
        for row in data.get("Response", {}).get("Data", [])
        if row.get("GroupName")
    }


# ==============================
# Get users
# ==============================
def get_users():
    data = post_json(
        f"{BASE_URL}/api/users/get",
        {"ApiVersion": API_VERSION, "ApiKey": API_KEY},
    )
    return {
        row["LoginId"]: row
        for row in data.get("Response", {}).get("Data", [])
        if row.get("LoginId")
    }


# ==============================
# Create/update a group
# ==============================
def upsert_group(group_name, payload, existing_groups):
    current = existing_groups.get(group_name)

    if current:
        url = f"{BASE_URL}/api/groups/{current['GroupId']}/update"
        result = post_json(url, payload)
        print(f"[UPDATE] {group_name}")
    else:
        url = f"{BASE_URL}/api/groups/create"
        result = post_json(url, payload)
        print(f"[CREATE] {group_name}")

    return result


# ==============================
# Set the group members
# ==============================
def build_member_map(memberships):
    result = {}
    for row in memberships:
        if not is_active(row):
            continue
        group_code = row["group_code"]
        result.setdefault(group_code, []).append(row)
    return result


# ==============================
# Build the parent-child relationships
# ==============================
def build_relation_map(relations):
    result = {}
    for row in relations:
        if not is_active(row):
            continue
        parent_code = row["parent_group_code"]
        result.setdefault(parent_code, []).append(row["child_group_code"])
    return result


# ==============================
# Build the request payload for Phase1
# ==============================
def build_payload_phase1(group_row, member_rows, users_by_id, pleasanter_users):
    members = []

    for row in member_rows:
        user = users_by_id[row["user_id"]]
        login_id = user["login_id"]
        pleasanter_user = pleasanter_users[login_id]
        user_id = pleasanter_user["UserId"]
        is_admin = to_bool_text(row.get("is_group_admin", "0"))
        members.append(f"User,{user_id},{is_admin}")

    return {
        "ApiVersion": API_VERSION,
        "ApiKey": API_KEY,
        "GroupName": group_row["group_name"],
        "Body": group_row.get("remark", ""),
        "GroupMembers": members,
        "GroupChildren": [],
    }


# ==============================
# Build the request payload for Phase2
# ==============================
def build_payload_phase2(
    group_row,
    member_rows,
    child_codes,
    users_by_id,
    groups_by_code,
    pleasanter_users,
    existing_groups,
):
    members = []

    for row in member_rows:
        user = users_by_id[row["user_id"]]
        login_id = user["login_id"]
        pleasanter_user = pleasanter_users[login_id]
        user_id = pleasanter_user["UserId"]
        is_admin = to_bool_text(row.get("is_group_admin", "0"))
        members.append(f"User,{user_id},{is_admin}")

    children = []

    for child_code in child_codes:
        child_name = groups_by_code[child_code]["group_name"]
        child_group = existing_groups[child_name]
        children.append(f"Group,{child_group['GroupId']},")

    return {
        "ApiVersion": API_VERSION,
        "ApiKey": API_KEY,
        "GroupName": group_row["group_name"],
        "Body": group_row.get("remark", ""),
        "GroupMembers": members,
        "GroupChildren": children,
    }


# ==============================
# Main process
# ==============================
def main():
    groups = read_csv(GROUPS_CSV)
    users = read_csv(USERS_CSV)
    memberships = read_csv(MEMBERSHIPS_CSV)
    relations = read_csv(RELATIONS_CSV)

    groups_by_code = {row["group_code"]: row for row in groups}
    users_by_id = {row["user_id"]: row for row in users}
    members_by_group = build_member_map(memberships)
    children_by_parent = build_relation_map(relations)

    pleasanter_users = get_users()
    existing_groups = get_groups()

    # Phase 1: Create/update all groups (without child groups)
    print("=== Phase 1 ===")
    for group in groups:
        payload = build_payload_phase1(
            group,
            members_by_group.get(group["group_code"], []),
            users_by_id,
            pleasanter_users,
        )
        upsert_group(group["group_name"], payload, existing_groups)
        existing_groups = get_groups()

    # Phase 2: Update the parent-child relationships
    print("=== Phase 2 ===")
    existing_groups = get_groups()

    for group in groups:
        payload = build_payload_phase2(
            group,
            members_by_group.get(group["group_code"], []),
            children_by_parent.get(group["group_code"], []),
            users_by_id,
            groups_by_code,
            pleasanter_users,
            existing_groups,
        )
        upsert_group(group["group_name"], payload, existing_groups)


if __name__ == "__main__":
    main()