Jarvis-Brain 0.1.5.5__tar.gz → 0.1.5.6__tar.gz

{jarvis_brain-0.1.5.5 → jarvis_brain-0.1.5.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Jarvis_Brain
-Version: 0.1.5.5
+Version: 0.1.5.6
 Summary: Jarvis brain mcp
 Requires-Python: >=3.10
 Requires-Dist: beautifulsoup4

{jarvis_brain-0.1.5.5 → jarvis_brain-0.1.5.6}/README.md

@@ -4,8 +4,8 @@
 
 一个基于 FastMCP 和 DrissionPage 的浏览器自动化 MCP 服务器
 
-[![Python Version](https://img.shields.io/badge/python-3.10~3.12-blue.svg)](https://www.python.org/downloads/)
-[![Version](https://img.shields.io/badge/version-0.1.4.5-green.svg)](https://github.com/yourusername/jarvis-mcp)
+[![Python Version](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+[![Version](https://img.shields.io/badge/version-0.1.5.5-green.svg)](https://github.com/yourusername/jarvis-mcp)
 
 ## 📖 简介
 

{jarvis_brain-0.1.5.5 → jarvis_brain-0.1.5.6}/mcp_tools/dp_tools.py

@@ -2,7 +2,6 @@ import json
 import os
 from typing import Any
 
-from DrissionPage._elements.none_element import NoneElement
 from fastmcp import FastMCP
 
 from tools.tools import compress_html, requests_html, dp_headless_html, assert_waf_cookie, dp_mcp_message_pack
@@ -71,7 +70,8 @@ def register_close_tab(mcp: FastMCP, browser_manager):
 
 def register_check_selector(mcp: FastMCP, browser_manager):
     @mcp.tool(name="check_selector", description="查找tab页中是否包含元素，并返回元素attr_name所对应的值")
-    async def check_selector(browser_port: int, tab_id: str, css_selector: str, attr_name: str = "text") -> dict[str, Any]:
+    async def check_selector(browser_port: int, tab_id: str, css_selector: str, attr_name: str = "text") -> dict[
+        str, Any]:
         _browser = browser_manager.get_browser(browser_port)
         target_tab = _browser.get_tab(tab_id)
         css_selector = css_selector
@@ -157,10 +157,18 @@ def register_assert_waf(mcp: FastMCP, browser_manager):
             head_headless_rate_difference=h_hless_rate_diff
         )
 
-# def register_highlight_element_captcha(mcp: FastMCP, browser_manager):
-#     @mcp.tool(name="highlight_element_captcha",
-#               description="将传入的Selector在页面上高亮，并截屏")
-#     async def highlight_element_captcha(browser_port: int, tab_id: str, selector: str) -> dict[str, Any]:
-#         _browser = browser_manager.get_browser(browser_port)
-#         tab = _browser.get_tab(tab_id)
-#         tab.ele
+
+def register_highlight_element_captcha(mcp: FastMCP, browser_manager):
+    @mcp.tool(name="highlight_element_captcha", description="将传入的Selector在页面上高亮，并截屏")
+    async def highlight_element_captcha(browser_port: int, tab_id: str, css_selector: str) -> dict[str, Any]:
+        _browser = browser_manager.get_browser(browser_port)
+        tab = _browser.get_tab(tab_id)
+        highlight_style = 'background-color: rgba(255, 255, 0, 0.2); outline: 2px solid yellow !important; outline-offset: 2px'
+        css_selector = css_selector
+        if "css:" not in css_selector:
+            css_selector = "css:" + css_selector
+        
+        target_eles = tab.eles(css_selector)
+        exist_flag = False
+        if len(target_eles) != 0:
+            exist_flag = True

{jarvis_brain-0.1.5.5 → jarvis_brain-0.1.5.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "Jarvis_Brain"  # 别人下载时用的名字，必须在 PyPI 上唯一
-version = "0.1.5.5"
+version = "0.1.5.6"
 description = "Jarvis brain mcp"
 dependencies = [
     "fastmcp",

{jarvis_brain-0.1.5.5 → jarvis_brain-0.1.5.6}/tools/browser_manager.py

@@ -3,6 +3,7 @@ import random
 from typing import Optional, Tuple
 import os
 from DrissionPage import ChromiumPage, ChromiumOptions
+import platform
 
 
 class BrowserManager:
@@ -22,6 +23,8 @@ class BrowserManager:
             random_port = random.randint(9223, 9934)
 
         co = ChromiumOptions().set_local_port(random_port)
+        if platform.system() != 'Windows':
+            co.set_argument('--no-sandbox')
         custom_data_dir = os.path.join(os.path.expanduser('~'), 'DrissionPage', "userData", f"{random_port}")
         co.set_user_data_path(custom_data_dir)  # 设置用户数据路径
         # if not os.path.exists(custom_data_dir):