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

{zerone_cli-0.1.3 → zerone_cli-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zerone-cli
-Version: 0.1.3
+Version: 0.1.4
 Summary: Zerone 私募股权数据 CLI 工具
 Author-email: Zerone <tech@zerone.com.cn>
 License: MIT
@@ -67,10 +67,10 @@ 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-exit-list "深创投" --exit-type FUND --page 1 --page-size 20
 
 # 查询投资列表
-zerone entity-invest-list "深创投" --invest-type 直接投资 --page 1 --page-size 20
+zerone entity-invest-list "深创投" --invest-type COMPANY --page 1 --page-size 20
 
 # 查询融资概率预测
 zerone model-financing-probability "字节跳动" --type COMPANY

{zerone_cli-0.1.3 → zerone_cli-0.1.4}/README.md

@@ -40,10 +40,10 @@ 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-exit-list "深创投" --exit-type FUND --page 1 --page-size 20
 
 # 查询投资列表
-zerone entity-invest-list "深创投" --invest-type 直接投资 --page 1 --page-size 20
+zerone entity-invest-list "深创投" --invest-type COMPANY --page 1 --page-size 20
 
 # 查询融资概率预测
 zerone model-financing-probability "字节跳动" --type COMPANY

{zerone_cli-0.1.3 → zerone_cli-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "zerone-cli"
-version = "0.1.3"
+version = "0.1.4"
 description = "Zerone 私募股权数据 CLI 工具"
 readme = "README.md"
 license = { text = "MIT" }

{zerone_cli-0.1.3 → zerone_cli-0.1.4}/zerone_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zerone-cli
-Version: 0.1.3
+Version: 0.1.4
 Summary: Zerone 私募股权数据 CLI 工具
 Author-email: Zerone <tech@zerone.com.cn>
 License: MIT
@@ -67,10 +67,10 @@ 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-exit-list "深创投" --exit-type FUND --page 1 --page-size 20
 
 # 查询投资列表
-zerone entity-invest-list "深创投" --invest-type 直接投资 --page 1 --page-size 20
+zerone entity-invest-list "深创投" --invest-type COMPANY --page 1 --page-size 20
 
 # 查询融资概率预测
 zerone model-financing-probability "字节跳动" --type COMPANY

zerone_cli-0.1.4/zerone_cli.egg-info/SOURCES.txt

@@ -0,0 +1,8 @@
+README.md
+pyproject.toml
+zerone_cli.egg-info/PKG-INFO
+zerone_cli.egg-info/SOURCES.txt
+zerone_cli.egg-info/dependency_links.txt
+zerone_cli.egg-info/entry_points.txt
+zerone_cli.egg-info/requires.txt
+zerone_cli.egg-info/top_level.txt

zerone_cli-0.1.4/zerone_cli.egg-info/top_level.txt

@@ -0,0 +1 @@
+

zerone_cli-0.1.3/cli/__init__.py

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

zerone_cli-0.1.3/cli/client.py

@@ -1,176 +0,0 @@
-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

@@ -1,19 +0,0 @@
-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

@@ -1,77 +0,0 @@
-"""
-查询基本信息命令模块
-"""
-
-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

@@ -1,47 +0,0 @@
-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

@@ -1,47 +0,0 @@
-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)

zerone_cli-0.1.3/cli/commands/entity_financial_list.py

@@ -1,42 +0,0 @@
-from typing import Optional
-
-import typer
-
-from cli.client import MCPClient
-from cli.utils.output import print_result
-
-
-def entity_financial_list(
-    entity_name: str = typer.Argument(..., help="实体名称，如'美团'"),
-    entity_type: Optional[str] = typer.Option(
-        None, "--type", "-t",
-        help="实体类型：LISTED/PREIPO/NEEQ"
-    ),
-    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"),
-):
-    """
-    查询实体财务数据列表（分页）
-
-    仅支持 LISTED/PREIPO/NEEQ 类型。
-
-    示例：
-        zerone entity-financial-list "美团" --type LISTED
-        zerone entity-financial-list "美团" --type LISTED --page 2
-    """
-    try:
-        with MCPClient() as client:
-            result = client.entity_financial_list(
-                entity_name=entity_name,
-                entity_type=entity_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)

zerone_cli-0.1.3/cli/commands/entity_invest_list.py

@@ -1,47 +0,0 @@
-from typing import Optional
-
-import typer
-
-from cli.client import MCPClient
-from cli.utils.output import print_result
-
-
-def entity_invest_list(
-    entity_name: str = typer.Argument(..., help="实体名称，如'高瓴资本'"),
-    entity_type: Optional[str] = typer.Option(
-        None, "--type", "-t",
-        help="实体类型：GP/LP/FUND"
-    ),
-    invest_type: Optional[str] = typer.Option(
-        None, "--invest-type", "-i",
-        help="投资类型：COMPANY/FUND/OTHER"
-    ),
-    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-invest-list "高瓴资本" --type GP
-        zerone entity-invest-list "高瓴资本" --type GP --invest-type COMPANY
-    """
-    try:
-        with MCPClient() as client:
-            result = client.entity_invest_list(
-                entity_name=entity_name,
-                entity_type=entity_type,
-                invest_type=invest_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)

zerone_cli-0.1.3/cli/commands/entity_managed_funds_list.py

@@ -1,42 +0,0 @@
-from typing import Optional
-
-import typer
-
-from cli.client import MCPClient
-from cli.utils.output import print_result
-
-
-def entity_managed_funds_list(
-    entity_name: str = typer.Argument(..., help="实体名称，如'红杉资本'"),
-    entity_type: Optional[str] = typer.Option(
-        None, "--type", "-t",
-        help="实体类型：GP/FUND_MANAGER"
-    ),
-    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 和 FUND_MANAGER 类型。
-
-    示例：
-        zerone entity-managed-funds-list "红杉资本" --type GP
-        zerone entity-managed-funds-list "红杉资本" --type GP --page 2
-    """
-    try:
-        with MCPClient() as client:
-            result = client.entity_managed_funds_list(
-                entity_name=entity_name,
-                entity_type=entity_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)

zerone_cli-0.1.3/cli/commands/entity_profile.py

@@ -1,40 +0,0 @@
-from typing import Optional
-
-import typer
-
-from cli.client import MCPClient
-from cli.utils.output import print_result
-
-
-def entity_profile(
-    entity_name: str = typer.Argument(..., help="实体名称，如'字节跳动'"),
-    entity_type: Optional[str] = typer.Option(
-        None, "--type", "-t",
-        help="实体类型：GP/FUND/LP/COMPANY/PREIPO/LISTED/NEEQ/FUND_MANAGER"
-    ),
-    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
-):
-    """
-    查询实体综合画像
-
-    返回基本信息、标签画像、概览数据、股东列表、融资历史、关联GP列表等。
-    支持全部 8 种实体类型：GP/FUND/LP/COMPANY/FUND_MANAGER/PREIPO/LISTED/NEEQ。
-
-    示例：
-        zerone entity-profile "字节跳动"
-        zerone entity-profile "红杉资本" --type GP
-        zerone entity-profile "字节跳动" --output json
-    """
-    try:
-        with MCPClient() as client:
-            result = client.entity_profile(
-                entity_name=entity_name,
-                entity_type=entity_type,
-            )
-            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)

zerone_cli-0.1.3/cli/commands/entity_search.py

@@ -1,31 +0,0 @@
-from typing import Optional
-
-import typer
-
-from cli.client import MCPClient
-from cli.utils.output import print_result
-
-
-def entity_search(
-    keyword: str = typer.Argument(..., help="搜索关键词"),
-    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
-):
-    """
-    实体模糊搜索
-
-    根据关键词搜索匹配的实体，返回候选实体列表。
-
-    示例：
-        zerone entity-search "字节跳动"
-        zerone entity-search "红杉" --output json
-    """
-    try:
-        with MCPClient() as client:
-            result = client.entity_search(keyword=keyword)
-            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)

