mcp-query-table 0.3.8__py3-none-any.whl → 0.3.9__py3-none-any.whl

mcp_query_table/__main__.py

@@ -17,7 +17,7 @@ def main():
     parser.add_argument("--executable_path", type=str, help="浏览器路径",
                         nargs="?", default=r'C:\Program Files\Google\Chrome\Application\chrome.exe')
     parser.add_argument("--user_data_dir", type=str, help="浏览器用户数据目录",
-                        nargs="?", default=rf'C:\Users\{getpass.getuser()}\AppData\Local\Google\Chrome\User Data')
+                        nargs="?", default=rf'C:\Users\{getpass.getuser()}\AppData\Local\Google\Chrome\User Data\Default')
     parser.add_argument("--transport", type=str, help="传输类型",
                         default='stdio', choices=['stdio', 'sse'])
     parser.add_argument("--host", type=str, help="MCP服务端绑定地址",

mcp_query_table/_version.py

@@ -1 +1 @@
-__version__ = "0.3.8"
+__version__ = "0.3.9"

mcp_query_table/sites/iwencai.py

@@ -10,10 +10,8 @@ import re
 import pandas as pd
 from loguru import logger
 from playwright.async_api import Page
-from playwright_stealth import stealth_async
 
 from mcp_query_table.enums import QueryType
-from mcp_query_table.utils import FixedConfig
 
 # 初次查询页面
 _PAGE1_ = 'https://www.iwencai.com/customized/chart/get-robot-data'
@@ -154,8 +152,6 @@ async def query(page: Page,
     querytype = _querytype_.get(type_, None)
     assert querytype is not None, f"不支持的类型:{type_}"
 
-    await stealth_async(page, FixedConfig())
-
     await page.route(re.compile(r'.*\.(?:jpg|jpeg|png|gif|webp)(?:$|\?)'), lambda route: route.abort())
 
     P.reset()

mcp_query_table/tool.py

@@ -4,11 +4,12 @@ import sys
 import time
 from pathlib import Path
 from typing import Optional
-from urllib.parse import urlparse
+from urllib.parse import urlparse, quote
 
 import pandas as pd
 from loguru import logger
 from playwright.async_api import async_playwright, Playwright, Page
+from playwright_stealth import Stealth
 
 from mcp_query_table.enums import QueryType, Site, Provider
 
@@ -204,6 +205,8 @@ class BrowserManager:
             self.context = await self.browser.new_context()
         else:
             self.context = self.browser.contexts[0]
+        # 爱问财，无头模式，需要使用 stealth 插件
+        await Stealth().apply_stealth_async(self.context)
 
         # 复用打开的page
         for page in self.context.pages:
@@ -268,6 +271,7 @@ async def query(
         查询结果
 
     """
+    query_input = quote(query_input.strip(), safe='')
 
     if site == Site.EastMoney:
         from mcp_query_table.sites.eastmoney import query

mcp_query_table/utils.py

@@ -1,10 +1,6 @@
-import random
-import string
 from pathlib import Path
 from typing import List, Tuple
 
-from playwright_stealth import StealthConfig
-
 
 def is_image(path: str) -> bool:
     """判断是否是图片文件"""
@@ -36,16 +32,3 @@ class GlobalVars:
 
     def get_text(self):
         return self.text
-
-
-# https://github.com/AtuboDad/playwright_stealth/issues/31#issuecomment-2342541305
-class FixedConfig(StealthConfig):
-
-    @property
-    def enabled_scripts(self):
-        key = "".join(random.choices(string.ascii_letters, k=10))
-        for script in super().enabled_scripts:
-            if "const opts" in script:
-                yield script.replace("const opts", f"window.{key}")
-                continue
-            yield script.replace("opts", f"window.{key}")

{mcp_query_table-0.3.8.dist-info → mcp_query_table-0.3.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mcp_query_table
-Version: 0.3.8
+Version: 0.3.9
 Summary: query table from website, support MCP
 Author-email: wukan <wu-kan@163.com>
 License: MIT License
@@ -33,8 +33,7 @@ Requires-Dist: loguru
 Requires-Dist: mcp
 Requires-Dist: pandas
 Requires-Dist: playwright
-Requires-Dist: playwright-stealth
-Requires-Dist: setuptools
+Requires-Dist: playwright-stealth>=2.0.0
 Requires-Dist: tabulate
 Description-Content-Type: text/markdown
 

{mcp_query_table-0.3.8.dist-info → mcp_query_table-0.3.9.dist-info}/RECORD

@@ -1,19 +1,19 @@
 mcp_query_table/__init__.py,sha256=K-0DU2hpeRvM9ZAuky0aWZtJLuOgKg7ZRd-pL9noc0o,330
-mcp_query_table/__main__.py,sha256=oePptyDeLtOHcR0XZxx-O12hO6LSe6cplb4gaJBG4rI,1453
-mcp_query_table/_version.py,sha256=7dTW0A5-FkrEuNOotvR8oW59M2lvIwYouVqfJzvXpKk,22
+mcp_query_table/__main__.py,sha256=K0r7y8PD9y9pFg3bAQxNo21KlQpGz1jqGg_zptPfenM,1461
+mcp_query_table/_version.py,sha256=xmkmdvq15kb61xdtCoa1YARnvHBnUgI-0GWIJYvHNeA,22
 mcp_query_table/enums.py,sha256=7bu0m0zJBIfiS-eHGURw1ZHWNXgsq6gH1SztUhCgF-Y,678
 mcp_query_table/server.py,sha256=D2-7ZmutijphasbLMosg9P5EOhJTB4RvC9-zmvCvc5k,3749
-mcp_query_table/tool.py,sha256=PpsDFwaKQ9NsfqSMCvWm1p0MSyxT1Uwpve198TOP8FQ,11157
-mcp_query_table/utils.py,sha256=VjKYLRPEa-W3qAOaNSJa9GnezrAAYykn4XY-R4P5NJg,1264
+mcp_query_table/tool.py,sha256=dW6Hcv7aB-hBotd7OBzCWQrpRck1h5812y6sRCUENVA,11380
+mcp_query_table/utils.py,sha256=MUKcklPF9TkABhM8wN0-kW0iy9AlmjL6oycZyxB_Qk8,722
 mcp_query_table/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_query_table/providers/baidu.py,sha256=S75D2zbpqG1r4Rxz7pJf5u2ZHNLO8nqV-LPEjlACtHg,3390
 mcp_query_table/providers/n.py,sha256=SLalpwHSdkXNoMCLQEx1TEFlo50dS7I9JNli5jz8w6k,3202
 mcp_query_table/providers/yuanbao.py,sha256=1wRMy7Z2JraM3MrgLDSdyg-EqX-D26ysx7CW1GFtVho,3292
 mcp_query_table/sites/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mcp_query_table/sites/eastmoney.py,sha256=LImjpYVuM5YnXwnNzB2hkKfHofocZZScetGqMOCHZpk,4477
-mcp_query_table/sites/iwencai.py,sha256=43sBrVCXgiIybv25lEwTS6dlR2jXXceSBVGCOb03woE,5194
+mcp_query_table/sites/iwencai.py,sha256=FlIQOAN6wPz0B8w5DFX-EHekPO5HPIYdLMhiX1dYV7s,5057
 mcp_query_table/sites/tdx.py,sha256=P-GNFUsS5_INy3sicaZbUBdudgQBZuhu_QyVvyw4yDg,4126
-mcp_query_table-0.3.8.dist-info/METADATA,sha256=4nnj_OrLYG0SMPlvSn9TpNI5IOx1DNsTfsWt22sVdqs,9372
-mcp_query_table-0.3.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-mcp_query_table-0.3.8.dist-info/licenses/LICENSE,sha256=rbvv_CTd7biGwT21tvhgQ2zkbPFXOoON7WFQWEdElBA,1063
-mcp_query_table-0.3.8.dist-info/RECORD,,
+mcp_query_table-0.3.9.dist-info/METADATA,sha256=w6Kk_cokNDdnTgr7APF6IhlSVOHJTrDAO0uGdhTaNvA,9353
+mcp_query_table-0.3.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_query_table-0.3.9.dist-info/licenses/LICENSE,sha256=rbvv_CTd7biGwT21tvhgQ2zkbPFXOoON7WFQWEdElBA,1063
+mcp_query_table-0.3.9.dist-info/RECORD,,