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

bocai_cli/commands/member.py

@@ -0,0 +1,288 @@
+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_optional_date_order
+from bocai_cli.errors import ConfigValidationError
+from bocai_cli.output import print_json
+
+app = typer.Typer(help="会员相关命令。", invoke_without_command=True)
+
+
+@app.callback()
+def member_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,
+    )
+
+
+@command_handler
+def studio_proposed_list(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+    follower_id: int = typer.Option(
+        -1,
+        "--follower-id",
+        help="跟进人ID，-1表示全部。",
+    ),
+    source_id: int = typer.Option(
+        -1,
+        "--source-id",
+        help="来源ID，-1表示全部。",
+    ),
+    tag_id: int = typer.Option(
+        -1,
+        "--tag-id",
+        help="标签ID，-1表示全部。",
+    ),
+    attend_start_date: str = typer.Option(
+        "",
+        "--attend-start-date",
+        help="加入开始时间，格式 YYYY-MM-DD。",
+    ),
+    attend_end_date: str = typer.Option(
+        "",
+        "--attend-end-date",
+        help="加入结束时间，格式 YYYY-MM-DD。",
+    ),
+    follow_status: int = typer.Option(
+        -1,
+        "--follow-status",
+        min=-1,
+        max=5,
+        help="跟进状态：-1全部，1未体验，2已体验，3跟进中，4完成，5中止。",
+    ),
+    page_num: int = typer.Option(
+        ...,
+        "--page-num",
+        min=0,
+        help="页码，从 0 开始。",
+    ),
+    page_size: int = typer.Option(
+        ...,
+        "--page-size",
+        min=1,
+        help="每页数量，必须大于等于 1。",
+    ),
+) -> None:
+    validate_optional_date_order(
+        attend_start_date,
+        attend_end_date,
+        start_option_name="attend-start-date",
+        end_option_name="attend-end-date",
+    )
+
+    api = _build_api()
+    data = api.member.studio_proposed_list(
+        center_id=center_id,
+        page_num=page_num,
+        page_size=page_size,
+        consultant_id=follower_id,
+        source=source_id,
+        tag_id=tag_id,
+        follow_status=follow_status,
+        attend_start_date=attend_start_date,
+        attend_end_date=attend_end_date,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+def _parse_checkin_status(raw_value: str) -> bool:
+    normalized = raw_value.strip().lower()
+    if normalized == "true":
+        return True
+    if normalized == "false":
+        return False
+    raise ConfigValidationError("--checkin-status 只支持 true 或 false。")
+
+
+@command_handler
+def studio_member_list(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+    follower_id: int = typer.Option(
+        -1,
+        "--follower-id",
+        help="跟进人ID，-1表示全部。",
+    ),
+    source_id: int = typer.Option(
+        -1,
+        "--source-id",
+        help="来源ID，-1表示全部。",
+    ),
+    tag_id: int = typer.Option(
+        -1,
+        "--tag-id",
+        help="标签ID，-1表示全部。",
+    ),
+    member_type: int = typer.Option(
+        -1,
+        "--member-type",
+        min=-1,
+        max=1,
+        help="会员类型：-1全部，0普通会员，1课程会员。",
+    ),
+    member_status: int = typer.Option(
+        -1,
+        "--member-status",
+        min=-1,
+        max=1,
+        help="会员状态：-1全部，0有效会员，1已失效。",
+    ),
+    follow_status: int = typer.Option(
+        -1,
+        "--follow-status",
+        min=-1,
+        max=5,
+        help="跟进状态：-1全部，1未体验，2已体验，3跟进中，4完成，5中止。",
+    ),
+    expire_start_date: str = typer.Option(
+        "",
+        "--expire-start-date",
+        help="将过期开始日期，格式 YYYY-MM-DD。",
+    ),
+    expire_end_date: str = typer.Option(
+        "",
+        "--expire-end-date",
+        help="将过期结束日期，格式 YYYY-MM-DD。",
+    ),
+    birthday_start_date: str = typer.Option(
+        "",
+        "--birthday-start-date",
+        help="生日开始日期，格式 YYYY-MM-DD。",
+    ),
+    birthday_end_date: str = typer.Option(
+        "",
+        "--birthday-end-date",
+        help="生日结束日期，格式 YYYY-MM-DD。",
+    ),
+    checkin_status: str = typer.Option(
+        "true",
+        "--checkin-status",
+        help="签到状态：true已签到，false未签到。",
+    ),
+    last_checkin_start_date: str = typer.Option(
+        "",
+        "--last-checkin-start-date",
+        help="最后签到开始日期，格式 YYYY-MM-DD。",
+    ),
+    last_checkin_end_date: str = typer.Option(
+        "",
+        "--last-checkin-end-date",
+        help="最后签到结束日期，格式 YYYY-MM-DD。",
+    ),
+    course_start_date: str = typer.Option(
+        "",
+        "--course-start-date",
+        help="最后上课开始日期，格式 YYYY-MM-DD。",
+    ),
+    course_end_date: str = typer.Option(
+        "",
+        "--course-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。",
+    ),
+) -> None:
+    validate_optional_date_order(
+        expire_start_date,
+        expire_end_date,
+        start_option_name="expire-start-date",
+        end_option_name="expire-end-date",
+    )
+    validate_optional_date_order(
+        birthday_start_date,
+        birthday_end_date,
+        start_option_name="birthday-start-date",
+        end_option_name="birthday-end-date",
+    )
+    validate_optional_date_order(
+        last_checkin_start_date,
+        last_checkin_end_date,
+        start_option_name="last-checkin-start-date",
+        end_option_name="last-checkin-end-date",
+    )
+    validate_optional_date_order(
+        course_start_date,
+        course_end_date,
+        start_option_name="course-start-date",
+        end_option_name="course-end-date",
+    )
+    parsed_checkin_status = _parse_checkin_status(checkin_status)
+
+    api = _build_api()
+    data = api.member.studio_member_list(
+        center_id=center_id,
+        page_num=page_num,
+        page_size=page_size,
+        consultant_id=follower_id,
+        source=source_id,
+        tag_id=tag_id,
+        member_type=member_type,
+        status=member_status,
+        follow_status=follow_status,
+        expire_start_date=expire_start_date,
+        expire_end_date=expire_end_date,
+        birthday_start_date=birthday_start_date,
+        birthday_end_date=birthday_end_date,
+        check_in_status=parsed_checkin_status,
+        last_checkin_start_date=last_checkin_start_date,
+        last_checkin_end_date=last_checkin_end_date,
+        course_start_date=course_start_date,
+        course_end_date=course_end_date,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("detail", help="获取会员信息")
+@command_handler
+def detail(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+    member_id: str = typer.Option(
+        ...,
+        "--member-id",
+        help="会员ID。",
+    ),
+) -> None:
+    api = _build_api()
+    data = api.member.detail(center_id=center_id, member_id=member_id)
+    print_json({"status": "ok", "data": data})
+
+
+studio_app = typer.Typer(help="会员相关命令。", invoke_without_command=True)
+studio_app.callback()(member_root)
+studio_app.command("proposed-list", help="获取准会员列表")(studio_proposed_list)
+studio_app.command("member-list", help="获取会员列表")(studio_member_list)
+studio_app.command("detail", help="获取会员信息")(detail)

bocai_cli/commands/month_dashboard.py

@@ -0,0 +1,235 @@
+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 monthdashboard_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 _run_dashboard_query(
+    method_name: str,
+    *,
+    center_id: str,
+    start_date: str,
+    end_date: str,
+) -> None:
+    validate_date_range(start_date, end_date)
+
+    api = _build_api()
+    method = getattr(api.monthdashboard, method_name)
+    data = method(
+        center_id=center_id,
+        start_date=start_date,
+        end_date=end_date,
+    )
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("month-payment", help="获取月付履约信息")
+@command_handler
+def month_payment(
+    center_id: str = typer.Option(
+        ...,
+        "--center-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:
+    _run_dashboard_query(
+        "statistics",
+        center_id=center_id,
+        start_date=start_date,
+        end_date=end_date,
+    )
+
+
+@app.command("recognize-income", help="获取会员卡确认收入信息")
+@command_handler
+def recognize_income(
+    center_id: str = typer.Option(
+        ...,
+        "--center-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:
+    _run_dashboard_query(
+        "recognize_income",
+        center_id=center_id,
+        start_date=start_date,
+        end_date=end_date,
+    )
+
+
+@app.command("clue-conversion", help="获取线索转化信息")
+@command_handler
+def clue_conversion(
+    center_id: str = typer.Option(
+        ...,
+        "--center-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:
+    _run_dashboard_query(
+        "clue_conversion",
+        center_id=center_id,
+        start_date=start_date,
+        end_date=end_date,
+    )
+
+
+@app.command("service-delivery", help="获取服务与交付信息")
+@command_handler
+def service_delivery(
+    center_id: str = typer.Option(
+        ...,
+        "--center-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:
+    _run_dashboard_query(
+        "service_delivery",
+        center_id=center_id,
+        start_date=start_date,
+        end_date=end_date,
+    )
+
+
+@app.command("member-count", help="获取会员总数信息")
+@command_handler
+def member_count(
+    center_id: str = typer.Option(
+        ...,
+        "--center-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:
+    _run_dashboard_query(
+        "member_count",
+        center_id=center_id,
+        start_date=start_date,
+        end_date=end_date,
+    )
+
+
+@app.command("attendance-count", help="获取出勤与活跃信息")
+@command_handler
+def attendance_count(
+    center_id: str = typer.Option(
+        ...,
+        "--center-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:
+    _run_dashboard_query(
+        "attendance_count",
+        center_id=center_id,
+        start_date=start_date,
+        end_date=end_date,
+    )
+
+
+@app.command("revenue-info", help="获取营收状态信息")
+@command_handler
+def revenue_info(
+    center_id: str = typer.Option(
+        ...,
+        "--center-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:
+    _run_dashboard_query(
+        "revenue_info",
+        center_id=center_id,
+        start_date=start_date,
+        end_date=end_date,
+    )

bocai_cli/commands/qywx.py

@@ -0,0 +1,112 @@
+from __future__ import annotations
+
+import json
+from pathlib import Path
+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.errors import ConfigValidationError
+from bocai_cli.output import print_json
+
+app = typer.Typer(help="企业微信相关命令。", invoke_without_command=True)
+
+
+@app.callback()
+def qywx_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_send_message_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 "phones" not in payload:
+        raise ConfigValidationError("--payload 缺少 phones。")
+    if not isinstance(payload["phones"], list):
+        raise ConfigValidationError("--payload.phones 必须是数组。")
+
+    if "type" not in payload:
+        raise ConfigValidationError("--payload 缺少 type。")
+
+    message_type = payload["type"]
+    if message_type not in ("file", "text"):
+        raise ConfigValidationError("--payload.type 只支持 file 或 text。")
+
+    if message_type == "file":
+        if "media_id" not in payload:
+            raise ConfigValidationError("--payload 缺少 media_id。")
+    if message_type == "text":
+        if "content" not in payload:
+            raise ConfigValidationError("--payload 缺少 content。")
+
+    return payload
+
+
+def _validate_file_path(file_path: Path) -> Path:
+    if not file_path.exists():
+        raise ConfigValidationError("--file 指定的文件不存在。")
+    if not file_path.is_file():
+        raise ConfigValidationError("--file 必须是普通文件。")
+    return file_path
+
+
+@app.command("send-message", help="发送企业微信消息")
+@command_handler
+def send_message(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+    payload: str = typer.Option(
+        ...,
+        "--payload",
+        help='发送消息内容，JSON 对象字符串。{\\\"phones\\\":[],\\\"type\\\":\\\"file\\\",\\\"media_id\\\":\\\"xxx\\\"} 或 {\\\"phones\\\":[],\\\"type\\\":\\\"text\\\",\\\"content\\\":\\\"xxx\\\"}',
+    ),
+) -> None:
+    parsed_payload = _parse_send_message_payload(payload)
+
+    api = _build_api()
+    data = api.qywx.send_message(center_id=center_id, payload=parsed_payload)
+    print_json({"status": "ok", "data": data})
+
+
+@app.command("upload", help="上传企业微信媒体文件")
+@command_handler
+def upload(
+    center_id: str = typer.Option(
+        ...,
+        "--center-id",
+        help="门店ID。",
+    ),
+    file_path: Path = typer.Option(
+        ...,
+        "--file",
+        help="待上传文件路径。",
+    ),
+) -> None:
+    validated_file_path = _validate_file_path(file_path)
+
+    api = _build_api()
+    data = api.qywx.upload(center_id=center_id, file_path=validated_file_path)
+    print_json({"status": "ok", "data": data})

bocai_cli/commands/registry.py

@@ -0,0 +1,70 @@
+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.card import studio_app as card_studio_app
+from bocai_cli.commands.center import app as center_app
+from bocai_cli.commands.center import studio_app as center_studio_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.customreport import app as customreport_app
+from bocai_cli.commands.guanyuan import app as guanyuan_app
+from bocai_cli.commands.guanyuan import studio_app as guanyuan_studio_app
+from bocai_cli.commands.member import app as member_app
+from bocai_cli.commands.member import studio_app as member_studio_app
+from bocai_cli.commands.month_dashboard import app as monthdashboard_app
+from bocai_cli.commands.qywx import app as qywx_app
+from bocai_cli.commands.reservation import app as reservation_app
+from bocai_cli.commands.report import app as report_app
+from bocai_cli.product import Product
+
+
+@dataclass(frozen=True)
+class CommandRegistration:
+    name: str
+    app: typer.Typer
+    studio_app: typer.Typer | None = None
+    gym_enabled: bool = True
+    studio_enabled: bool = True
+    aliases: tuple[str, ...] = ()
+
+
+COMMAND_REGISTRY: tuple[CommandRegistration, ...] = (
+    CommandRegistration(name="auth", app=auth_app),
+    CommandRegistration(name="card", app=card_app, studio_app=card_studio_app),
+    CommandRegistration(name="center", app=center_app, studio_app=center_studio_app),
+    CommandRegistration(name="checkin", app=checkin_app),
+    CommandRegistration(name="course", app=course_app, studio_enabled=False),
+    CommandRegistration(name="customreport", app=customreport_app),
+    CommandRegistration(name="guanyuan", app=guanyuan_app, studio_app=guanyuan_studio_app),
+    CommandRegistration(
+        name="member",
+        app=member_app,
+        studio_app=member_studio_app,
+        gym_enabled=False,
+    ),
+    CommandRegistration(name="month-dashboard", app=monthdashboard_app),
+    CommandRegistration(name="qywx", app=qywx_app),
+    CommandRegistration(name="reservation", app=reservation_app),
+    CommandRegistration(name="report", app=report_app),
+)
+
+
+def register_commands(root_app: typer.Typer, *, product: Product = "gym") -> None:
+    for registration in COMMAND_REGISTRY:
+        if product == "gym" and not registration.gym_enabled:
+            continue
+        if product == "studio" and not registration.studio_enabled:
+            continue
+        command_app = (
+            registration.studio_app
+            if product == "studio" and registration.studio_app is not None
+            else registration.app
+        )
+        root_app.add_typer(command_app, name=registration.name)
+        for alias in registration.aliases:
+            root_app.add_typer(command_app, name=alias, hidden=True)