全球最受欢迎的网站设计师作品展示网站

全球最受欢迎的网站,设计师作品展示网站,杭州室内设计设计公司前十排名,门户网站编辑流程BrowserUse14-源码-ScreenShot模块-整理ScreenShot模块 1-源码部分【下载】基于图片的Base64进行图片的#xff0c;【查看】指定本地的文件路径进行查看 浏览器使用代理的截图存储服务。 import base64 from pathlib import Pathimport a…BrowserUse14-源码-ScreenShot模块-整理ScreenShot模块1-源码部分【下载】基于图片的Base64进行图片的【查看】指定本地的文件路径进行查看 浏览器使用代理的截图存储服务。 importbase64frompathlibimportPathimportanyiofrombrowser_use_manual.observabilityimportobserve_debugclassScreenshotService:简单的截图存储服务将截图保存到磁盘def__init__(self,agent_directory:str|Path):初始化时传入代理目录路径self.agent_directoryPath(agent_directory)ifisinstance(agent_directory,str)elseagent_directory# 创建 screenshots 子目录self.screenshots_dirself.agent_directory/screenshotsself.screenshots_dir.mkdir(parentsTrue,exist_okTrue)observe_debug(ignore_inputTrue,ignore_outputTrue,namestore_screenshot)asyncdefstore_screenshot(self,screenshot_b64:str,step_number:int)-str:将截图保存到磁盘并返回完整路径字符串screenshot_filenamefstep_{step_number}.pngscreenshot_pathself.screenshots_dir/screenshot_filename# 解码 base64 并保存到磁盘screenshot_database64.b64decode(screenshot_b64)asyncwithawaitanyio.open_file(screenshot_path,wb)asf:awaitf.write(screenshot_data)returnstr(screenshot_path)observe_debug(ignore_inputTrue,ignore_outputTrue,nameget_screenshot_from_disk)asyncdefget_screenshot(self,screenshot_path:str)-str|None:从磁盘路径加载截图并以 base64 格式返回ifnotscreenshot_path:returnNonepathPath(screenshot_path)ifnotpath.exists():returnNone# 从磁盘读取并编码为 base64asyncwithawaitanyio.open_file(path,rb)asf:screenshot_dataawaitf.read()returnbase64.b64encode(screenshot_data).decode(utf-8)2-测试用例测试【截图】模块的【本地图片的下载和读取】和【远程文件的下载和读取】#!/usr/bin/env python3 Screenshot 截图服务测试用例 测试读取本地图片文件并转换为base64编码然后使用截图服务进行测试 # 添加项目根目录到 Python 路径以支持导入importsysfrompathlibimportPath sys.path.insert(0,str(Path(__file__).parent.parent.parent))importbase64importasyncioimportloggingimporthttpxfrombrowser_use_manual.screenshots.serviceimportScreenshotService# 配置日志logging.basicConfig(levellogging.INFO,format%(levelname)s - %(name)s - %(message)s)loggerlogging.getLogger(__name__)asyncdeftest_get_screenshot_from_file(image_path:str):测试从文件读取截图并转换为base64# 检查文件是否存在pathPath(image_path)ifnotpath.exists():logger.error(f测试图片文件不存在:{image_path})returnFalse# 读取图片文件并转换为base64withopen(path,rb)asimage_file:image_dataimage_file.read()image_b64base64.b64encode(image_data).decode(utf-8)logger.info(f图片文件 {image_path} 已转换为base64长度:{len(image_b64)}字符)# 创建截图服务实例agent_dirPath(__file__).parent.parent/test_datascreenshot_serviceScreenshotService(agent_dir)# 使用截图服务存储截图stored_pathawaitscreenshot_service.store_screenshot(image_b64,1)logger.info(f截图已存储到:{stored_path})# 使用截图服务读取截图retrieved_b64awaitscreenshot_service.get_screenshot(stored_path)ifretrieved_b64image_b64:logger.info(✓ 成功: 从磁盘读取的截图与原图一致)returnTrueelse:logger.error(✗ 失败: 从磁盘读取的截图与原图不一致)returnFalseasyncdeftest_get_screenshot_from_url(image_url:str):测试从URL下载图片并转换为base64try:# 下载远程图片asyncwithhttpx.AsyncClient()asclient:responseawaitclient.get(image_url)response.raise_for_status()image_dataresponse.content image_b64base64.b64encode(image_data).decode(utf-8)logger.info(f远程图片 {image_url} 已下载并转换为base64长度:{len(image_b64)}字符)# 创建截图服务实例agent_dirPath(__file__).parent.parent/test_datascreenshot_serviceScreenshotService(agent_dir)# 使用截图服务存储截图stored_pathawaitscreenshot_service.store_screenshot(image_b64,2)logger.info(f截图已存储到:{stored_path})# 使用截图服务读取截图retrieved_b64awaitscreenshot_service.get_screenshot(stored_path)ifretrieved_b64image_b64:logger.info(✓ 成功: 从磁盘读取的截图与下载的图片一致)returnTrueelse:logger.error(✗ 失败: 从磁盘读取的截图与下载的图片不一致)returnFalseexceptExceptionase:logger.error(f下载或处理远程图片时出错:{e})returnFalseasyncdefrun_all_tests():运行所有测试# 1-测试本地文件如果指定了有效路径test_image_path/Users/rong/Documents/EnzoApplication/WorkSpace/Python/20251209_1_Python_playwright_manual/browser-use-manual-file/test_screen_image.jpgpathPath(test_image_path)ifpath.exists():logger.info(f开始测试本地图片文件:{test_image_path})local_successawaittest_get_screenshot_from_file(test_image_path)else:logger.warning(f本地测试图片文件不存在:{test_image_path})local_successTrue# 不影响远程图片测试# 2-测试远程图片remote_image_urlhttps://i-blog.csdnimg.cn/direct/bee22ad9c3b74f07b8568843e48fe08e.pnglogger.info(f开始测试远程图片URL:{remote_image_url})remote_successawaittest_get_screenshot_from_url(remote_image_url)returnlocal_successandremote_successdefmain():# 指定要测试的图片文件路径logger.info(测试截图服务包括本地文件和远程图片)# 运行所有测试successasyncio.run(run_all_tests())ifsuccess:print(\n 截图服务测试全部通过!)else:print(\n❌ 截图服务某些测试失败!)if__name____main__:main()