zerone_cli-0.1.3/cli/commands/financing.py

@@ -1,84 +0,0 @@
-"""
-查询融资概率命令模块
-"""
-
-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:
-    """
-    执行查询融资概率
-
-    注意：此命令仅支持项目公司类型（company/preipo/list/neeq）
-
-    Args:
-        entity_name: 实体名称
-        entity_type: 实体类型
-
-    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_financing_probability(
-            entity_name=entity_name,
-            entity_type=entity_type,
-            app_id=app_id,
-            app_key=app_key,
-        )
-        return result
-    finally:
-        await client.close()
-
-
-def get_financing_probability(
-    entity_name: str = typer.Argument(..., help="实体名称，如'字节跳动'"),
-    entity_type: Optional[str] = typer.Option(
-        None, "--type", "-t", help="实体类型：company/preipo/list/neeq"
-    ),
-    output: str = typer.Option(
-        "table", "--output", "-o", help="输出格式：table/json/csv"
-    ),
-):
-    """
-    查询公司融资概率（2年）
-
-    注意：此命令仅支持项目公司类型（company/preipo/list/neeq）
-
-    示例：
-        zerone-cli get-financing-probability "蔚来汽车"
-        zerone-cli get-financing-probability "字节跳动" --type company
-        zerone-cli get-financing-probability "蔚来汽车" --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/model_financing_probability.py

