Jarvis-Brain 0.1.11.10__tar.gz → 0.1.13.9__tar.gz

{jarvis_brain-0.1.11.10 → jarvis_brain-0.1.13.9}/PKG-INFO

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

{jarvis_brain-0.1.11.10 → jarvis_brain-0.1.13.9}/mcp_tools/dp_tools.py

@@ -7,7 +7,7 @@ import os
 import time
 from typing import Any
 
-import DrissionPage
+from DrissionPage.errors import NoRectError
 from fastmcp import FastMCP
 
 from tools.browser_manager import BrowserManager
@@ -248,22 +248,17 @@ def register_click_action(mcp: FastMCP, browser_manager):
         if "css:" not in css_selector:
             css_selector = "css:" + css_selector
         target_eles = target_tab.eles(css_selector)
-        # click_success = False
-        # element_clickable = False
-        # if len(target_eles) == 1:
         target_element = target_eles[target_element_index]
         element_clickable = target_element.states.is_clickable
         try:
-            target_element.click()
-            click_success = True
+            click_success = target_element.click()
+            click_success = click_success is not False
         except Exception as e:
             click_success = False
         if target_element_index > 0:
             message = f"tab页:【{tab_id}】点击【{css_selector}】【index={target_element_index}】的元素 {'成功' if click_success else '失败'} 了"
         else:
             message = f"tab页:【{tab_id}】点击【{css_selector}】 {'成功' if click_success else '失败'} 了"
