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

bocai_cli/__init__.py

@@ -0,0 +1,3 @@
+__all__ = ["__version__"]
+
+__version__ = "0.2.0"

bocai_cli/api/__init__.py

@@ -0,0 +1,29 @@
+from bocai_cli.api.auth_client import AuthClient
+from bocai_cli.api.card_client import CardClient
+from bocai_cli.api.center_client import CenterClient
+from bocai_cli.api.checkin_client import CheckinClient
+from bocai_cli.api.course_client import CourseClient
+from bocai_cli.api.facade import ApiFacade
+from bocai_cli.api.guanyuan_client import GuanyuanClient
+from bocai_cli.api.member_client import MemberClient
+from bocai_cli.api.month_dashboard_client import MonthDashboardClient
+from bocai_cli.api.qywx_client import QywxClient
+from bocai_cli.api.report_client import ReportClient
+from bocai_cli.api.reservation_client import ReservationClient
+from bocai_cli.api.transport import ApiTransport
+
+__all__ = [
+    "ApiFacade",
+    "ApiTransport",
+    "AuthClient",
+    "CardClient",
+    "CenterClient",
+    "CheckinClient",
+    "CourseClient",
+    "GuanyuanClient",
+    "MemberClient",
+    "MonthDashboardClient",
+    "QywxClient",
+    "ReservationClient",
+    "ReportClient",
+]

bocai_cli/api/auth_client.py

@@ -0,0 +1,57 @@
+from __future__ import annotations
+
+from typing import Any
+
+from bocai_cli.api.transport import ApiTransport
+from bocai_cli.errors import ApiRequestError
+
+
+def _extract_access_token(payload: object) -> str | None:
+    if isinstance(payload, dict):
+        for field in ("accessToken", "access_token"):
+            value = payload.get(field)
+            if isinstance(value, str) and value.strip():
+                return value.strip()
+    return None
+
+
+class AuthClient:
+    def __init__(self, transport: ApiTransport) -> None:
+        self._transport = transport
+
+    def start(self) -> Any:
+        return self._transport.request(
+            "POST",
+            "/api/auth/start",
+            auth_required=False,
+        )
+
+    def check_status(self, *, auth_code: str) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/auth/status",
+            params={"auth_code": auth_code},
+            auth_required=False,
+        )
+
+    def refresh_user_token(self, *, access_token: str, refresh_token: str) -> Any:
+        return self._transport.request(
+            "POST",
+            "/api/auth/refreshUserToken",
+            json_body={"accessToken": access_token, "refreshToken": refresh_token},
+            auth_required=False,
+        )
+
+    def fetch_access_token(self, *, token_url: str, app_id: str, secret: str) -> str:
+        payload = self._transport.request(
+            "POST",
+            token_url,
+            json_body={"appId": app_id, "appSecret": secret},
+            absolute_url=True,
+            auth_required=False,
+        )
+        value = _extract_access_token(payload)
+        if value is not None:
+            return value
+
+        raise ApiRequestError("token 接口响应缺少 accessToken 或 access_token。")

bocai_cli/api/card_client.py

