custom_controller.py•2.52 kB
# -*- coding: utf-8 -*-
import logging
import pyperclip
from browser_use.agent.views import ActionResult
from browser_use.browser.context import BrowserContext
from browser_use.controller.service import Controller
logger = logging.getLogger(__name__)
class CustomController(Controller):
"""
A custom controller registering two clipboard actions: copy and paste.
"""
def __init__(self):
super().__init__()
self._register_custom_actions()
def _register_custom_actions(self) -> None:
"""Register all custom browser actions for this controller."""
@self.registry.action("Copy text to clipboard")
def copy_to_clipboard(text: str) -> ActionResult:
"""
Copy the given text to the system's clipboard.
Returns an ActionResult with the same text as extracted_content.
"""
try:
pyperclip.copy(text)
# Be cautious about logging the actual text, if sensitive
logger.debug("Copied text to clipboard.")
return ActionResult(extracted_content=text)
except Exception as e:
logger.error(f"Error copying text to clipboard: {e}")
return ActionResult(error=str(e), extracted_content=None)
@self.registry.action("Paste text from clipboard", requires_browser=True)
async def paste_from_clipboard(browser: BrowserContext) -> ActionResult:
"""
Paste whatever is currently in the system's clipboard
into the active browser page by simulating keyboard typing.
"""
try:
text = pyperclip.paste()
except Exception as e:
logger.error(f"Error reading text from clipboard: {e}")
return ActionResult(error=str(e), extracted_content=None)
try:
# Send text to browser
page = await browser.get_current_page()
if page is None:
# It's possible there's no current page
raise RuntimeError("No active browser page found.")
await page.keyboard.type(text)
logger.debug("Pasted text from clipboard into the browser.")
return ActionResult(extracted_content=text)
except Exception as e:
logger.error(f"Error pasting text into the browser: {e}")
return ActionResult(error=str(e), extracted_content=None)