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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zerone-cli
-Version: 0.1.4
+Version: 0.1.7
 Summary: Zerone 私募股权数据 CLI 工具
 Author-email: Zerone <tech@zerone.com.cn>
 License: MIT
@@ -79,7 +79,7 @@ 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
+zerone evt-company-financing --start-date 2024-01-01 --end-date 2024-12-31 --province 北京  --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
@@ -94,10 +94,10 @@ zerone evt-company-repurchase --start-date 2024-01-01 --end-date 2024-12-31 --pr
 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-company-ipo --start-date 2024-01-01 --end-date 2024-12-31 --city 合肥 --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-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

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

@@ -52,7 +52,7 @@ 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
+zerone evt-company-financing --start-date 2024-01-01 --end-date 2024-12-31 --province 北京  --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
@@ -67,10 +67,10 @@ zerone evt-company-repurchase --start-date 2024-01-01 --end-date 2024-12-31 --pr
 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-company-ipo --start-date 2024-01-01 --end-date 2024-12-31 --city 合肥 --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-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

zerone_cli-0.1.7/cli/__init__.py

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

zerone_cli-0.1.7/cli/client.py

@@ -0,0 +1,422 @@
+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 evt_company_financing(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        company_tag: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_company_financing",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "company_tag": company_tag,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def evt_lp_investment(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_lp_investment",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def evt_company_secondary(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        company_tag: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_company_secondary",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "company_tag": company_tag,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def evt_company_repurchase(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        company_tag: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_company_repurchase",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "company_tag": company_tag,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def evt_ma_company_deal(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        target_company_tag: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_ma_company_deal",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "target_company_tag": target_company_tag,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def evt_company_ipo(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        company_tag: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_company_ipo",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "company_tag": company_tag,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def evt_listing_company_unlock(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        company_tag: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_listing_company_unlock",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "company_tag": company_tag,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def evt_listing_company_reduction(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        company_tag: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_listing_company_reduction",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "company_tag": company_tag,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def evt_s_fund_deal(
+        self,
+        start_date: str = "",
+        end_date: str = "",
+        reg_province: str = "",
+        reg_city: str = "",
+        reg_county: str = "",
+        page: int = 1,
+        page_size: int = 20,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/evt_s_fund_deal",
+            json_data={
+                "start_date": start_date,
+                "end_date": end_date,
+                "reg_province": reg_province,
+                "reg_city": reg_city,
+                "reg_county": reg_county,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    def model_company_match_gp(
+        self,
+        company_name: str,
+        page: int = 1,
+        page_size: int = 5,
+    ) -> dict:
+        return self._request(
+            "POST",
+            "/rest-api/tool/model_company_match_gp",
+            json_data={
+                "company_name": company_name,
+                "page": page,
+                "page_size": page_size,
+            },
+        )
+
+    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.7/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.7/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.7/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.7/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)