@@ -0,0 +1,222 @@
+from __future__ import annotations
+
+from typing import Any
+
+from bocai_cli.api.transport import ApiTransport
+
+
+class CardClient:
+    def __init__(self, transport: ApiTransport) -> None:
+        self._transport = transport
+
+    def sale_performance(
+        self,
+        *,
+        start_date: str,
+        end_date: str,
+        page_num: int,
+        page_size: int,
+        type_: int,
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] = {
+            "startDate": start_date,
+            "endDate": end_date,
+            "pageNum": page_num,
+            "pageSize": page_size,
+            "type": type_,
+        }
+        if center_id is not None:
+            params["centerId"] = center_id
+        return self._transport.request("GET", "/api/bocai/card/sale", params=params)
+
+    def performance_clear(
+        self,
+        *,
+        transaction_ids: list[str],
+        phone: str,
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] | None = None
+        if center_id is not None:
+            params = {"centerId": center_id}
+
+        return self._transport.request(
+            "POST",
+            "/api/bocai/course/performance/clear",
+            params=params,
+            json_body={"transactionIds": transaction_ids, "phone": phone},
+        )
+
+    def performance_update(
+        self,
+        *,
+        phone: str,
+        transactions: list[dict[str, Any]],
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] | None = None
+        if center_id is not None:
+            params = {"centerId": center_id}
+
+        return self._transport.request(
+            "POST",
+            "/api/bocai/course/performance/update",
+            params=params,
+            json_body={"phone": phone, "transactions": transactions},
+        )
+
+    def performance_new(
+        self,
+        *,
+        type_: int,
+        payload: dict[str, Any],
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] = {"type": type_}
+        if center_id is not None:
+            params["centerId"] = center_id
+
+        return self._transport.request(
+            "POST",
+            "/api/bocai/card/performance/new",
+            params=params,
+            json_body=payload,
+        )
+
+    def studio_contract_list(
+        self,
+        *,
+        center_id: str,
+        page_num: int,
+        page_size: int,
+        type_: int,
+        card_type: int = -1,
+        card_ids: list[int] | None = None,
+        status: int = 99,
+        remain_start: int | None = None,
+        remain_end: int | None = None,
+        sale_type: int = -1,
+        contract_status: int = -1,
+        seller_id: int = -1,
+        tag_id: int = -1,
+        card_create_start_date: str = "",
+        card_create_end_date: str = "",
+        card_activate_start_date: str = "",
+        card_activate_end_date: str = "",
+        card_expire_start_date: str = "",
+        card_expire_end_date: str = "",
+        course_start_date: str = "",
+        course_end_date: str = "",
+    ) -> Any:
+        json_body: dict[str, Any] = {
+            "pageNum": page_num,
+            "pageSize": page_size,
+            "keyword": "",
+            "displayCenter": False,
+            "cardType": card_type,
+            "cardIds": card_ids or [],
+            "status": status,
+            "consultantId": seller_id,
+            "followUpConsultantId": -1,
+            "saleType": sale_type,
+            "cardCreateTimeStart": card_create_start_date,
+            "cardCreateTimeEnd": card_create_end_date,
+            "cardActiveTimeStart": card_activate_start_date,
+            "cardActiveTimeEnd": card_activate_end_date,
+            "expireTimeStart": card_expire_start_date,
+            "expireTimeEnd": card_expire_end_date,
+            "electronicContractStatus": str(contract_status),
+            "isPrimary": -1,
+            "tagId": tag_id,
+            "courseStartTime": course_start_date,
+            "courseEndTime": course_end_date,
+            "searchRemain": remain_start is not None or remain_end is not None,
+            "remainStart": "" if remain_start is None else str(remain_start),
+            "remainEnd": "" if remain_end is None else remain_end,
+        }
+        if type_ == 1:
+            json_body["isCourseCard"] = False
+        elif type_ == 2:
+            json_body["isCourseCard"] = True
+
+        return self._transport.request(
+            "POST",
+            "/api/bocai/card/studio_contract_list",
+            params={
+                "centerId": center_id,
+                "pageNum": page_num,
+                "pageSize": page_size,
+            },
+            json_body=json_body,
+        )
+
+    def contract_detail(self, *, center_id: str, contract_id: str) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/card/contract_detail",
+            params={"centerId": center_id, "contractId": contract_id},
+        )
+
+    def studio_sale(
+        self,
+        *,
+        center_id: str,
+        start_date: str,
+        end_date: str,
+        type_: int,
+        seller_id: int = -1,
+        card_ids: list[int] | None = None,
+        tag_ids: list[int] | None = None,
+        page_num: int = 0,
+        page_size: int = 20,
+    ) -> Any:
+        json_body: dict[str, Any] = {
+            "startDate": start_date,
+            "endDate": end_date,
+            "tagIds": tag_ids or [],
+            "cardIds": card_ids or [],
+        }
+        if type_ == 1:
+            json_body["isCourseCard"] = False
+        elif type_ == 2:
+            json_body["isCourseCard"] = True
+
+        return self._transport.request(
+            "POST",
+            "/api/bocai/card/studio_sale",
+            params={
+                "centerId": center_id,
+                "consultantId": seller_id,
+                "pageNum": page_num,
+                "pageSize": page_size,
+            },
+            json_body=json_body,
+        )
+
+    def studio_private_consume_performance(
+        self,
+        *,
+        center_id: str,
+        start_date: str,
+        end_date: str,
+        trainer_id: int = -1,
+        teach_mode: int = -1,
+        page_num: int = 0,
+        page_size: int = 20,
+    ) -> Any:
+        return self._transport.request(
+            "POST",
+            "/api/bocai/card/studio_private_consume_performance",
+            params={
+                "centerId": center_id,
+                "pageNum": page_num,
+                "pageSize": page_size,
+            },
+            json_body={
+                "trainerId": trainer_id,
+                "startDate": start_date,
+                "endDate": end_date,
+                "teachMode": teach_mode,
+            },
+        )

bocai_cli/api/center_client.py

