erp-integral-mcp 1.2.0__tar.gz

erp_integral_mcp-1.2.0/PKG-INFO

@@ -0,0 +1,88 @@
+Metadata-Version: 2.4
+Name: erp-integral-mcp
+Version: 1.2.0
+Summary: 鼎尖 ERP 业务动作积分统计 MCP：积分项查询、按日/按积分项汇总
+License: Proprietary
+Project-URL: Homepage, https://open.workbuddy.cn/docs/connector
+Project-URL: Documentation, https://open.workbuddy.cn/docs/connector
+Keywords: mcp,dingjian,erp,integral,settlement,stdio,uvx,workbuddy
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: mcp>=1.2.0
+Requires-Dist: httpx>=0.27.0
+Requires-Dist: pydantic>=2.0
+Requires-Dist: pydantic-settings>=2.0
+Requires-Dist: python-dotenv>=1.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.24; extra == "dev"
+
+# erp-integral-mcp
+
+鼎尖 ERP **业务动作积分**统计 MCP：第三方 SHA1 签名鉴权，查询积分项清单，并按**日历日**或**积分项**汇总每位员工的作业积分。
+
+## 接口
+
+| Tool | ERP API | 说明 |
+| --- | --- | --- |
+| `search_persons` | `basedata/person/dataPickerData` | 姓名/工号 → personId（F7） |
+| `list_integral_jobsets` | `settlement/integralDetail/getIntegralConditon` | 积分项（jobSet）清单 |
+| `query_integral_stats` | `settlement/integralDetail/integralCountData` | 人员积分；支持 `personId`；`group_by=week` |
+
+`columnType`：
+
+- `INTEGRALDATE`：按日，`scoreN` = 当月第 N 日积分
+- `INTEGRALJOBSET`：按积分项，字段名为积分项 `id`，需结合 `jobSetList` 解读
+
+## 环境变量
+
+与 `erp-property-mcp` 相同：
+
+```bash
+ERP_BASE_URL=https://your-host/dingjian/
+ERP_SECRET=...
+ERP_CU_ID=...
+ERP_USER_ID=...
+ERP_DC=...
+```
+
+## 本地运行
+
+```bash
+cd mcp/erp-integral-mcp
+cp .env.example .env   # 填入凭证
+python -m venv .venv && source .venv/bin/activate
+pip install -e ".[dev]"
+pytest
+erp-integral-mcp       # stdio MCP
+```
+
+## uvx / WorkBuddy
+
+```json
+{
+  "mcpServers": {
+    "erp_integral": {
+      "command": "uvx",
+      "args": ["erp-integral-mcp@1.0.0"],
+      "env": {
+        "ERP_BASE_URL": "${ERP_BASE_URL}",
+        "ERP_SECRET": "${ERP_SECRET}",
+        "ERP_CU_ID": "${ERP_CU_ID}",
+        "ERP_USER_ID": "${ERP_USER_ID}",
+        "ERP_DC": "${ERP_DC}"
+      }
+    }
+  }
+}
+```
+
+连接器打包：
+
+```bash
+./scripts/pack-connector.sh
+```
+
+## Skill
+
+业务作业 SOP 见 `skills/erp-integral-stats/SKILL.md`（仓库同步副本：`mcp/skills/erp-integral-stats/SKILL.md`）。

erp_integral_mcp-1.2.0/README.md

