mcp-query-table 0.3.9__tar.gz → 0.3.10__tar.gz

{mcp_query_table-0.3.9 → mcp_query_table-0.3.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp_query_table
-Version: 0.3.9
+Version: 0.3.10
 Summary: query table from website, support MCP
 Author-email: wukan <wu-kan@163.com>
 License: MIT License

mcp_query_table-0.3.10/mcp_query_table/_version.py

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

{mcp_query_table-0.3.9 → mcp_query_table-0.3.10}/mcp_query_table/server.py

@@ -53,9 +53,10 @@ async def query(
         query_type: Annotated[QueryType, Field(default=QueryType.CNStock,
                                                description="查询类型。支持`A股`、`指数`、`基金`、`港股`、`美股`等")],
         max_page: Annotated[int, Field(default=1, ge=1, le=10, description="最大页数。只查第一页即可")],
+        rename: Annotated[bool, Field(default=False, description="是否重命名列名")],
         site: Annotated[Site, Field(default=Site.THS, description="站点。支持`东方财富`、`通达信`、`同花顺`")]
 ) -> str:
-    return await qsv.query(query_input, query_type, max_page, site)
+    return await qsv.query(query_input, query_type, max_page, rename, site)
 
 
 # chat功能不通过mcp暴露，因为在Cline等客户端中本就有LLM功能，反而导致返回的数据没有正确提交

{mcp_query_table-0.3.9 → mcp_query_table-0.3.10}/mcp_query_table/sites/eastmoney.py

@@ -81,7 +81,7 @@ class Pagination:
             datas.extend(v)
         return datas
 
-    def get_dataframe(self):
+    def get_dataframe(self, rename: bool):
         columns = {x['key']: x['title'] for x in self.columns}
         dtypes = {x['key']: convert_type(x['dataType']) for x in self.columns}
 
@@ -98,7 +98,10 @@ class Pagination:
             except ValueError:
                 logger.info("转换失败 {}:{}", k, v)
 
-        return df.rename(columns=columns)
+        if rename:
+            return df.rename(columns=columns)
+        else:
+            return df
 
 
 P = Pagination()
@@ -121,7 +124,8 @@ async def on_response(response):
 async def query(page: Page,
                 q: str = "收盘价>100元",
                 type_: QueryType = 'stock',
-                max_page: int = 5) -> pd.DataFrame:
+                max_page: int = 5,
+                rename: bool = True) -> pd.DataFrame:
     type = _type_.get(type_, None)
     assert type is not None, f"不支持的类型:{type_}"
 
@@ -141,4 +145,4 @@ async def query(page: Page,
             await page.get_by_role("button", name="下一页").click()
         await on_response(await response_info.value)
 
-    return P.get_dataframe()
+    return P.get_dataframe(rename)

{mcp_query_table-0.3.9 → mcp_query_table-0.3.10}/mcp_query_table/sites/iwencai.py

@@ -80,7 +80,7 @@ class Pagination:
             datas.extend(v)
         return datas
 
-    def get_dataframe(self):
+    def get_dataframe(self, rename: bool):
         columns = {x['key']: x['index_name'] for x in self.columns}
         dtypes = {x['key']: convert_type(x['type']) for x in self.columns}
 
@@ -94,7 +94,10 @@ class Pagination:
             except ValueError:
                 logger.info("转换失败 {}:{}", k, v)
 
-        return df.rename(columns=columns)
+        if rename:
+            return df.rename(columns=columns)
+        else:
+            return df
 
 
 P = Pagination()
@@ -148,7 +151,8 @@ async def on_response(response):
 async def query(page: Page,
                 w: str = "收盘价>1000元",
                 type_: QueryType = 'stock',
-                max_page: int = 5) -> pd.DataFrame:
+                max_page: int = 5,
+                rename: bool = False) -> pd.DataFrame:
     querytype = _querytype_.get(type_, None)
     assert querytype is not None, f"不支持的类型:{type_}"
 
@@ -168,4 +172,4 @@ async def query(page: Page,
             await page.get_by_text("下页").click()
         await on_response(await response_info.value)
 
-    return P.get_dataframe()
+    return P.get_dataframe(rename)

{mcp_query_table-0.3.9 → mcp_query_table-0.3.10}/mcp_query_table/sites/tdx.py

@@ -78,7 +78,7 @@ class Pagination:
             datas.extend(v)
         return datas
 
-    def get_dataframe(self):
+    def get_dataframe(self, rename: bool):
         dtypes = [convert_type(x) for x in self.dtypes]
         df = pd.DataFrame(self.get_list(), columns=self.columns)
         for i, v in enumerate(dtypes):
@@ -128,7 +128,8 @@ async def on_response2(response):
 async def query(page: Page,
                 message: str = "收盘价>100元",
                 type_: QueryType = 'AG',
-                max_page: int = 5) -> pd.DataFrame:
+                max_page: int = 5,
+                rename: bool = False) -> pd.DataFrame:
     queryType = _queryType_.get(type_, None)
     assert queryType is not None, f"不支持的类型:{type_}"
 
@@ -147,4 +148,4 @@ async def query(page: Page,
             await page.get_by_role("button", name="下一页").click()
         await on_response1(await response_info.value)
 
-    return P.get_dataframe()
+    return P.get_dataframe(rename)

{mcp_query_table-0.3.9 → mcp_query_table-0.3.10}/mcp_query_table/tool.py

@@ -249,7 +249,9 @@ async def query(
         query_input: str = "收盘价>100元",
         query_type: QueryType = QueryType.CNStock,
         max_page: int = 5,
-        site: Site = Site.THS) -> pd.DataFrame:
+        rename: bool = False,
+        site: Site = Site.THS,
+        ) -> pd.DataFrame:
     """查询表格
 
     Parameters
@@ -262,6 +264,8 @@ async def query(
         查询类型, by default QueryType.astock
     max_page : int, optional
         最大页数, by default 5
+    rename: bool
+        是否重命名列名, by default False
     site : Site, optional
         站点, by default Site.iwencai
 
@@ -275,13 +279,13 @@ async def query(
 
     if site == Site.EastMoney:
         from mcp_query_table.sites.eastmoney import query
-        return await query(page, query_input, query_type, max_page)
+        return await query(page, query_input, query_type, max_page, rename)
     if site == Site.THS:
         from mcp_query_table.sites.iwencai import query
-        return await query(page, query_input, query_type, max_page)
+        return await query(page, query_input, query_type, max_page, rename)
     if site == Site.TDX:
         from mcp_query_table.sites.tdx import query
-        return await query(page, query_input, query_type, max_page)
+        return await query(page, query_input, query_type, max_page, rename)
 
     raise ValueError(f"未支持的站点:{site}")
 

mcp_query_table-0.3.9/mcp_query_table/_version.py

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