@@ -0,0 +1,55 @@
+from __future__ import annotations
+
+from typing import Any
+
+from bocai_cli.api.transport import ApiTransport
+
+
+class CenterClient:
+    def __init__(self, transport: ApiTransport) -> None:
+        self._transport = transport
+
+    def list(self) -> Any:
+        return self._transport.request("GET", "/api/bocai/centers")
+
+    def manager_phone(self, *, center_id: str) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/managerPhones",
+            params={"centerId": center_id},
+        )
+
+    def card_list(self, *, center_id: str, type_: int) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/center/card/list",
+            params={"centerId": center_id, "type": type_},
+        )
+
+    def source_list(self, *, center_id: str) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/center/source/list",
+            params={"centerId": center_id},
+        )
+
+    def studio_seller_list(self, *, center_id: str) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/center/consultant/list",
+            params={"centerId": center_id},
+        )
+
+    def studio_trainer_list(self, *, center_id: str) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/center/trainer/list",
+            params={"centerId": center_id},
+        )
+
+    def tag_list(self, *, center_id: str, type_: int) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/center/tag/list",
+            params={"centerId": center_id, "type": type_},
+        )

bocai_cli/api/checkin_client.py

@@ -0,0 +1,29 @@
+from __future__ import annotations
+
+from typing import Any
+
+from bocai_cli.api.transport import ApiTransport
+
+
+class CheckinClient:
+    def __init__(self, transport: ApiTransport) -> None:
+        self._transport = transport
+
+    def list(
+        self,
+        *,
+        start_date: str,
+        end_date: str,
+        page_num: int,
+        page_size: int,
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] = {
+            "startDate": start_date,
+            "endDate": end_date,
+            "pageNum": page_num,
+            "pageSize": page_size,
+        }
+        if center_id is not None:
+            params["centerId"] = center_id
+        return self._transport.request("GET", "/api/bocai/checkin", params=params)

bocai_cli/api/course_client.py

@@ -0,0 +1,85 @@
+from __future__ import annotations
+
+from typing import Any
+
+from bocai_cli.api.transport import ApiTransport
+
+
+class CourseClient:
+    def __init__(self, transport: ApiTransport) -> None:
+        self._transport = transport
+
+    def sale_performance(
+        self,
+        *,
+        start_date: str,
+        end_date: str,
+        page_num: int,
+        page_size: int,
+        type_: int,
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] = {
+            "startDate": start_date,
+            "endDate": end_date,
+            "pageNum": page_num,
+            "pageSize": page_size,
+            "type": type_,
+        }
+        if center_id is not None:
+            params["centerId"] = center_id
+        return self._transport.request("GET", "/api/bocai/course/sale", params=params)
+
+    def performance_clear(
+        self,
+        *,
+        transaction_ids: list[str],
+        phone: str,
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] | None = None
+        if center_id is not None:
+            params = {"centerId": center_id}
+
+        return self._transport.request(
+            "POST",
+            "/api/bocai/course/performance/clear",
+            params=params,
+            json_body={"transactionIds": transaction_ids, "phone": phone},
+        )
+
+    def performance_update(
+        self,
+        *,
+        phone: str,
+        transactions: list[dict[str, Any]],
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] | None = None
+        if center_id is not None:
+            params = {"centerId": center_id}
+
+        return self._transport.request(
+            "POST",
+            "/api/bocai/course/performance/update",
+            params=params,
+            json_body={"phone": phone, "transactions": transactions},
+        )
+
+    def performance_new(
+        self,
+        *,
+        type_: int,
+        payload: dict[str, Any],
+        center_id: str | None = None,
+    ) -> Any:
+        params: dict[str, Any] = {"type": type_}
+        if center_id is not None:
+            params["centerId"] = center_id
+
+        return self._transport.request(
+            "POST",
+            "/api/bocai/course/performance/new",
+            params=params,
+            json_body=payload,
+        )

bocai_cli/api/facade.py

@@ -0,0 +1,40 @@
+from __future__ import annotations
+
+from bocai_cli.api.auth_client import AuthClient
+from bocai_cli.api.card_client import CardClient
+from bocai_cli.api.center_client import CenterClient
+from bocai_cli.api.checkin_client import CheckinClient
+from bocai_cli.api.course_client import CourseClient
+from bocai_cli.api.guanyuan_client import GuanyuanClient
+from bocai_cli.api.member_client import MemberClient
+from bocai_cli.api.month_dashboard_client import MonthDashboardClient
+from bocai_cli.api.qywx_client import QywxClient
+from bocai_cli.api.report_client import ReportClient
+from bocai_cli.api.reservation_client import ReservationClient
+from bocai_cli.api.transport import ApiTransport
+
+
+class ApiFacade:
+    def __init__(
+        self,
+        *,
+        base_url: str | None,
+        token: str | None,
+        timeout_seconds: float = 30.0,
+    ) -> None:
+        self.transport = ApiTransport(
+            base_url=base_url,
+            token=token,
+            timeout_seconds=timeout_seconds,
+        )
+        self.auth = AuthClient(self.transport)
+        self.card = CardClient(self.transport)
+        self.center = CenterClient(self.transport)
+        self.checkin = CheckinClient(self.transport)
+        self.course = CourseClient(self.transport)
+        self.guanyuan = GuanyuanClient(self.transport)
+        self.member = MemberClient(self.transport)
+        self.monthdashboard = MonthDashboardClient(self.transport)
+        self.qywx = QywxClient(self.transport)
+        self.reservation = ReservationClient(self.transport)
+        self.report = ReportClient(self.transport)

