bocai-cli-core 0.2.0__py3-none-any.whl

bocai_cli/commands/center.py

@@ -0,0 +1,138 @@
+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.output import print_json
+
+app = typer.Typer(help="门店相关命令。", invoke_without_command=True)
+
+
+@app.callback()
+def center_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_centers() -> None:
+    api = _build_api()
+    data = api.center.list()
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("manager-phone", help="获取场馆店长手机号列表")
+@command_handler
+def manager_phone(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    api = _build_api()
+    data = api.center.manager_phone(center_id=center_id)
+    print_json({"status": "ok", "data": data})
+
+
+@command_handler
+def card_list(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+    type_: int = typer.Option(
+        ...,
+        "--type",
+        min=0,
+        max=2,
+        help="卡种类型：0全部，1会员卡，2课程卡。",
+    ),
+) -> None:
+    api = _build_api()
+    data = api.center.card_list(center_id=center_id, type_=type_)
+    print_json({"status": "ok", "data": data})
+
+
+@command_handler
+def source_list(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    api = _build_api()
+    data = api.center.source_list(center_id=center_id)
+    print_json({"status": "ok", "data": data})
+
+
+@command_handler
+def studio_seller_list(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    api = _build_api()
+    data = api.center.studio_seller_list(center_id=center_id)
+    print_json({"status": "ok", "data": data})
+
+
+@command_handler
+def studio_trainer_list(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    api = _build_api()
+    data = api.center.studio_trainer_list(center_id=center_id)
+    print_json({"status": "ok", "data": data})
+
+
+@command_handler
+def tag_list(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+    type_: int = typer.Option(
+        ...,
+        "--type",
+        min=1,
+        max=3,
+        help="标签类型：1会员标签，2准会员标签，3会员卡标签。",
+    ),
+) -> None:
+    api = _build_api()
+    data = api.center.tag_list(center_id=center_id, type_=type_)
+    print_json({"status": "ok", "data": data})
+
+
+studio_app = typer.Typer(help="门店相关命令。", invoke_without_command=True)
+studio_app.callback()(center_root)
+studio_app.command("list", help="获取门店列表")(list_centers)
+studio_app.command("manager-phone", help="获取场馆店长手机号列表")(manager_phone)
+studio_app.command("card-list", help="获取门店卡种列表")(card_list)
+studio_app.command("source-list", help="获取会员来源列表")(source_list)
+studio_app.command("seller-list", help="获取跟进人/销售人列表")(studio_seller_list)
+studio_app.command("trainer-list", help="获取教练列表")(studio_trainer_list)
+studio_app.command("tag-list", help="获取标签列表")(tag_list)

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 = typer.Option(
+        ...,
+        "--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 = typer.Option(
+        ...,
+        "--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 = typer.Option(
+        ...,
+        "--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 = typer.Option(
+        ...,
+        "--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\\\",\n\\\"contractNum\\\":\\\"xxx\\\",\\\"count\\\":3,\\\"payAmount\\\":1000,\\\"payMethod\\\":\\\"\\\",\n\\\"payAmount2\\\":1000,\\\"payMethod2\\\":\\\"\\\",\\\"performanceAmount\\\":2000,\n\\\"performance\\\":[{\\\"sellerName\\\":\\\"xxx\\\",\\\"amount\\\":2000}],\\\"saleType\\\":\\\"新购课\\\", \\\"remarks\\\":\\\"测试备注\\\",\\\"noPerformancePhone\\\":\\\"xxx\\\"}}]} ",
+    ),
+    center_id: str = typer.Option(
+        ...,
+        "--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/customreport.py

@@ -0,0 +1,77 @@
+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="customreport相关命令。", invoke_without_command=True)
+
+
+@app.callback()
+def customreport_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("ds-list", help="获取定制数据集列表")
+@command_handler
+def ds_list(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+) -> None:
+    api = _build_api()
+    data = api.guanyuan.ds_list(center_id=center_id)
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("ds-data", help="获取定制数据集数据")
+@command_handler
+def ds_data(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+    ds_id: str = typer.Option(
+        ...,
+        "--ds-id",
+        help="数据集ID。",
+    ),
+    start_date: str = typer.Option(
+        ...,
+        "--start-date",
+        help="业务开始日期，格式 YYYY-MM-DD。",
+    ),
+    end_date: str = typer.Option(
+        ...,
+        "--end-date",
+        help="业务结束日期，格式 YYYY-MM-DD。",
+    ),
+) -> None:
+    validate_date_range(start_date, end_date)
+
+    api = _build_api()
+    data = api.guanyuan.ds_data(
+        center_id=center_id,
+        ds_id=ds_id,
+        start_date=start_date,
+        end_date=end_date,
+    )
+    print_json({"status": "ok", "data": data})

bocai_cli/commands/guanyuan.py

@@ -0,0 +1,94 @@
+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.output import print_json
+
+app = typer.Typer(help="观远相关命令。", invoke_without_command=True)
+
+
+@app.callback()
+def guanyuan_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 _register_absence_commands(
+    application: typer.Typer,
+    *,
+    default_type: int,
+    min_type: int,
+    max_type: int,
+    type_help: str,
+) -> None:
+    @application.command("absence-statistics", help="门店未出勤统计")
+    @command_handler
+    def absence_statistics(
+        center_id: str = typer.Option(
+            ...,
+            "--center-id",
+            help="门店ID。",
+        ),
+        type_: int = typer.Option(
+            default_type,
+            "--type",
+            min=min_type,
+            max=max_type,
+            help=type_help,
+        ),
+    ) -> None:
+        api = _build_api()
+        data = api.guanyuan.absence_statistics(center_id=center_id, type_=type_)
+        print_json({"status": "ok", "data": data})
+
+    @application.command("absence-detail", help="门店未出勤明细")
+    @command_handler
+    def absence_detail(
+        center_id: str = typer.Option(
+            ...,
+            "--center-id",
+            help="门店ID。",
+        ),
+        type_: int = typer.Option(
+            default_type,
+            "--type",
+            min=min_type,
+            max=max_type,
+            help=type_help,
+        ),
+    ) -> None:
+        api = _build_api()
+        data = api.guanyuan.absence_detail(center_id=center_id, type_=type_)
+        print_json({"status": "ok", "data": data})
+
+
+_register_absence_commands(
+    app,
+    default_type=2,
+    min_type=2,
+    max_type=3,
+    type_help="未出勤类型：2健身房私教未出勤，3健身房培训未出勤。",
+)
+
+studio_app = typer.Typer(help="观远相关命令。", invoke_without_command=True)
+studio_app.callback()(guanyuan_root)
+_register_absence_commands(
+    studio_app,
+    default_type=1,
+    min_type=1,
+    max_type=1,
+    type_help="未出勤类型：1未出勤。",
+)