開発者向け機能：API：テナント操作：テナント停止

import argparse
import datetime
import os

import requests

# =========================
# 設定
# =========================
BASE_URL = "【URL】"  # PleasanterサーバーのベースURL
API_KEY = os.environ.get("PLEASANTER_API_KEY")  # 特権ユーザのAPIキー。直書きせず環境変数から取得する
if not API_KEY:
    raise SystemExit("エラー: 環境変数 PLEASANTER_API_KEY が設定されていません。")

API_VERSION = 1.1  # マルチテナント管理APIのApiVersion固定値
DATE_FORMAT = "%Y-%m-%d"  # --suspend-date で受け付ける日付形式
PAGE_SIZE = 200  # Get実行時のページサイズ（Api.jsonの既定値に合わせる）

TENANTS_CREATE_PATH = "/api/tenants/Create"  # テナント作成APIのエンドポイントパス
TENANTS_GET_PATH = "/api/tenants/Get"  # テナント一覧取得APIのエンドポイントパス


# =========================
# 共通処理
# =========================
def call_api(path: str, extra_payload: dict | None = None) -> dict:
    """共通ヘッダ（ApiVersion・ApiKey）を付与してPOSTし、レスポンスJSONを返す"""
    payload = {"ApiVersion": API_VERSION, "ApiKey": API_KEY, **(extra_payload or {})}
    try:
        r = requests.post(f"{BASE_URL}{path}", json=payload, timeout=30)
    except requests.exceptions.RequestException as e:
        raise SystemExit(f"エラー: APIサーバーへの接続に失敗しました。({e})")
    try:
        return r.json()
    except ValueError:
        raise SystemExit(f"エラー: APIから不正な応答が返されました。(HTTP {r.status_code})")


def print_result(body: dict) -> bool:
    """StatusCode・Messageを加工せずラベル付きで表示する。StatusCode=200かどうかを返す"""
    message = (body.get("Response") or {}).get("Message") or body.get("Message")
    print(f"  ステータス : {body.get('StatusCode')}")
    print(f"  メッセージ : {message}")
    return body.get("StatusCode") == 200


def fetch_all_tenants() -> list[dict]:
    """
    テナント一覧を全件取得する（delete時の安全確認などに使う内部処理）。
    PageSizeを明示指定し、Offsetを進めながらページングする。
    """
    tenants, offset = [], 0
    while True:
        body = call_api(TENANTS_GET_PATH, {"Offset": offset, "PageSize": PAGE_SIZE})
        if body.get("StatusCode") != 200:
            print_result(body)
            raise SystemExit(1)
        page = (body.get("Response") or {}).get("Data") or []
        tenants.extend(page)
        if len(page) < PAGE_SIZE:
            return tenants
        offset += PAGE_SIZE


def valid_date(value: str) -> str:
    """--suspend-date引数がYYYY-MM-DD形式であることを検証する"""
    try:
        datetime.datetime.strptime(value, DATE_FORMAT)
    except ValueError:
        raise argparse.ArgumentTypeError(f"日付は{DATE_FORMAT}形式で指定してください。")
    return value


# =========================
# サブコマンド処理
# =========================
def cmd_create(args):
    """createサブコマンドの処理：テナントと初期管理ユーザを作成する"""
    payload = {
        "TenantName": args.tenant_name,
        "LoginId": args.login_id,
        "Name": args.name,
        "NotifyMailAddress": args.notify_mail,
    }
    if args.title:
        payload["Title"] = args.title
    if args.lang:
        payload["Language"] = args.lang
    if not print_result(call_api(TENANTS_CREATE_PATH, payload)):
        raise SystemExit(1)


def cmd_get(args):
    """getサブコマンドの処理：全テナントを取得して一覧表示する"""
    tenants = fetch_all_tenants()
    print(f"  取得件数 : {len(tenants)} 件")
    for t in tenants:
        print(f"    TenantId : {t.get('TenantId')}  TenantName : {t.get('TenantName')}")


def cmd_suspend(args):
    """suspendサブコマンドの処理：指定テナントを停止する"""
    payload = {"SuspendDate": args.suspend_date} if args.suspend_date else {}
    if not print_result(call_api(f"/api/tenants/{args.tenant_id}/Suspend", payload)):
        raise SystemExit(1)


def cmd_resume(args):
    """resumeサブコマンドの処理：指定テナントを再開する"""
    if not print_result(call_api(f"/api/tenants/{args.tenant_id}/Resume")):
        raise SystemExit(1)


def cmd_delete(args):
    """
    誤操作防止のため、一覧取得 → テナント名の入力確認 → 完全一致時のみ削除申請、の順で行う。
    """
    tenants = fetch_all_tenants()
    target = next((t for t in tenants if t.get("TenantId") == args.tenant_id), None)
    if target is None:
        raise SystemExit(f"エラー: テナントID {args.tenant_id} は一覧の中に見つかりませんでした。削除申請を中断します。")

    tenant_name = target.get("TenantName")
    prompt = f"削除申請対象: {tenant_name}（テナントID: {args.tenant_id}）。確認のため、テナント名を入力してください: "
    if input(prompt) != tenant_name:
        raise SystemExit("テナント名が一致しないため、削除申請を中断しました。")

    if not print_result(call_api(f"/api/tenants/{args.tenant_id}/Delete")):
        raise SystemExit(1)


def build_arg_parser():
    """5つのサブコマンド（create/get/suspend/resume/delete）を持つArgumentParserを構築する"""
    parser = argparse.ArgumentParser(description="Pleasanter マルチテナント管理APIクライアント")
    sub = parser.add_subparsers(dest="command", required=True)

    p = sub.add_parser("create", help="テナントを作成する")
    p.add_argument("--tenant-name", required=True)
    p.add_argument("--login-id", required=True)
    p.add_argument("--name", required=True)
    p.add_argument("--notify-mail", required=True)
    p.add_argument("--title")
    p.add_argument("--lang")

    sub.add_parser("get", help="テナント一覧を取得する")

    p = sub.add_parser("suspend", help="テナントを停止する")
    p.add_argument("--tenant-id", type=int, required=True)
    p.add_argument("--suspend-date", type=valid_date, help="YYYY-MM-DD、省略時は即時停止")

    p = sub.add_parser("resume", help="テナントを再開する")
    p.add_argument("--tenant-id", type=int, required=True)

    p = sub.add_parser("delete", help="テナントの削除を申請する")
    p.add_argument("--tenant-id", type=int, required=True)

    return parser


COMMAND_HANDLERS = {
    "create": cmd_create,
    "get": cmd_get,
    "suspend": cmd_suspend,
    "resume": cmd_resume,
    "delete": cmd_delete,
}


# =========================
# メイン
# =========================
def main():
    """エントリポイント：引数を解析し、対応するサブコマンド処理を呼び出す"""
    args = build_arg_parser().parse_args()
    COMMAND_HANDLERS[args.command](args)


if __name__ == "__main__":
    main()