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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zerone-cli
-Version: 0.1.3
+Version: 0.1.6
 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.6}/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.6/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.6/cli/commands/evt_company_financing.py

@@ -0,0 +1,48 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def evt_company_financing(
+    start_date: Optional[str] = typer.Option(None, "--start-date", help="融资日期-开始，YYYY-MM-DD"),
+    end_date: Optional[str] = typer.Option(None, "--end-date", help="融资日期-截止，YYYY-MM-DD"),
+    reg_province: Optional[str] = typer.Option(None, "--province", help="公司注册省级行政区划"),
+    reg_city: Optional[str] = typer.Option(None, "--city", help="公司注册地级行政区划"),
+    reg_county: Optional[str] = typer.Option(None, "--county", help="公司注册县级行政区划"),
+    company_tag: Optional[str] = typer.Option(None, "--tag", help="项目公司标签（经营业务类）"),
+    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"),
+):
+    """
+    查询市场投融资事件列表
+
+    支持按时间范围、公司注册地区域（省/市/县）、项目公司标签筛选，
+    返回符合条件的融资事件及投资方信息。
+
+    示例：
+        zerone evt-company-financing --start-date 2024-01-01 --end-date 2024-12-31
+        zerone evt-company-financing --province 北京 --tag 人工智能
+    """
+    try:
+        with MCPClient() as client:
+            result = client.evt_company_financing(
+                start_date=start_date or "",
+                end_date=end_date or "",
+                reg_province=reg_province or "",
+                reg_city=reg_city or "",
+                reg_county=reg_county or "",
+                company_tag=company_tag or "",
+                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.6/cli/commands/evt_company_ipo.py

@@ -0,0 +1,48 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def evt_company_ipo(
+    start_date: Optional[str] = typer.Option(None, "--start-date", help="IPO日期-开始，YYYY-MM-DD"),
+    end_date: Optional[str] = typer.Option(None, "--end-date", help="IPO日期-截止，YYYY-MM-DD"),
+    reg_province: Optional[str] = typer.Option(None, "--province", help="公司注册省级行政区划"),
+    reg_city: Optional[str] = typer.Option(None, "--city", help="公司注册地级行政区划"),
+    reg_county: Optional[str] = typer.Option(None, "--county", help="公司注册县级行政区划"),
+    company_tag: Optional[str] = typer.Option(None, "--tag", help="项目公司标签（经营业务类）"),
+    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"),
+):
+    """
+    查询IPO事件列表
+
+    支持按时间范围、公司注册地区域（省/市/县）、项目公司标签筛选，
+    返回符合条件的IPO事件。
+
+    示例：
+        zerone evt-company-ipo --start-date 2024-01-01 --end-date 2024-12-31
+        zerone evt-company-ipo --province 北京 --tag 半导体
+    """
+    try:
+        with MCPClient() as client:
+            result = client.evt_company_ipo(
+                start_date=start_date or "",
+                end_date=end_date or "",
+                reg_province=reg_province or "",
+                reg_city=reg_city or "",
+                reg_county=reg_county or "",
+                company_tag=company_tag or "",
+                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.6/cli/commands/evt_company_repurchase.py

@@ -0,0 +1,48 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def evt_company_repurchase(
+    start_date: Optional[str] = typer.Option(None, "--start-date", help="回购日期-开始，YYYY-MM-DD"),
+    end_date: Optional[str] = typer.Option(None, "--end-date", help="回购日期-截止，YYYY-MM-DD"),
+    reg_province: Optional[str] = typer.Option(None, "--province", help="公司注册省级行政区划"),
+    reg_city: Optional[str] = typer.Option(None, "--city", help="公司注册地级行政区划"),
+    reg_county: Optional[str] = typer.Option(None, "--county", help="公司注册县级行政区划"),
+    company_tag: Optional[str] = typer.Option(None, "--tag", help="项目公司标签（经营业务类）"),
+    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"),
+):
+    """
+    查询回购事件列表
+
+    支持按时间范围、公司注册地区域（省/市/县）、项目公司标签筛选，
+    返回符合条件的回购事件。
+
+    示例：
+        zerone evt-company-repurchase --start-date 2024-01-01 --end-date 2024-12-31
+        zerone evt-company-repurchase --province 江苏
+    """
+    try:
+        with MCPClient() as client:
+            result = client.evt_company_repurchase(
+                start_date=start_date or "",
+                end_date=end_date or "",
+                reg_province=reg_province or "",
+                reg_city=reg_city or "",
+                reg_county=reg_county or "",
+                company_tag=company_tag or "",
+                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.6/cli/commands/evt_company_secondary.py

@@ -0,0 +1,48 @@
+from typing import Optional
+
+import typer
+
+from cli.client import MCPClient
+from cli.utils.output import print_result
+
+
+def evt_company_secondary(
+    start_date: Optional[str] = typer.Option(None, "--start-date", help="转让日期-开始，YYYY-MM-DD"),
+    end_date: Optional[str] = typer.Option(None, "--end-date", help="转让日期-截止，YYYY-MM-DD"),
+    reg_province: Optional[str] = typer.Option(None, "--province", help="公司注册省级行政区划"),
+    reg_city: Optional[str] = typer.Option(None, "--city", help="公司注册地级行政区划"),
+    reg_county: Optional[str] = typer.Option(None, "--county", help="公司注册县级行政区划"),
+    company_tag: Optional[str] = typer.Option(None, "--tag", help="项目公司标签（经营业务类）"),
+    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"),
+):
+    """
+    查询老股转让事件列表
+
+    支持按时间范围、公司注册地区域（省/市/县）、项目公司标签筛选，
+    返回符合条件的老股转让事件。
+
+    示例：
+        zerone evt-company-secondary --start-date 2024-01-01 --end-date 2024-12-31
+        zerone evt-company-secondary --province 上海
+    """
+    try:
+        with MCPClient() as client:
+            result = client.evt_company_secondary(
+                start_date=start_date or "",
+                end_date=end_date or "",
+                reg_province=reg_province or "",
+                reg_city=reg_city or "",
+                reg_county=reg_county or "",
+                company_tag=company_tag or "",
+                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)