bocai-cli 0.1.0__py3-none-any.whl

bocai_cli/commands/card.py

@@ -0,0 +1,260 @@
+from __future__ import annotations
+
+import json
+
+from typing import Any
+
+import typer
+
+from bocai_cli.api import ApiFacade
+from bocai_cli.command_utils import command_handler
+from bocai_cli.config import load_config
+from bocai_cli.date_range_validation import validate_date_range
+from bocai_cli.errors import ConfigValidationError
+from bocai_cli.output import print_json
+
+app = typer.Typer(help="卡相关命令。", invoke_without_command=True)
+
+
+@app.callback()
+def card_root(ctx: typer.Context) -> None:
+    if ctx.invoked_subcommand is None:
+        typer.echo(ctx.get_help())
+        raise typer.Exit(code=0)
+
+
+def _build_api() -> ApiFacade:
+    config = load_config()
+    return ApiFacade(
+        base_url=config.base_url,
+        token=config.token,
+        timeout_seconds=config.timeout_seconds,
+    )
+
+
+def _parse_transaction_ids(raw_transaction_ids: str) -> list[str]:
+    values = [item.strip() for item in raw_transaction_ids.split(",")]
+    non_empty_values = [item for item in values if item]
+
+    unique_values: list[str] = []
+    seen: set[str] = set()
+    for value in non_empty_values:
+        if value not in seen:
+            seen.add(value)
+            unique_values.append(value)
+
+    if not unique_values:
+        raise ConfigValidationError("--transaction-ids 至少要包含一个有效ID。")
+
+    return unique_values
+
+
+def _parse_performance_update_payload(raw_payload: str) -> list[dict[str, Any]]:
+    try:
+        payload = json.loads(raw_payload)
+    except json.JSONDecodeError as exc:
+        raise ConfigValidationError("--payload 必须是合法 JSON。") from exc
+
+    if not isinstance(payload, list):
+        raise ConfigValidationError("--payload 必须是数组。")
+
+    if not payload:
+        raise ConfigValidationError("--payload 至少要包含一条交易记录。")
+
+    for index, item in enumerate(payload, start=1):
+        if not isinstance(item, dict):
+            raise ConfigValidationError(f"--payload 第 {index} 项必须是对象。")
+        if "transactionId" not in item:
+            raise ConfigValidationError(f"--payload 第 {index} 项缺少 transactionId。")
+        if "performance" not in item:
+            raise ConfigValidationError(f"--payload 第 {index} 项缺少 performance。")
+
+    return payload
+
+
+def _parse_performance_new_payload(raw_payload: str) -> dict[str, Any]:
+    try:
+        payload = json.loads(raw_payload)
+    except json.JSONDecodeError as exc:
+        raise ConfigValidationError("--payload 必须是合法 JSON。") from exc
+
+    if not isinstance(payload, dict):
+        raise ConfigValidationError("--payload 必须是对象。")
+
+    if "data" not in payload:
+        raise ConfigValidationError("--payload 缺少 data。")
+
+    data = payload["data"]
+    if not isinstance(data, list):
+        raise ConfigValidationError("--payload.data 必须是数组。")
+
+    if not data:
+        raise ConfigValidationError("--payload.data 至少要包含一条合同记录。")
+
+    required_fields = (
+        "seq",
+        "date",
+        "activateNow",
+        "phone",
+        "name",
+        "cardName",
+        "contractNum",
+        "count",
+        "payAmount",
+        "payMethod",
+        "performanceAmount",
+        "performance",
+        "saleType",
+    )
+    for index, item in enumerate(data, start=1):
+        if not isinstance(item, dict):
+            raise ConfigValidationError(f"--payload.data 第 {index} 项必须是对象。")
+        for field in required_fields:
+            if field not in item:
+                raise ConfigValidationError(f"--payload.data 第 {index} 项缺少 {field}。")
+
+    return payload
+
+
+@app.command("sale-performance", help="获取售卡业绩数据")
+@command_handler
+def sale_performance(
+    start_date: str = typer.Option(
+        ...,
+        "--start-date",
+        help="开始日期，格式 YYYY-MM-DD。",
+    ),
+    end_date: str = typer.Option(
+        ...,
+        "--end-date",
+        help="结束日期，格式 YYYY-MM-DD。",
+    ),
+    page_num: int = typer.Option(
+        ...,
+        "--page-num",
+        min=0,
+        help="页码，必须大于等于 0。",
+    ),
+    page_size: int = typer.Option(
+        ...,
+        "--page-size",
+        min=1,
+        help="每页条数，必须大于等于 1。",
+    ),
+    type_: int = typer.Option(
+        ...,
+        "--type",
+        min=1,
+        max=2,
+        help="卡类型：1会员卡，2培训卡。",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    validate_date_range(start_date, end_date)
+
+    api = _build_api()
+    data = api.card.sale_performance(
+        start_date=start_date,
+        end_date=end_date,
+        page_num=page_num,
+        page_size=page_size,
+        type_=type_,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("sale-performance-clear", help="清除售卡业绩数据")
+@command_handler
+def sale_performance_clear(
+    transaction_ids: str = typer.Option(
+        ...,
+        "--transaction-ids",
+        help="交易ID列表，逗号分隔。",
+    ),
+    phone: str = typer.Option(
+        ...,
+        "--phone",
+        help="不分业绩员工手机号。",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    parsed_transaction_ids = _parse_transaction_ids(transaction_ids)
+
+    api = _build_api()
+    data = api.card.performance_clear(
+        transaction_ids=parsed_transaction_ids,
+        phone=phone,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("sale-performance-update", help="更新售卡业绩数据")
+@command_handler
+def sale_performance_update(
+    payload: str = typer.Option(
+        ...,
+        "--payload",
+        help="业绩更新列表，JSON 数组字符串。[{\\\"transactionId\\\":1, \\\"performance\\\":[{\\\"sellerName\\\":\\\"测试\", \\\"amount\\\":10.0}]}]",
+    ),
+    phone: str = typer.Option(
+        ...,
+        "--phone",
+        help="不分业绩员工手机号。",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    parsed_payload = _parse_performance_update_payload(payload)
+
+    api = _build_api()
+    data = api.card.performance_update(
+        phone=phone,
+        transactions=parsed_payload,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("sale-performance-new", help="录卡合同业绩数据")
+@command_handler
+def sale_performance_new(
+    type_: int = typer.Option(
+        ...,
+        "--type",
+        min=1,
+        max=2,
+        help="卡类型：1会员卡，2培训卡。",
+    ),
+    payload: str = typer.Option(
+        ...,
+        "--payload",
+        help="待录单合同列表，JSON 对象字符串。{\\\"data\":[{\\\"seq\\\":1,\\\"date\\\":\"2026-01-01\\\",\\\"activateNow\\\":true,\\\"phone\\\":\\\"xxx\\\",\\\"name\\\":\\\"xxxx\\\",\\\"cardName\\\":\\\"xxx\\\",\\\"contractNum\\\":\\\"xxx\\\",\\\"count\\\":3,\\\"payAmount\\\":1000,\\\"payMethod\\\":\\\"\\\",\\\"performanceAmount\\\":1000,\\\"performance\\\":[{\\\"sellerName\\\":\\\"xxx\\\",\\\"amount\\\":1000}],\\\"saleType\\\":\\\"新卡\\\",\\\"remarks\\\":\\\"\\\"}]} ",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    parsed_payload = _parse_performance_new_payload(payload)
+
+    api = _build_api()
+    data = api.card.performance_new(
+        type_=type_,
+        payload=parsed_payload,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})

bocai_cli/commands/checkin.py

@@ -0,0 +1,71 @@
+from __future__ import annotations
+
+import typer
+
+from bocai_cli.api import ApiFacade
+from bocai_cli.command_utils import command_handler
+from bocai_cli.config import load_config
+from bocai_cli.date_range_validation import validate_date_range
+from bocai_cli.output import print_json
+
+app = typer.Typer(help="签到相关命令。", invoke_without_command=True)
+
+
+@app.callback()
+def checkin_root(ctx: typer.Context) -> None:
+    if ctx.invoked_subcommand is None:
+        typer.echo(ctx.get_help())
+        raise typer.Exit(code=0)
+
+
+def _build_api() -> ApiFacade:
+    config = load_config()
+    return ApiFacade(
+        base_url=config.base_url,
+        token=config.token,
+        timeout_seconds=config.timeout_seconds,
+    )
+
+
+@app.command("list", help="获取签到记录数据")
+@command_handler
+def list_checkin(
+    start_date: str = typer.Option(
+        ...,
+        "--start-date",
+        help="开始日期，格式 YYYY-MM-DD。",
+    ),
+    end_date: str = typer.Option(
+        ...,
+        "--end-date",
+        help="结束日期，格式 YYYY-MM-DD。",
+    ),
+    page_num: int = typer.Option(
+        ...,
+        "--page-num",
+        min=0,
+        help="页码，必须大于等于 0。",
+    ),
+    page_size: int = typer.Option(
+        ...,
+        "--page-size",
+        min=1,
+        help="每页条数，必须大于等于 1。",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    validate_date_range(start_date, end_date)
+
+    api = _build_api()
+    data = api.checkin.list(
+        start_date=start_date,
+        end_date=end_date,
+        page_num=page_num,
+        page_size=page_size,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})

bocai_cli/commands/course.py

@@ -0,0 +1,260 @@
+from __future__ import annotations
+
+import json
+
+from typing import Any
+
+import typer
+
+from bocai_cli.api import ApiFacade
+from bocai_cli.command_utils import command_handler
+from bocai_cli.config import load_config
+from bocai_cli.date_range_validation import validate_date_range
+from bocai_cli.errors import ConfigValidationError
+from bocai_cli.output import print_json
+
+app = typer.Typer(help="课程相关命令。", invoke_without_command=True)
+
+
+@app.callback()
+def course_root(ctx: typer.Context) -> None:
+    if ctx.invoked_subcommand is None:
+        typer.echo(ctx.get_help())
+        raise typer.Exit(code=0)
+
+
+def _build_api() -> ApiFacade:
+    config = load_config()
+    return ApiFacade(
+        base_url=config.base_url,
+        token=config.token,
+        timeout_seconds=config.timeout_seconds,
+    )
+
+
+def _parse_transaction_ids(raw_transaction_ids: str) -> list[str]:
+    values = [item.strip() for item in raw_transaction_ids.split(",")]
+    non_empty_values = [item for item in values if item]
+
+    unique_values: list[str] = []
+    seen: set[str] = set()
+    for value in non_empty_values:
+        if value not in seen:
+            seen.add(value)
+            unique_values.append(value)
+
+    if not unique_values:
+        raise ConfigValidationError("--transaction-ids 至少要包含一个有效ID。")
+
+    return unique_values
+
+
+def _parse_performance_update_payload(raw_payload: str) -> list[dict[str, Any]]:
+    try:
+        payload = json.loads(raw_payload)
+    except json.JSONDecodeError as exc:
+        raise ConfigValidationError("--payload 必须是合法 JSON。") from exc
+
+    if not isinstance(payload, list):
+        raise ConfigValidationError("--payload 必须是数组。")
+
+    if not payload:
+        raise ConfigValidationError("--payload 至少要包含一条交易记录。")
+
+    for index, item in enumerate(payload, start=1):
+        if not isinstance(item, dict):
+            raise ConfigValidationError(f"--payload 第 {index} 项必须是对象。")
+        if "transactionId" not in item:
+            raise ConfigValidationError(f"--payload 第 {index} 项缺少 transactionId。")
+        if "performance" not in item:
+            raise ConfigValidationError(f"--payload 第 {index} 项缺少 performance。")
+
+    return payload
+
+
+def _parse_performance_new_payload(raw_payload: str) -> dict[str, Any]:
+    try:
+        payload = json.loads(raw_payload)
+    except json.JSONDecodeError as exc:
+        raise ConfigValidationError("--payload 必须是合法 JSON。") from exc
+
+    if not isinstance(payload, dict):
+        raise ConfigValidationError("--payload 必须是对象。")
+
+    if "data" not in payload:
+        raise ConfigValidationError("--payload 缺少 data。")
+
+    data = payload["data"]
+    if not isinstance(data, list):
+        raise ConfigValidationError("--payload.data 必须是数组。")
+
+    if not data:
+        raise ConfigValidationError("--payload.data 至少要包含一条合同记录。")
+
+    required_fields = (
+        "seq",
+        "date",
+        "activateNow",
+        "phone",
+        "name",
+        "courseName",
+        "contractNum",
+        "count",
+        "payAmount",
+        "payMethod",
+        "performanceAmount",
+        "performance",
+        "saleType",
+    )
+    for index, item in enumerate(data, start=1):
+        if not isinstance(item, dict):
+            raise ConfigValidationError(f"--payload.data 第 {index} 项必须是对象。")
+        for field in required_fields:
+            if field not in item:
+                raise ConfigValidationError(f"--payload.data 第 {index} 项缺少 {field}。")
+
+    return payload
+
+
+@app.command("sale-performance", help="获取售课业绩数据")
+@command_handler
+def sale_performance(
+    start_date: str = typer.Option(
+        ...,
+        "--start-date",
+        help="开始日期，格式 YYYY-MM-DD。",
+    ),
+    end_date: str = typer.Option(
+        ...,
+        "--end-date",
+        help="结束日期，格式 YYYY-MM-DD。",
+    ),
+    page_num: int = typer.Option(
+        ...,
+        "--page-num",
+        min=0,
+        help="页码，必须大于等于 0。",
+    ),
+    page_size: int = typer.Option(
+        ...,
+        "--page-size",
+        min=1,
+        help="每页条数，必须大于等于 1。",
+    ),
+    type_: int = typer.Option(
+        ...,
+        "--type",
+        min=1,
+        max=2,
+        help="课程类型：1私教课，2培训课。",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    validate_date_range(start_date, end_date)
+
+    api = _build_api()
+    data = api.course.sale_performance(
+        start_date=start_date,
+        end_date=end_date,
+        page_num=page_num,
+        page_size=page_size,
+        type_=type_,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("sale-performance-clear", help="清除售课业绩数据")
+@command_handler
+def sale_performance_clear(
+    transaction_ids: str = typer.Option(
+        ...,
+        "--transaction-ids",
+        help="交易ID列表，逗号分隔。",
+    ),
+    phone: str = typer.Option(
+        ...,
+        "--phone",
+        help="不分业绩员工手机号。",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    parsed_transaction_ids = _parse_transaction_ids(transaction_ids)
+
+    api = _build_api()
+    data = api.course.performance_clear(
+        transaction_ids=parsed_transaction_ids,
+        phone=phone,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("sale-performance-update", help="更新售课业绩数据")
+@command_handler
+def sale_performance_update(
+    payload: str = typer.Option(
+        ...,
+        "--payload",
+        help="业绩更新列表，JSON 数组字符串。[{\\\"transactionId\\\":1, \\\"performance\\\":[{\\\"sellerName\\\":\\\"测试\", \\\"amount\\\":10.0}]}]",
+    ),
+    phone: str = typer.Option(
+        ...,
+        "--phone",
+        help="不分业绩员工手机号。",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    parsed_payload = _parse_performance_update_payload(payload)
+
+    api = _build_api()
+    data = api.course.performance_update(
+        phone=phone,
+        transactions=parsed_payload,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("sale-performance-new", help="录课程合同业绩数据")
+@command_handler
+def sale_performance_new(
+    type_: int = typer.Option(
+        ...,
+        "--type",
+        min=1,
+        max=2,
+        help="课程类型：1私教课，2培训课。",
+    ),
+    payload: str = typer.Option(
+        ...,
+        "--payload",
+        help="待录单合同列表，JSON 对象字符串。{\\\"data\\\":[{\\\"seq\\\":1,\\\"date\\\":\\\"2026-01-01\\\",\\\"activateNow\\\":true,\\\"phone\\\":\\\"xxx\\\",\\\"name\\\":\\\"xxxx\\\",\\\"courseName\\\":\\\"xxx\\\",\\\"contractNum\\\":\\\"xxx\\\",\\\"count\\\":3,\\\"payAmount\\\":1000,\\\"payMethod\\\":\\\"\\\",\\\"performanceAmount\\\":1000,\\\"performance\\\":[{\\\"sellerName\\\":\\\"xxx\\\",\\\"amount\\\":1000}],\\\"saleType\\\":\\\"新购课\\\", \\\"remarks\\\":\\\"测试备注\\\"}]} ",
+    ),
+    center_id: str | None = typer.Option(
+        None,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    parsed_payload = _parse_performance_new_payload(payload)
+
+    api = _build_api()
+    data = api.course.performance_new(
+        type_=type_,
+        payload=parsed_payload,
+        center_id=center_id,
+    )
+    print_json({"status": "ok", "data": data})

bocai_cli/commands/registry.py

@@ -0,0 +1,36 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+
+import typer
+
+from bocai_cli.commands.auth import app as auth_app
+from bocai_cli.commands.card import app as card_app
+from bocai_cli.commands.checkin import app as checkin_app
+from bocai_cli.commands.course import app as course_app
+from bocai_cli.commands.reservation import app as reservation_app
+from bocai_cli.commands.report import app as report_app
+
+
+@dataclass(frozen=True)
+class CommandRegistration:
+    name: str
+    app: typer.Typer
+    aliases: tuple[str, ...] = ()
+
+
+COMMAND_REGISTRY: tuple[CommandRegistration, ...] = (
+    CommandRegistration(name="auth", app=auth_app),
+    CommandRegistration(name="card", app=card_app),
+    CommandRegistration(name="checkin", app=checkin_app),
+    CommandRegistration(name="course", app=course_app),
+    CommandRegistration(name="reservation", app=reservation_app),
+    CommandRegistration(name="report", app=report_app),
+)
+
+
+def register_commands(root_app: typer.Typer) -> None:
+    for registration in COMMAND_REGISTRY:
+        root_app.add_typer(registration.app, name=registration.name)
+        for alias in registration.aliases:
+            root_app.add_typer(registration.app, name=alias, hidden=True)