zerone-cli 0.1.1__tar.gz → 0.1.3__tar.gz

zerone_cli-0.1.3/PKG-INFO

@@ -0,0 +1,169 @@
+Metadata-Version: 2.4
+Name: zerone-cli
+Version: 0.1.3
+Summary: Zerone 私募股权数据 CLI 工具
+Author-email: Zerone <tech@zerone.com.cn>
+License: MIT
+Project-URL: Homepage, https://ai.zerone.com.cn
+Project-URL: Documentation, https://ai.zerone.com.cn
+Keywords: cli,private-equity,finance,mcp,zerone
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Operating System :: OS Independent
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Financial and Insurance Industry
+Classifier: Topic :: Office/Business :: Financial
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: typer>=0.9.0
+Requires-Dist: httpx>=0.24.0
+Requires-Dist: pydantic>=2.0.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: pandas>=2.0.0
+
+# Zerone CLI
+
+Zerone 私募股权数据命令行工具，通过 API Key 直接查询 Zerone 数据平台。
+
+## 安装
+
+```bash
+pip install zerone-cli
+```
+
+## 快速开始
+
+### 1. 配置 API Key
+
+```bash
+zerone config set-api-key YOUR_API_KEY
+```
+
+> 如何获取 API Key？请访问 [https://ai.zerone.com.cn](https://ai.zerone.com.cn) 注册账号后申请。
+
+### 2. 查看帮助
+
+```bash
+zerone --help
+```
+
+### 3. 开始查询
+
+```bash
+# 实体模糊搜索
+zerone entity-search "字节跳动"
+
+# 查询实体综合画像
+zerone entity-profile "中国石油"
+
+# 查询实体财务数据列表
+zerone entity-financial-list "中国石油" --page 1 --page-size 20
+
+# 查询管理基金列表
+zerone entity-managed-funds-list "深创投" --page 1 --page-size 20
+
+# 查询退出列表
+zerone entity-exit-list "深创投" --exit-type IPO --page 1 --page-size 20
+
+# 查询投资列表
+zerone entity-invest-list "深创投" --invest-type 直接投资 --page 1 --page-size 20
+
+# 查询融资概率预测
+zerone model-financing-probability "字节跳动" --type COMPANY
+
+# 查询机构匹配推荐
+zerone model-company-match-gp "字节跳动" --page 1 --page-size 5
+
+# 查询市场投融资事件
+zerone evt-company-financing --start-date 2024-01-01 --end-date 2024-12-31 --province 北京 --city 北京市 --tag 人工智能 --page 1 --page-size 20
+
+# 查询LP出资事件
+zerone evt-lp-investment --start-date 2024-01-01 --end-date 2024-12-31 --province 广东 --city 深圳 --page 1 --page-size 20
+
+# 查询老股转让事件
+zerone evt-company-secondary --start-date 2024-01-01 --end-date 2024-12-31 --province 上海 --tag 半导体 --page 1 --page-size 20
+
+# 查询回购事件
+zerone evt-company-repurchase --start-date 2024-01-01 --end-date 2024-12-31 --province 浙江 --tag 医疗健康 --page 1 --page-size 20
+
+# 查询并购事件
+zerone evt-ma-company-deal --start-date 2024-01-01 --end-date 2024-12-31 --province 北京 --tag 新能源 --page 1 --page-size 20
+
+# 查询IPO事件
+zerone evt-company-ipo --start-date 2024-01-01 --end-date 2024-12-31 --province 广东 --tag 新材料 --page 1 --page-size 20
+
+# 查询解禁事件
+zerone evt-listing-company-unlock --start-date 2024-01-01 --end-date 2024-12-31 --province 江苏 --tag 高端制造 --page 1 --page-size 20
+
+# 查询减持事件
+zerone evt-listing-company-reduction --start-date 2024-01-01 --end-date 2024-12-31 --province 上海 --tag 集成电路 --page 1 --page-size 20
+
+# 查询S基金交易事件
+zerone evt-s-fund-deal --start-date 2024-01-01 --end-date 2024-12-31 --province 北京 --page 1 --page-size 20
+```
+
+## 命令列表
+
+| 命令 | 说明 |
+|------|------|
+| `zerone config set-api-key` | 设置 API Key |
+| `zerone config set-server-url` | 设置服务器地址（可选） |
+| `zerone config show` | 查看当前配置 |
+| `zerone entity-search` | 实体模糊搜索 |
+| `zerone entity-profile` | 查询实体综合画像 |
+| `zerone entity-financial-list` | 查询实体财务数据列表（分页） |
+| `zerone entity-managed-funds-list` | 查询管理基金列表（分页） |
+| `zerone entity-exit-list` | 查询退出列表（分页） |
+| `zerone entity-invest-list` | 查询投资列表（分页） |
+| `zerone model-financing-probability` | 查询融资概率预测模型 |
+| `zerone model-company-match-gp` | 基于推荐引擎为项目公司匹配潜在投资机构 |
+| `zerone evt-company-financing` | 查询市场投融资事件列表 |
+| `zerone evt-lp-investment` | 查询LP出资事件列表 |
+| `zerone evt-company-secondary` | 查询老股转让事件列表 |
+| `zerone evt-company-repurchase` | 查询回购事件列表 |
+| `zerone evt-ma-company-deal` | 查询并购事件列表 |
+| `zerone evt-company-ipo` | 查询IPO事件列表 |
+| `zerone evt-listing-company-unlock` | 查询解禁事件列表 |
+| `zerone evt-listing-company-reduction` | 查询减持事件列表 |
+| `zerone evt-s-fund-deal` | 查询S基金交易事件列表 |
+
+## 常用参数说明
+
+### 日期参数
+- `--start-date`: 开始日期，格式 YYYY-MM-DD
+- `--end-date`: 结束日期，格式 YYYY-MM-DD
+
+### 地区参数
+- `--province`: 省级行政区划
+- `--city`: 地级行政区划
+- `--county`: 县级行政区划
+
+### 分页参数
+- `--page`: 页码，从1开始，默认1
+- `--page-size`: 每页条数，默认20，上限20
+
+### 其他参数
+- `--tag`: 公司标签（经营业务类）
+- `--output`: 输出格式，可选 table/json/csv，默认table
+
+## 配置说明
+
+配置文件保存在 `~/.zerone/config.yaml`，支持以下环境变量覆盖：
+
+| 环境变量 | 说明 |
+|----------|------|
+| `ZERONE_API_KEY` | API Key |
+| `ZERONE_SERVER_URL` | 服务器地址 |
+
+## 系统要求
+
+- Python 3.9+
+- 有效的 Zerone API Key
+
+## 许可证
+
+MIT

zerone_cli-0.1.3/README.md

@@ -0,0 +1,142 @@
+# Zerone CLI
+
+Zerone 私募股权数据命令行工具，通过 API Key 直接查询 Zerone 数据平台。
+
+## 安装
+
+```bash
+pip install zerone-cli
+```
+
+## 快速开始
+
+### 1. 配置 API Key
+
+```bash
+zerone config set-api-key YOUR_API_KEY
+```
+
+> 如何获取 API Key？请访问 [https://ai.zerone.com.cn](https://ai.zerone.com.cn) 注册账号后申请。
+
+### 2. 查看帮助
+
+```bash
+zerone --help
+```
+
+### 3. 开始查询
+
+```bash
+# 实体模糊搜索
+zerone entity-search "字节跳动"
+
+# 查询实体综合画像
+zerone entity-profile "中国石油"
+
+# 查询实体财务数据列表
+zerone entity-financial-list "中国石油" --page 1 --page-size 20
+
+# 查询管理基金列表
+zerone entity-managed-funds-list "深创投" --page 1 --page-size 20
+
+# 查询退出列表
+zerone entity-exit-list "深创投" --exit-type IPO --page 1 --page-size 20
+
+# 查询投资列表
+zerone entity-invest-list "深创投" --invest-type 直接投资 --page 1 --page-size 20
+
+# 查询融资概率预测
+zerone model-financing-probability "字节跳动" --type COMPANY
+
+# 查询机构匹配推荐
+zerone model-company-match-gp "字节跳动" --page 1 --page-size 5
+
+# 查询市场投融资事件
+zerone evt-company-financing --start-date 2024-01-01 --end-date 2024-12-31 --province 北京 --city 北京市 --tag 人工智能 --page 1 --page-size 20
+
+# 查询LP出资事件
+zerone evt-lp-investment --start-date 2024-01-01 --end-date 2024-12-31 --province 广东 --city 深圳 --page 1 --page-size 20
+
+# 查询老股转让事件
+zerone evt-company-secondary --start-date 2024-01-01 --end-date 2024-12-31 --province 上海 --tag 半导体 --page 1 --page-size 20
+
+# 查询回购事件
+zerone evt-company-repurchase --start-date 2024-01-01 --end-date 2024-12-31 --province 浙江 --tag 医疗健康 --page 1 --page-size 20
+
+# 查询并购事件
+zerone evt-ma-company-deal --start-date 2024-01-01 --end-date 2024-12-31 --province 北京 --tag 新能源 --page 1 --page-size 20
+
+# 查询IPO事件
+zerone evt-company-ipo --start-date 2024-01-01 --end-date 2024-12-31 --province 广东 --tag 新材料 --page 1 --page-size 20
+
+# 查询解禁事件
+zerone evt-listing-company-unlock --start-date 2024-01-01 --end-date 2024-12-31 --province 江苏 --tag 高端制造 --page 1 --page-size 20
+
+# 查询减持事件
+zerone evt-listing-company-reduction --start-date 2024-01-01 --end-date 2024-12-31 --province 上海 --tag 集成电路 --page 1 --page-size 20
+
+# 查询S基金交易事件
+zerone evt-s-fund-deal --start-date 2024-01-01 --end-date 2024-12-31 --province 北京 --page 1 --page-size 20
+```
+
+## 命令列表
+
+| 命令 | 说明 |
+|------|------|
+| `zerone config set-api-key` | 设置 API Key |
+| `zerone config set-server-url` | 设置服务器地址（可选） |
+| `zerone config show` | 查看当前配置 |
+| `zerone entity-search` | 实体模糊搜索 |
+| `zerone entity-profile` | 查询实体综合画像 |
+| `zerone entity-financial-list` | 查询实体财务数据列表（分页） |
+| `zerone entity-managed-funds-list` | 查询管理基金列表（分页） |
+| `zerone entity-exit-list` | 查询退出列表（分页） |
+| `zerone entity-invest-list` | 查询投资列表（分页） |
+| `zerone model-financing-probability` | 查询融资概率预测模型 |
+| `zerone model-company-match-gp` | 基于推荐引擎为项目公司匹配潜在投资机构 |
+| `zerone evt-company-financing` | 查询市场投融资事件列表 |
+| `zerone evt-lp-investment` | 查询LP出资事件列表 |
+| `zerone evt-company-secondary` | 查询老股转让事件列表 |
+| `zerone evt-company-repurchase` | 查询回购事件列表 |
+| `zerone evt-ma-company-deal` | 查询并购事件列表 |
+| `zerone evt-company-ipo` | 查询IPO事件列表 |
+| `zerone evt-listing-company-unlock` | 查询解禁事件列表 |
+| `zerone evt-listing-company-reduction` | 查询减持事件列表 |
+| `zerone evt-s-fund-deal` | 查询S基金交易事件列表 |
+
+## 常用参数说明
+
+### 日期参数
+- `--start-date`: 开始日期，格式 YYYY-MM-DD
+- `--end-date`: 结束日期，格式 YYYY-MM-DD
+
+### 地区参数
+- `--province`: 省级行政区划
+- `--city`: 地级行政区划
+- `--county`: 县级行政区划
+
+### 分页参数
+- `--page`: 页码，从1开始，默认1
+- `--page-size`: 每页条数，默认20，上限20
+
+### 其他参数
+- `--tag`: 公司标签（经营业务类）
+- `--output`: 输出格式，可选 table/json/csv，默认table
+
+## 配置说明
+
+配置文件保存在 `~/.zerone/config.yaml`，支持以下环境变量覆盖：
+
+| 环境变量 | 说明 |
+|----------|------|
+| `ZERONE_API_KEY` | API Key |
+| `ZERONE_SERVER_URL` | 服务器地址 |
+
+## 系统要求
+
+- Python 3.9+
+- 有效的 Zerone API Key
+
+## 许可证
+
+MIT

zerone_cli-0.1.3/cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.0.0"

zerone_cli-0.1.3/cli/client.py

@@ -0,0 +1,176 @@
+from typing import Optional
+
+import httpx
+
+from cli.config import config_manager
+
+
+class MCPClient:
+
+    def __init__(self):
+        self.api_key = config_manager.get_api_key()
+        self.server_url = config_manager.get_server_url()
+        self.client = httpx.Client(timeout=30.0)
+
+    def _request(
+        self,
+        method: str,
+        path: str,
+        json_data: Optional[dict] = None,
+        params: Optional[dict] = None,
+    ) -> dict:
+        url = f"{self.server_url}{path}"
+        headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json",
+        }
+
+        try:
+            response = self.client.request(
+                method=method,
+                url=url,
+                headers=headers,
+                json=json_data,
+                params=params,
+            )
+            response.raise_for_status()
+            data = response.json()
+            if not data.get("success", True):
+                error = data.get("error", {})
+                raise ValueError(f"[{error.get('code', 'UNKNOWN')}] {error.get('message', '未知错误')}")
+            return data
+        except httpx.HTTPStatusError as e:
+            try:
+                error_data = e.response.json()
+                detail = error_data.get("detail", "")
+                if detail:
+                    raise ValueError(f"[HTTP {e.response.status_code}] {detail}")
+            except ValueError:
+                raise
+            except Exception:
+                pass
+            raise ValueError(f"请求失败: {e.response.status_code} - {e.response.text}")
+        except httpx.ConnectError:
+            raise ValueError(f"无法连接到 MCP Server，请检查 server_url 配置：{self.server_url}")
+        except httpx.TimeoutException:
+            raise ValueError("请求超时，请检查网络连接或稍后重试")
+
+    def entity_search(self, keyword: str) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_search",
+            json_data={"keyword": keyword},
+        )
+
+    def entity_profile(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_profile",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+            },
+        )
+
+    def entity_financial_list(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_financial_list",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def entity_managed_funds_list(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_managed_funds_list",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def entity_exit_list(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+        exit_type: Optional[str] = None,
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_exit_list",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+                "exit_type": exit_type or "",
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def entity_invest_list(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+        invest_type: Optional[str] = None,
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/entity_invest_list",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+                "invest_type": invest_type or "",
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def model_financing_probability(
+        self,
+        entity_name: str,
+        entity_type: Optional[str] = None,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/model_financing_probability",
+            json_data={
+                "entity_name": entity_name,
+                "entity_type": entity_type or "",
+            },
+        )
+
+    def close(self):
+        self.client.close()
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        self.close()

zerone_cli-0.1.3/cli/commands/__init__.py

@@ -0,0 +1,19 @@
+from cli.commands import config_cmd
+from cli.commands.entity_search import entity_search
+from cli.commands.entity_profile import entity_profile
+from cli.commands.entity_financial_list import entity_financial_list
+from cli.commands.entity_managed_funds_list import entity_managed_funds_list
+from cli.commands.entity_exit_list import entity_exit_list
+from cli.commands.entity_invest_list import entity_invest_list
+from cli.commands.model_financing_probability import model_financing_probability
+
+__all__ = [
+    "config_cmd",
+    "entity_search",
+    "entity_profile",
+    "entity_financial_list",
+    "entity_managed_funds_list",
+    "entity_exit_list",
+    "entity_invest_list",
+    "model_financing_probability",
+]

zerone_cli-0.1.3/cli/commands/basic_info.py

@@ -0,0 +1,77 @@
+"""
+查询基本信息命令模块
+"""
+
+import asyncio
+from typing import Optional
+
+import typer
+
+from cli.config import config_manager
+from cli.utils.output import print_result
+from mcp_server.tools.executor import ToolExecutor
+from mcp_server.router.entity_router import EntityRouter
+from mcp_server.router.type_router import TypeRouter
+from mcp_server.services.entity_service import EntityService
+from mcp_server.utils.field_mapper import FieldMapper
+from mcp_server.openapi_client.client import OpenAPIClient
+
+
+async def execute(entity_name: str, entity_type: Optional[str] = None) -> dict:
+    """
+    执行查询基本信息
+
+    Args:
+        entity_name: 实体名称
+        entity_type: 实体类型 (gp/lp/fund/company/preipo/list/neeq)
+
+    Returns:
+        查询结果字典
+    """
+    app_id, app_key = config_manager.get_credentials()
+    config = config_manager.load()
+
+    client = OpenAPIClient(base_url=config.base_url)
+    entity_service = EntityService()
+    executor = ToolExecutor(
+        entity_router=EntityRouter(entity_service),
+        type_router=TypeRouter(),
+        field_mapper=FieldMapper(),
+        openapi_client=client,
+    )
+
+    try:
+        result = await executor.execute_get_basic_info(
+            entity_name=entity_name,
+            entity_type=entity_type,
+            app_id=app_id,
+            app_key=app_key,
+        )
+        return result
+    finally:
+        await client.close()
+
+
+def get_basic_info(
+    entity_name: str = typer.Argument(..., help="实体名称，如'字节跳动'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t", help="实体类型：gp/lp/fund/company/preipo/list/neeq"
+    ),
+    output: str = typer.Option(
+        "table", "--output", "-o", help="输出格式：table/json/csv"
+    ),
+):
+    """
+    查询实体基本信息
+
+    示例：
+        zerone-cli get-basic-info "字节跳动"
+        zerone-cli get-basic-info "红杉资本" --type gp
+        zerone-cli get-basic-info "字节跳动" --output json
+    """
+    try:
+        result = asyncio.run(execute(entity_name, entity_type))
+        print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"[red]错误: {e}[/red]", err=True)
+        raise typer.Exit(1)

zerone_cli-0.1.3/cli/commands/config_cmd.py

@@ -0,0 +1,47 @@
+import typer
+
+from cli.config import config_manager
+
+app = typer.Typer(help="管理 CLI 配置")
+
+
+@app.command("show")
+def show_config():
+    """显示当前配置"""
+    config = config_manager.load()
+    typer.echo("=== 当前配置 ===")
+    typer.echo(f"api_key: {'*' * 10 if config.api_key else '(未设置)'}")
+    typer.echo(f"server_url: {config.server_url}")
+    typer.echo(f"output_format: {config.output_format}")
+
+
+@app.command("set-api-key")
+def set_api_key(api_key: str = typer.Argument(..., help="API Key（从平台获取）")):
+    """设置 API Key"""
+    config = config_manager.load()
+    config.api_key = api_key
+    config_manager.save(config)
+    typer.echo("✓ API Key 已设置")
+
+
+@app.command("set-server-url")
+def set_server_url(url: str = typer.Argument(..., help="MCP Server 地址，如 http://localhost:8080")):
+    """设置 MCP Server 地址"""
+    config = config_manager.load()
+    config.server_url = url
+    config_manager.save(config)
+    typer.echo(f"✓ Server URL 已设置: {url}")
+
+
+@app.command("set-output-format")
+def set_output_format(
+    fmt: str = typer.Argument(..., help="输出格式 (table/json/csv)")
+):
+    """设置默认输出格式"""
+    if fmt not in ("table", "json", "csv"):
+        typer.echo("✗ 无效的输出格式，请选择: table, json, csv")
+        raise typer.Exit(1)
+    config = config_manager.load()
+    config.output_format = fmt
+    config_manager.save(config)
+    typer.echo(f"✓ Output format 已设置: {fmt}")

zerone_cli-0.1.3/cli/commands/entity_exit_list.py

@@ -0,0 +1,47 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def entity_exit_list(
+    entity_name: str = typer.Argument(..., help="实体名称，如'高瓴资本'"),
+    entity_type: Optional[str] = typer.Option(
+        None, "--type", "-t",
+        help="实体类型：GP/LP/FUND"
+    ),
+    exit_type: Optional[str] = typer.Option(
+        None, "--exit-type", "-e",
+        help="退出类型：COMPANY/FUND"
+    ),
+    page: int = typer.Option(1, "--page", "-p", help="页码，从 1 开始"),
+    page_size: int = typer.Option(20, "--page-size", "-s", help="每页条数，默认 20"),
+    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
+):
+    """
+    查询退出列表（分页）
+
+    支持 GP/LP/FUND 类型。
+
+    示例：
+        zerone entity-exit-list "高瓴资本" --type GP
+        zerone entity-exit-list "高瓴资本" --type GP --exit-type COMPANY
+    """
+    try:
+        with MCPClient() as client:
+            result = client.entity_exit_list(
+                entity_name=entity_name,
+                entity_type=entity_type,
+                exit_type=exit_type,
+                page=page,
+                page_size=page_size,
+            )
+            print_result(result, output)
+    except ValueError as e:
+        typer.echo(f"错误: {e}", err=True)
+        raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"请求失败: {e}", err=True)
+        raise typer.Exit(1)