@@ -0,0 +1,69 @@
+# erp-integral-mcp
+
+鼎尖 ERP **业务动作积分**统计 MCP：第三方 SHA1 签名鉴权，查询积分项清单，并按**日历日**或**积分项**汇总每位员工的作业积分。
+
+## 接口
+
+| Tool | ERP API | 说明 |
+| --- | --- | --- |
+| `search_persons` | `basedata/person/dataPickerData` | 姓名/工号 → personId（F7） |
+| `list_integral_jobsets` | `settlement/integralDetail/getIntegralConditon` | 积分项（jobSet）清单 |
+| `query_integral_stats` | `settlement/integralDetail/integralCountData` | 人员积分；支持 `personId`；`group_by=week` |
+
+`columnType`：
+
+- `INTEGRALDATE`：按日，`scoreN` = 当月第 N 日积分
+- `INTEGRALJOBSET`：按积分项，字段名为积分项 `id`，需结合 `jobSetList` 解读
+
+## 环境变量
+
+与 `erp-property-mcp` 相同：
+
+```bash
+ERP_BASE_URL=https://your-host/dingjian/
+ERP_SECRET=...
+ERP_CU_ID=...
+ERP_USER_ID=...
+ERP_DC=...
+```
+
+## 本地运行
+
+```bash
+cd mcp/erp-integral-mcp
+cp .env.example .env   # 填入凭证
+python -m venv .venv && source .venv/bin/activate
+pip install -e ".[dev]"
+pytest
+erp-integral-mcp       # stdio MCP
+```
+
+## uvx / WorkBuddy
+
+```json
+{
+  "mcpServers": {
+    "erp_integral": {
+      "command": "uvx",
+      "args": ["erp-integral-mcp@1.0.0"],
+      "env": {
+        "ERP_BASE_URL": "${ERP_BASE_URL}",
+        "ERP_SECRET": "${ERP_SECRET}",
+        "ERP_CU_ID": "${ERP_CU_ID}",
+        "ERP_USER_ID": "${ERP_USER_ID}",
+        "ERP_DC": "${ERP_DC}"
+      }
+    }
+  }
+}
+```
+
+连接器打包：
+
+```bash
+./scripts/pack-connector.sh
+```
+
+## Skill
+
+业务作业 SOP 见 `skills/erp-integral-stats/SKILL.md`（仓库同步副本：`mcp/skills/erp-integral-stats/SKILL.md`）。

erp_integral_mcp-1.2.0/pyproject.toml

@@ -0,0 +1,40 @@
+[build-system]
+requires = ["setuptools>=68", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "erp-integral-mcp"
+version = "1.2.0"
+description = "鼎尖 ERP 业务动作积分统计 MCP：积分项查询、按日/按积分项汇总"
+readme = "README.md"
+requires-python = ">=3.10"
+license = { text = "Proprietary" }
+keywords = ["mcp", "dingjian", "erp", "integral", "settlement", "stdio", "uvx", "workbuddy"]
+dependencies = [
+    "mcp>=1.2.0",
+    "httpx>=0.27.0",
+    "pydantic>=2.0",
+    "pydantic-settings>=2.0",
+    "python-dotenv>=1.0.0",
+]
+
+[project.urls]
+Homepage = "https://open.workbuddy.cn/docs/connector"
+Documentation = "https://open.workbuddy.cn/docs/connector"
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.0",
+    "pytest-asyncio>=0.24",
+]
+
+[project.scripts]
+erp-integral-mcp = "erp_integral_mcp.__main__:main"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.pytest.ini_options]
+asyncio_mode = "auto"
+testpaths = ["tests"]
+pythonpath = ["tests", "src"]

erp_integral_mcp-1.2.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

erp_integral_mcp-1.2.0/src/erp_integral_mcp/__init__.py

@@ -0,0 +1,3 @@
+"""鼎尖 ERP 业务动作积分统计 MCP。"""
+
+__version__ = "1.2.0"

erp_integral_mcp-1.2.0/src/erp_integral_mcp/__main__.py

@@ -0,0 +1,9 @@
+from erp_integral_mcp.server import run
+
+
+def main() -> None:
+    run()
+
+
+if __name__ == "__main__":
+    main()

erp_integral_mcp-1.2.0/src/erp_integral_mcp/config.py