bocai_cli/api/guanyuan_client.py

@@ -0,0 +1,50 @@
+from __future__ import annotations
+
+from typing import Any
+
+from bocai_cli.api.transport import ApiTransport
+
+
+class GuanyuanClient:
+    def __init__(self, transport: ApiTransport) -> None:
+        self._transport = transport
+
+    def absence_statistics(self, *, center_id: str, type_: int = 1) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/guanyuan/absence/statistics",
+            params={"centerId": center_id, "type": type_},
+        )
+
+    def absence_detail(self, *, center_id: str, type_: int = 1) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/guanyuan/absence/detail",
+            params={"centerId": center_id, "type": type_},
+        )
+
+    def ds_list(self, *, center_id: str) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/guanyuan/ds/list",
+            params={"centerId": center_id},
+        )
+
+    def ds_data(
+        self,
+        *,
+        center_id: str,
+        ds_id: str,
+        start_date: str,
+        end_date: str,
+    ) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/guanyuan/ds/data",
+            params={
+                "centerId": center_id,
+                "dsId": ds_id,
+                "startDate": start_date,
+                "endDate": end_date,
+            },
+        )

bocai_cli/api/member_client.py

@@ -0,0 +1,97 @@
+from __future__ import annotations
+
+from typing import Any
+
+from bocai_cli.api.transport import ApiTransport
+
+
+class MemberClient:
+    def __init__(self, transport: ApiTransport) -> None:
+        self._transport = transport
+
+    def studio_proposed_list(
+        self,
+        *,
+        center_id: str,
+        page_num: int,
+        page_size: int,
+        consultant_id: int = -1,
+        source: int = -1,
+        tag_id: int = -1,
+        follow_status: int = -1,
+        attend_start_date: str = "",
+        attend_end_date: str = "",
+    ) -> Any:
+        return self._transport.request(
+            "POST",
+            "/api/bocai/member/studio_proposed_list",
+            params={
+                "centerId": center_id,
+                "pageNum": page_num,
+                "pageSize": page_size,
+            },
+            json_body={
+                "consultantId": consultant_id,
+                "source": source,
+                "tagId": tag_id,
+                "followupStatus": str(follow_status),
+                "attendStartDate": attend_start_date,
+                "attendEndDate": attend_end_date,
+            },
+        )
+
+    def studio_member_list(
+        self,
+        *,
+        center_id: str,
+        page_num: int,
+        page_size: int,
+        consultant_id: int = -1,
+        source: int = -1,
+        tag_id: int = -1,
+        member_type: int = -1,
+        status: int = -1,
+        follow_status: int = -1,
+        expire_start_date: str = "",
+        expire_end_date: str = "",
+        birthday_start_date: str = "",
+        birthday_end_date: str = "",
+        check_in_status: bool = True,
+        last_checkin_start_date: str = "",
+        last_checkin_end_date: str = "",
+        course_start_date: str = "",
+        course_end_date: str = "",
+    ) -> Any:
+        return self._transport.request(
+            "POST",
+            "/api/bocai/member/studio_member_list",
+            params={
+                "centerId": center_id,
+                "pageNum": page_num,
+                "pageSize": page_size,
+            },
+            json_body={
+                "consultantId": consultant_id,
+                "status": status,
+                "birthdayStart": birthday_start_date,
+                "birthDayEnd": birthday_end_date,
+                "lastCheckInStartTs": last_checkin_start_date,
+                "lastCheckInEndTs": last_checkin_end_date,
+                "checkInStatus": check_in_status,
+                "expireStartTime": expire_start_date,
+                "expireEndTime": expire_end_date,
+                "memberType": None if member_type == -1 else member_type,
+                "source": source,
+                "tagId": tag_id,
+                "followupStatus": None if follow_status == -1 else follow_status,
+                "courseStartTime": course_start_date,
+                "courseEndTime": course_end_date,
+            },
+        )
+
+    def detail(self, *, center_id: str, member_id: str) -> Any:
+        return self._transport.request(
+            "GET",
+            "/api/bocai/member/detail",
+            params={"centerId": center_id, "traineeId": member_id},
+        )