@@ -1,39 +0,0 @@
-from typing import Optional
-
-import typer
-
-from cli.client import MCPClient
-from cli.utils.output import print_result
-
-
-def model_financing_probability(
-    entity_name: str = typer.Argument(..., help="实体名称，如'蔚来汽车'"),
-    entity_type: Optional[str] = typer.Option(
-        None, "--type", "-t",
-        help="实体类型：COMPANY/PREIPO/LISTED/NEEQ"
-    ),
-    output: str = typer.Option("table", "--output", "-o", help="输出格式：table/json/csv"),
-):
-    """
-    查询项目公司融资概率预测模型
-
-    仅支持 COMPANY/PREIPO/LISTED/NEEQ 类型。
-    返回融资概率（2年）和模型指标数据。
-
-    示例：
-        zerone model-financing-probability "蔚来汽车"
-        zerone model-financing-probability "蔚来汽车" --type COMPANY
-    """
-    try:
-        with MCPClient() as client:
-            result = client.model_financing_probability(
-                entity_name=entity_name,
-                entity_type=entity_type,
-            )
-            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)

zerone_cli-0.1.3/cli/commands/shareholders.py

@@ -1,88 +0,0 @@
-"""
-查询股东信息命令模块
-"""
-
-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,
-    page: int = 1,
-    page_size: int = 20,
-) -> dict:
-    """
-    执行查询股东信息
-
-    Args:
-        entity_name: 实体名称
-        entity_type: 实体类型
-        page: 页码，从 1 开始
-        page_size: 每页数量
-
-    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_shareholders(
-            entity_name=entity_name,
-            entity_type=entity_type,
-            page=page,
-            page_size=page_size,
-            app_id=app_id,
-            app_key=app_key,
-        )
-        return result
-    finally:
-        await client.close()
-
-
-def get_shareholders(
-    entity_name: str = typer.Argument(..., help="实体名称，如'字节跳动'"),
-    entity_type: Optional[str] = typer.Option(
-        None, "--type", "-t", help="实体类型：gp/lp/fund/company/preipo/list/neeq"
-    ),
-    page: int = typer.Option(1, "--page", "-p", help="页码"),
-    page_size: int = typer.Option(20, "--page-size", "-s", help="每页数量"),
-    output: str = typer.Option(
-        "table", "--output", "-o", help="输出格式：table/json/csv"
-    ),
-):
-    """
-    查询实体股东信息（支持分页）
-
-    示例：
-        zerone-cli get-shareholders "美团"
-        zerone-cli get-shareholders "高瓴资本" --type gp --page 1 --page-size 10
-        zerone-cli get-shareholders "美团" --output csv
-    """
-    try:
-        result = asyncio.run(execute(entity_name, entity_type, page, page_size))
-        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/config.py

@@ -1,69 +0,0 @@
-import os
-import yaml
-from pathlib import Path
-from typing import Optional
-
-from pydantic import BaseModel
-
-SERVER_URL_PROD = "https://ai.zerone.com.cn/mcp"
-SERVER_URL_DEV = "http://ai-dev.zerone.com.cn/mcp"
-
-_env = os.getenv("ZERONE_ENV", "prod").lower()
-DEFAULT_SERVER_URL = SERVER_URL_DEV if _env == "dev" else SERVER_URL_PROD
-
-
-class CLIConfig(BaseModel):
-    api_key: Optional[str] = None
-    server_url: str = DEFAULT_SERVER_URL
-    output_format: str = "table"
-
-
-class ConfigManager:
-
-    CONFIG_DIR = Path.home() / ".zerone"
-    CONFIG_FILE = CONFIG_DIR / "config.yaml"
-
-    def __init__(self):
-        self._config: Optional[CLIConfig] = None
-
-    def _ensure_config_dir(self):
-        self.CONFIG_DIR.mkdir(parents=True, exist_ok=True)
-
-    def load(self) -> CLIConfig:
-        if self._config:
-            return self._config
-
-        if self.CONFIG_FILE.exists():
-            with open(self.CONFIG_FILE, "r", encoding="utf-8") as f:
-                data = yaml.safe_load(f) or {}
-        else:
-            data = {}
-
-        data["api_key"] = os.getenv("ZERONE_API_KEY", data.get("api_key"))
-        data["server_url"] = os.getenv("ZERONE_SERVER_URL", data.get("server_url", DEFAULT_SERVER_URL))
-
-        self._config = CLIConfig(**data)
-        return self._config
-
-    def save(self, config: CLIConfig):
-        self._ensure_config_dir()
-        self._config = config
-        with open(self.CONFIG_FILE, "w", encoding="utf-8") as f:
-            yaml.dump(config.model_dump(), f, allow_unicode=True, default_flow_style=False)
-
-    def get_api_key(self) -> str:
-        config = self.load()
-        if not config.api_key:
-            raise ValueError(
-                "未配置 API Key。\n"
-                "请运行：zerone config set-api-key <your-api-key>\n"
-                "或设置环境变量：ZERONE_API_KEY"
-            )
-        return config.api_key
-
-    def get_server_url(self) -> str:
-        config = self.load()
-        return config.server_url
-
-
-config_manager = ConfigManager()

zerone_cli-0.1.3/cli/main.py

@@ -1,29 +0,0 @@
-import typer
-
-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
-
-app = typer.Typer(
-    name="zerone",
-    help="Zerone 私募股权数据 CLI 工具 - MCP HTTP 客户端",
-    no_args_is_help=True,
-)
-
-app.add_typer(config_cmd.app, name="config")
-
-app.command("entity-search")(entity_search)
-app.command("entity-profile")(entity_profile)
-app.command("entity-financial-list")(entity_financial_list)
-app.command("entity-managed-funds-list")(entity_managed_funds_list)
-app.command("entity-exit-list")(entity_exit_list)
-app.command("entity-invest-list")(entity_invest_list)
-app.command("model-financing-probability")(model_financing_probability)
-
-if __name__ == "__main__":
-    app()

zerone_cli-0.1.3/cli/utils/__init__.py

@@ -1,3 +0,0 @@
-from .output import print_result, print_error
-
-__all__ = ["print_result", "print_error"]

zerone_cli-0.1.3/cli/utils/output.py

@@ -1,97 +0,0 @@
-"""
-CLI 输出格式化模块
-
-支持 table、json、csv 三种输出格式。
-"""
-
-import csv
-import io
-import json
-from typing import Any
-
-from rich.console import Console
-from rich.table import Table
-
-console = Console()
-
-
-def print_result(result: dict, output_format: str = "table"):
-    """
-    主入口函数，根据格式输出结果
-
-    Args:
-        result: 包含 data、metadata 的结果字典
-        output_format: 输出格式 (table/json/csv)
-    """
-    if "error" in result:
-        print_error(result)
-        return
-
-    data = result.get("data", {})
-    metadata = result.get("metadata", {})
-
-    if output_format == "json":
-        print_json(data)
-    elif output_format == "csv":
-        print_csv(data)
-    else:
-        print_table(data, metadata)
-
-
-def print_table(data: Any, metadata: dict):
-    """表格格式输出"""
-    if isinstance(data, list):
-        if not data:
-            console.print("[yellow]没有查询到数据[/yellow]")
-            return
-        keys = list(data[0].keys()) if data else []
-        table = Table(show_header=True, header_style="bold cyan", box=None)
-        for key in keys:
-            table.add_column(key, style="white")
-        for row in data:
-            table.add_row(*[str(row.get(k, "")) for k in keys])
-        console.print(table)
-    elif isinstance(data, dict):
-        table = Table(show_header=True, header_style="bold cyan", box=None)
-        table.add_column("字段", style="cyan")
-        table.add_column("值", style="white")
-        for key, value in data.items():
-            table.add_row(key, str(value) if value is not None else "")
-        console.print(table)
-    else:
-        console.print(str(data))
-
-    entity_name = metadata.get("entityName", "")
-    entity_type = metadata.get("entityType", "")
-    if entity_name:
-        console.print(f"\n[dim]实体名称: {entity_name} | 类型: {entity_type}[/dim]")
-
-
-def print_json(data: Any):
-    """JSON 格式输出"""
-    console.print(json.dumps(data, ensure_ascii=False, indent=2))
-
-
-def print_csv(data: Any):
-    """CSV 格式输出"""
-    output = io.StringIO()
-    if isinstance(data, list):
-        if not data:
-            return
-        writer = csv.DictWriter(output, fieldnames=data[0].keys())
-        writer.writeheader()
-        writer.writerows(data)
-    elif isinstance(data, dict):
-        writer = csv.DictWriter(output, fieldnames=data.keys())
-        writer.writeheader()
-        writer.writerow(data)
-    else:
-        output.write(str(data))
-    console.print(output.getvalue())
-
-
-def print_error(error: dict):
-    """错误信息输出"""
-    code = error.get("code", error.get("error_code", "UNKNOWN"))
-    message = error.get("message", error.get("msg", "未知错误"))
-    console.print(f"[bold red]错误 {code}:[/bold red] {message}")

zerone_cli-0.1.3/zerone_cli.egg-info/SOURCES.txt

@@ -1,27 +0,0 @@
-README.md
-pyproject.toml
-cli/__init__.py
-cli/client.py
-cli/config.py
-cli/main.py
-cli/commands/__init__.py
-cli/commands/basic_info.py
-cli/commands/config_cmd.py
-cli/commands/entity_exit_list.py
-cli/commands/entity_financial_list.py
-cli/commands/entity_invest_list.py
-cli/commands/entity_managed_funds_list.py
-cli/commands/entity_profile.py
-cli/commands/entity_search.py
-cli/commands/financing.py
-cli/commands/model_financing_probability.py
-cli/commands/shareholders.py
-cli/utils/__init__.py
-cli/utils/errors.py
-cli/utils/output.py
-zerone_cli.egg-info/PKG-INFO
-zerone_cli.egg-info/SOURCES.txt
-zerone_cli.egg-info/dependency_links.txt
-zerone_cli.egg-info/entry_points.txt
-zerone_cli.egg-info/requires.txt
-zerone_cli.egg-info/top_level.txt

zerone_cli-0.1.3/zerone_cli.egg-info/top_level.txt

@@ -1 +0,0 @@
-cli