@@ -0,0 +1,59 @@
+from functools import lru_cache
+from pathlib import Path
+
+from dotenv import load_dotenv
+from pydantic import Field
+from pydantic_settings import BaseSettings, SettingsConfigDict
+
+from erp_integral_mcp.erp.auth import require_config
+
+PACKAGE_ROOT = Path(__file__).resolve().parents[2]
+ENV_FILE = PACKAGE_ROOT / ".env"
+
+GET_CONDITION_API = "settlement/integralDetail/getIntegralConditon"
+COUNT_DATA_API = "settlement/integralDetail/integralCountData"
+PERSON_PICKER_API = "basedata/person/dataPickerData"
+COLUMN_BY_DATE = "INTEGRALDATE"
+COLUMN_BY_JOBSET = "INTEGRALJOBSET"
+
+load_dotenv(ENV_FILE)
+
+
+class Settings(BaseSettings):
+    model_config = SettingsConfigDict(
+        env_file=str(ENV_FILE),
+        env_file_encoding="utf-8",
+        extra="ignore",
+    )
+
+    erp_base_url: str = Field(alias="ERP_BASE_URL")
+    erp_secret: str = Field(alias="ERP_SECRET")
+    erp_cu_id: str = Field(alias="ERP_CU_ID")
+    erp_user_id: str = Field(alias="ERP_USER_ID")
+    erp_dc: str = Field(alias="ERP_DC")
+    erp_timeout_seconds: int = Field(default=60, alias="ERP_TIMEOUT_SECONDS")
+
+    @property
+    def base_url(self) -> str:
+        return self.erp_base_url.rstrip("/") + "/"
+
+    def validate_credentials(self) -> None:
+        require_config(
+            {
+                "ERP_SECRET": self.erp_secret,
+                "ERP_CU_ID": self.erp_cu_id,
+                "ERP_USER_ID": self.erp_user_id,
+                "ERP_DC": self.erp_dc,
+            },
+            [
+                "ERP_SECRET",
+                "ERP_CU_ID",
+                "ERP_USER_ID",
+                "ERP_DC",
+            ],
+        )
+
+
+@lru_cache
+def get_settings() -> Settings:
+    return Settings()  # type: ignore[call-arg]

erp_integral_mcp-1.2.0/src/erp_integral_mcp/context.py

@@ -0,0 +1,17 @@
+from __future__ import annotations
+
+from functools import lru_cache
+
+from erp_integral_mcp.config import Settings, get_settings
+from erp_integral_mcp.erp.client import ErpClient
+
+
+class AppContext:
+    def __init__(self, settings: Settings) -> None:
+        self.settings = settings
+        self.client = ErpClient(settings)
+
+
+@lru_cache
+def get_context() -> AppContext:
+    return AppContext(get_settings())

erp_integral_mcp-1.2.0/src/erp_integral_mcp/erp/__init__.py

@@ -0,0 +1 @@
+"""ERP 第三方接口客户端。"""

erp_integral_mcp-1.2.0/src/erp_integral_mcp/erp/auth.py

@@ -0,0 +1,65 @@
+"""鼎尖 ERP 第三方签名鉴权（对齐《第三方接口规范》Java 示例）。"""
+
+from __future__ import annotations
+
+import hashlib
+import time
+from typing import Mapping
+
+
+def build_signature(
+    *,
+    secret: str,
+    third_timestamp: str,
+    third_cu_id: str,
+    third_user_id: str,
+    third_dc: str,
+    api_path: str,
+) -> str:
+    """将 secret/timestamp/cuId/userId/dc/apiPath 字典序排序后拼接，再做 SHA1 hex。"""
+    parts = [
+        secret,
+        third_timestamp,
+        third_cu_id,
+        third_user_id,
+        third_dc,
+        api_path,
+    ]
+    parts.sort()
+    return hashlib.sha1("".join(parts).encode("utf-8")).hexdigest()
+
+
+def build_auth_headers(
+    *,
+    secret: str,
+    third_cu_id: str,
+    third_user_id: str,
+    third_dc: str,
+    api_path: str,
+    timestamp_ms: str | None = None,
+) -> dict[str, str]:
+    """生成第三方开放接口所需请求头。"""
+    third_timestamp = timestamp_ms or str(int(time.time() * 1000))
+    signature = build_signature(
+        secret=secret,
+        third_timestamp=third_timestamp,
+        third_cu_id=third_cu_id,
+        third_user_id=third_user_id,
+        third_dc=third_dc,
+        api_path=api_path,
+    )
+    return {
+        "thirdtimestamp": third_timestamp,
+        "thirdCuId": third_cu_id,
+        "thirdUserId": third_user_id,
+        "thirdDc": third_dc,
+        "djthirdsign": signature,
+        "content-type": "application/json;charset=UTF-8",
+        "accept": "application/json, text/plain, */*",
+    }
+
+
+def require_config(values: Mapping[str, str | None], keys: list[str]) -> None:
+    missing = [key for key in keys if not (values.get(key) or "").strip()]
+    if missing:
+        raise ValueError("缺少 ERP 第三方凭证，请配置环境变量: " + ", ".join(missing))