-        # else:
-        #     message = f"tab页:【{tab_id}】传入的css_selector找到了{len(target_eles)}个元素，请确保传入的css_selector可以找到唯一的一个元素"
         return dp_mcp_message_pack(
             message=message,
             browser_port=browser_port,
@@ -289,23 +284,28 @@ def register_scroll_action(mcp: FastMCP, browser_manager):
         if forward == "down":
             if pixel is None:
                 target_tab.scroll.to_half()
-            target_tab.scroll.down(pixel)
+            else:
+                target_tab.scroll.down(pixel)
         elif forward == "up":
             if pixel is None:
                 target_tab.scroll.to_top()
-            target_tab.scroll.up(pixel)
+            else:
+                target_tab.scroll.up(pixel)
         elif forward == "left":
             if pixel is None:
                 target_tab.scroll.to_leftmost()
-            target_tab.scroll.left(pixel)
+            else:
+                target_tab.scroll.left(pixel)
         elif forward == "right":
             if pixel is None:
                 target_tab.scroll.to_rightmost()
-            target_tab.scroll.right(pixel)
+            else:
+                target_tab.scroll.right(pixel)
         else:
             if pixel is None:
                 target_tab.scroll.to_half()
-            target_tab.scroll.down()
+            else:
+                target_tab.scroll.down()
         message = f"已完成对tab页:【{tab_id}】forward={forward} 的滑动"
         return dp_mcp_message_pack(
             message=message,
@@ -314,6 +314,117 @@ def register_scroll_action(mcp: FastMCP, browser_manager):
         )
 
 
+def register_mouse_hover(mcp: FastMCP, browser_manager):
+    @mcp.tool(name="mouse_hover", description="将鼠标悬停在元素上【这个功能使用了Drissionpage的action行为链功能】")
+    async def mouse_hover(browser_port: int, tab_id: str, css_selector: str, target_element_index: int = 0) -> dict[
+        str, Any]:
+        _browser = browser_manager.get_browser(browser_port)
+        target_tab = _browser.get_tab(tab_id)
+        css_selector = css_selector
+        if "css:" not in css_selector:
+            css_selector = "css:" + css_selector
+        target_eles = target_tab.eles(css_selector)
+        target_element = target_eles[target_element_index]
+        try:
+            target_tab.actions.move_to(target_element)
+            target_element.hover()
+            hover_success = True
+        except Exception as e:
+            hover_success = False
+        if target_element_index > 0:
+            message = f"tab页:【{tab_id}】hover【{css_selector}】【index={target_element_index}】的元素 {'成功' if hover_success else '失败'} 了"
+        else:
+            message = f"tab页:【{tab_id}】hover【{css_selector}】 {'成功' if hover_success else '失败'} 了"
+        # else:
+        #     message = f"tab页:【{tab_id}】传入的css_selector找到了{len(target_eles)}个元素，请确保传入的css_selector可以找到唯一的一个元素"
+        return dp_mcp_message_pack(
+            message=message,
+            browser_port=browser_port,
+            tab_id=tab_id,
+            css_selector=css_selector,
+            hoversuccess=hover_success,
+            extra_message="hover成功，页面可能有更新，请重新获取页面html，并重新分析页面Selector" if hover_success else ""
+        )
+
+
+def register_get_ele_info(mcp: FastMCP, browser_manager):
+    @mcp.tool(name="get_ele_info", description="返回元素有关一系列的信息。"
+                                               "参数说明："
+                                               "browser_port：浏览器端口号。"
+                                               "tab_id：tab页的id。"
+                                               "css_selector：目标选择器"
+                                               "element_index：如果选择器定位到的是一个列表【或者可以定位到多个元素】，则该值用于定位这个列表中具体元素，默认值为0【列表中的第一个元素】"
+                                               "返回值说明："
+                                               "element_tag：此属性返回元素的标签名。"
+                                               "element_attrs_key：此属性以list的形式返回元素所有属性的key。"
+                                               "element_rect_size：此属性以元组形式返回元素的大小【如果元素没有位置及大小，则返回空元组】。"
+                                               "is_in_viewport：此属性以布尔值方式返回元素是否在视口中，以元素可以接受点击的点为判断。"
+                                               "is_whole_in_viewport：此属性以布尔值方式返回元素是否整个在视口中。"
+                                               "is_alive：此属性以布尔值形式返回当前元素是否仍可用。用于判断是否因页面刷新而导致元素失效。"
+                                               "is_checked：此属性以布尔值返回表单单选或多选元素是否选中。"
+                                               "is_selected：此属性以布尔值返回<select>元素中的项是否选中。"
+                                               "is_enabled：此属性以布尔值返回元素是否可用。"
+                                               "is_displayed：此属性以布尔值返回元素是否可见。"
+                                               "is_covered：此属性返回元素是否被其它元素覆盖/遮挡。如被覆盖/遮挡，返回覆盖元素的 id【无障碍树的id，无法用于Selector】，否则返回False。"
+                                               "is_clickable：此属性返回元素是否可被模拟点击，从是否有大小、是否可用、是否显示、是否响应点击判断，不判断是否被遮挡。"
+                                               "has_rect：此属性返回元素是否拥有大小和位置信息，有则返回四个角在页面上的坐标组成的列表，没有则返回False。")
+    async def get_ele_info(browser_port: int, tab_id: str, css_selector: str, element_index: int = 0) -> dict[
+        str, Any]:
+        _browser = browser_manager.get_browser(browser_port)
+        target_tab = _browser.get_tab(tab_id)
+        css_selector = css_selector
+        if "css:" not in css_selector:
+            css_selector = "css:" + css_selector
+        target_eles = target_tab.eles(css_selector)
+        try:
+            target_element = target_eles[element_index]
+        except IndexError:
+            return dp_mcp_message_pack(
+                message="报错：IndexError: list index out of range。请检查Selector中是否包含了如：first-child、nth-child等字段，如果有则去掉。必须使用element_index来控制元素的选择",
+                browser_port=browser_port,
+                tab_id=tab_id,
+                css_selector=css_selector,
+                element_index=element_index,
+            )
+        try:
+            has_rect = target_element.states.has_rect,
+            element_rect_size = tuple()
+            if not has_rect:
+                element_rect_size = target_element.rect.size,
+            try:
+                child_count=target_element.child_count
+            except Exception:
+                child_count=0
+            return dp_mcp_message_pack(
+                message="元素可以被正常的选择到，以下是元素相关的一系列信息",
+                browser_port=browser_port,
+                tab_id=tab_id,
+                css_selector=css_selector,
+                element_index=element_index,
+                element_tag=target_element.tag,
+                element_attrs_key=list(target_element.attrs.keys()),
+                element_child_count=child_count,
+                element_rect_size=element_rect_size,
+                is_in_viewport=target_element.states.is_in_viewport,
+                is_whole_in_viewport=target_element.states.is_whole_in_viewport,
+                is_alive=target_element.states.is_alive,
+                is_checked=target_element.states.is_checked,
+                is_selected=target_element.states.is_selected,
+                is_enabled=target_element.states.is_enabled,
+                is_displayed=target_element.states.is_displayed,
+                is_covered=target_element.states.is_covered,
+                is_clickable=target_element.states.is_clickable,
+                has_rect=has_rect
+            )
+        except NoRectError:
+            return dp_mcp_message_pack(
+                message="报错：NoRectError: 该元素没有位置及大小。你传入的css_selector和element_index选出的元素没有rect，请确认该元素是否可见",
+                browser_port=browser_port,
+                tab_id=tab_id,
+                css_selector=css_selector,
+                element_index=element_index,
+            )
+
 def register_get_screenshot(mcp: FastMCP, browser_manager):
     @mcp.tool(name="get_tab_screenshot",
               description="尝试对传入tab页进行截图，并将截图压缩为1M大小png图片，会返回截图保存路径")
@@ -324,10 +435,10 @@ def register_get_screenshot(mcp: FastMCP, browser_manager):
         if not os.path.exists(html_source_code_local_save_path):
             os.makedirs(html_source_code_local_save_path)
         timestamp = int(time.time() * 1000)
-        # time.sleep(1)
-        origin_png = target_tab.get_screenshot(as_bytes="jpg", full_page=True)
-        compress_png = compress_image_bytes(origin_png, 0.5)
-        image_path = os.path.join(html_source_code_local_save_path, f"{browser_port}_{tab_id}_{timestamp}.jpg")
+        time.sleep(3)
+        origin_png = target_tab.get_screenshot(as_bytes="png")
+        compress_png = compress_image_bytes(origin_png)
+        image_path = os.path.join(html_source_code_local_save_path, f"{browser_port}_{tab_id}_{timestamp}.png")
         with open(image_path, "wb") as f:
             f.write(compress_png)
         return dp_mcp_message_pack(

{jarvis_brain-0.1.11.10 → jarvis_brain-0.1.13.9}/mcp_tools/main.py

@@ -24,6 +24,8 @@ if "TeamNode-Dp" in enabled_modules:
     # 页面交互
     register_click_action(mcp, browser_manager)
     register_scroll_action(mcp, browser_manager)
+    register_mouse_hover(mcp, browser_manager)
+    register_get_ele_info(mcp, browser_manager)
 
 if "JarvisNode" in enabled_modules:
     register_assert_waf(mcp, browser_manager)

{jarvis_brain-0.1.11.10 → jarvis_brain-0.1.13.9}/pyproject.toml

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

{jarvis_brain-0.1.11.10 → jarvis_brain-0.1.13.9}/tools/tools.py

@@ -10,7 +10,7 @@ import base64
 from PIL import Image
 import io
 
-compress_html_js = """
+compress_html_js1 = """
 function getSimplifiedDOM(node) {
     // 1. 处理文本节点
     if (node.nodeType === Node.TEXT_NODE) {
@@ -102,6 +102,101 @@ function getSimplifiedDOM(node) {
 return getSimplifiedDOM(document.body);
 """
 
+# 我自己优化后的版本，逻辑为：删除不可见元素、标签的任何属性value的长度大于20时直接删除这个属性、id和class采用简写方式：id=>#，class=>.
+compress_html_js="""
+function getSimplifiedDOM(node) {
+    // 全局配置：最大属性值长度
+    const MAX_ATTR_LEN = 40;
+
+    // 1. 处理文本节点
+    if (node.nodeType === Node.TEXT_NODE) {
+        const text = node.textContent.trim();
+        return text ? text.slice(0, 100) + (text.length > 100 ? '...' : '') : null;
+    }
+
+    // 2. 过滤无用标签
+    const ignoreTags = ['SCRIPT', 'STYLE', 'NOSCRIPT', 'IFRAME', 'LINK', 'META', 'AUDIO', 'VIDEO', 'CANVAS'];
+    if (ignoreTags.includes(node.tagName)) return null;
+    if (node.nodeType !== Node.ELEMENT_NODE) return null;
+
+    // 3. 过滤不可见元素
+    const style = window.getComputedStyle(node);
+    if (style.display === 'none' || style.visibility === 'hidden' || style.opacity === '0') return null;
+    
+    const rect = node.getBoundingClientRect();
+    if ((rect.width === 0 || rect.height === 0) && style.overflow !== 'visible') return null;
+
+    // --- 开始构建标签字符串 ---
+    const tagName = node.tagName.toLowerCase();
+    let tagStr = tagName;
+
+    const id = node.id;
+    const className = node.getAttribute('class');
+
+    // A. 处理 ID 简写 (#id)
+    // 限制提高到 40
+    if (id && id.length <= MAX_ATTR_LEN) {
+        tagStr += `#${id}`;
+    }
+
+    // B. 处理 Class 简写 (.class)
+    // 限制提高到 40
+    if (className && typeof className === 'string' && className.length <= MAX_ATTR_LEN) {
+        const classes = className.trim().split(/\s+/);
+        if (classes.length > 0) {
+            tagStr += `.${classes.join('.')}`;
+        }
+    }
+
+    let propsStr = '';
+
+    // C. 处理属性
+    if (node.hasAttributes()) {
+        for (const attr of node.attributes) {
+            const name = attr.name;
+            const value = attr.value;
+
+            // 1. 跳过 ID 和 Class (已在 tagStr 处理，或因过长被丢弃)
+            if (name === 'id' || name === 'class') continue;
+
+            // 2. 黑名单：直接删除 style 和 aria-label
+            if (name === 'style' || name === 'aria-label') continue;
+
+            // 3. 特殊标签：path 标签删除所有属性
+            if (tagName === 'path') continue;
+
+            // 4. 【长度与白名单逻辑】
+            // 如果不是 src 且不是 href，同时长度又超过了 40，则删除
+            const isLinkAttr = (name === 'src' || name === 'href');
+            
+            if (!isLinkAttr && value.length > MAX_ATTR_LEN) {
+                continue; 
+            }
+
+            // 5. 拼接保留的属性
+            propsStr += ` ${name}="${value.replace(/"/g, '&quot;')}"`;
+        }
+    }
+
+    // 4. 递归子节点
+    let childNodes = Array.from(node.childNodes);
+    if (node.shadowRoot) {
+        childNodes = [...childNodes, ...Array.from(node.shadowRoot.childNodes)];
+    }
+    
+    const children = childNodes
+        .map(getSimplifiedDOM)
+        .filter(n => n !== null);
+    
+    // 5. 组装输出
+    if (children.length === 0) {
+        return `<${tagStr}${propsStr} />`;
+    }
+    return `<${tagStr}${propsStr}>${children.join('')}</${tagName}>`; 
+}
+
+return getSimplifiedDOM(document.body);
+"""
 
 # 使用requests获取html，用于测试是否使用了瑞数和jsl
 def requests_html(url):
@@ -268,11 +363,3 @@ def compress_image_bytes(input_bytes, target_size_mb=1):
 
     return output_bytes
 
-# todo: 大致盘一下各种判定的逻辑【以下的所有压缩比之间的差距均取“绝对值”】
-#  1. 如果requests、无头、有头获取到的压缩比之间从差距都在15%以内，则认定该页面是静态页面，此时优先使用requests请求
-#  2. 如果requests的status_code为特定的412，或者521，则判定是瑞数和jsl。[此时还有一个特点：requests的压缩比会与其他两种方式获取到的压缩比差距非常大(一两千的那种)]
-#  3. 如果requests、无头、有头获取到的压缩比之间差距都在40%以上，则判定该页面只可以用有头采集
-#  4. 如果无头和有头获取到的压缩比之间差距小于15%，但是requests和无头的差距大于40%，则认定该页面可以使用无头浏览器采集
-#  5. 如果requests和有头获取到的压缩比之间差距小于15%，但是无头和有头的差距大于40%，则认定该页面优先使用有头浏览器采集
-#  【此时可能是：1.使用了别的检测无头的waf。2.网站使用瑞数，但是这次请求没有拦截requests（不知道是不是瑞数那边故意设置的），
-#   此时如果想进一步判定是否是瑞数，可以使用有头浏览器取一下cookies，如果cookies里面存在瑞数的cookie，那么就可以断定是瑞数】