erp_integral_mcp-1.2.0/src/erp_integral_mcp/erp/client.py

@@ -0,0 +1,93 @@
+from __future__ import annotations
+
+import asyncio
+import logging
+from typing import Any
+from urllib.parse import urljoin
+
+import httpx
+
+from erp_integral_mcp.config import (
+    COUNT_DATA_API,
+    GET_CONDITION_API,
+    PERSON_PICKER_API,
+    Settings,
+)
+from erp_integral_mcp.erp.auth import build_auth_headers
+
+logger = logging.getLogger(__name__)
+
+
+class ErpClient:
+    """调用鼎尖 ERP 第三方开放接口（签名鉴权，无需 Bearer Token）。"""
+
+    def __init__(self, settings: Settings) -> None:
+        self._settings = settings
+        settings.validate_credentials()
+
+    def _url(self, api_path: str) -> str:
+        base = self._settings.base_url
+        return urljoin(base, f"{api_path}?djrqformat=pcjson")
+
+    async def _post(self, api_path: str, body: dict[str, Any]) -> dict[str, Any]:
+        last_error: Exception | None = None
+        for attempt in range(3):
+            headers = build_auth_headers(
+                secret=self._settings.erp_secret,
+                third_cu_id=self._settings.erp_cu_id,
+                third_user_id=self._settings.erp_user_id,
+                third_dc=self._settings.erp_dc,
+                api_path=api_path,
+            )
+            try:
+                async with httpx.AsyncClient(
+                    timeout=self._settings.erp_timeout_seconds
+                ) as client:
+                    resp = await client.post(
+                        self._url(api_path),
+                        headers=headers,
+                        json=body,
+                    )
+                if resp.status_code >= 400:
+                    raise RuntimeError(
+                        f"ERP HTTP {resp.status_code} {api_path}: {resp.text[:300]}"
+                    )
+                data = resp.json()
+            except Exception as exc:  # noqa: BLE001
+                last_error = exc
+                await asyncio.sleep(0.6 * (attempt + 1))
+                continue
+
+            if not isinstance(data, dict):
+                raise RuntimeError(f"ERP 返回非 JSON 对象: {type(data)}")
+
+            if data.get("resultType") in (903, "903"):
+                last_error = RuntimeError(str(data.get("resultMsg") or "QPS限流"))
+                await asyncio.sleep(0.8 * (attempt + 1))
+                continue
+
+            if data.get("resultType") not in (None, 1):
+                raise RuntimeError(
+                    f"ERP 业务失败 resultType={data.get('resultType')} "
+                    f"resultMsg={data.get('resultMsg')}"
+                )
+            return data
+
+        if last_error:
+            raise RuntimeError(f"ERP 请求失败: {last_error}") from last_error
+        raise RuntimeError("ERP 请求失败")
+
+    async def get_integral_condition(
+        self, *, search_type: str = "integralCount"
+    ) -> dict[str, Any]:
+        return await self._post(
+            GET_CONDITION_API,
+            {"searchType": search_type},
+        )
+
+    async def integral_count_data(self, body: dict[str, Any]) -> dict[str, Any]:
+        return await self._post(COUNT_DATA_API, body)
+
+    async def search_persons(self, body: dict[str, Any]) -> dict[str, Any]:
+        """人员 F7：basedata/person/dataPickerData（与积分页员工选择器同源）。"""
+        return await self._post(PERSON_PICKER_API, body)