mirror of
https://github.com/browser-use/browser-use.git
synced 2026-09-14 19:59:47 +08:00
ShawnPana
7706c55532
- Create commands/utils.py with SDK singleton and format_duration helper - Merge task_management.py into cloud_task.py (SDK wrappers + CLI handlers) - Create cloud_session.py from session parts of cloud_task.py + session_management.py - Delete sdk.py, task_management.py, session_management.py - Update imports in main.py, agent.py, profile.py, __init__.py Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
47 lines
1.1 KiB
Python
47 lines
1.1 KiB
Python
"""Shared utilities for CLI command handlers."""
|
|
|
|
from datetime import datetime, timezone
|
|
|
|
from browser_use_sdk import BrowserUse
|
|
|
|
_client: BrowserUse | None = None
|
|
|
|
|
|
def get_sdk_client() -> BrowserUse:
|
|
"""Get authenticated SDK client (singleton)."""
|
|
global _client
|
|
if _client is None:
|
|
from browser_use.skill_cli.api_key import require_api_key
|
|
|
|
api_key = require_api_key('Cloud API')
|
|
_client = BrowserUse(api_key=api_key)
|
|
return _client
|
|
|
|
|
|
def format_duration(started_at: datetime | None, finished_at: datetime | None) -> str:
|
|
"""Format duration between two timestamps, or elapsed time if still running."""
|
|
if not started_at:
|
|
return ''
|
|
|
|
try:
|
|
if finished_at:
|
|
end = finished_at
|
|
else:
|
|
end = datetime.now(timezone.utc)
|
|
|
|
delta = end - started_at
|
|
total_seconds = int(delta.total_seconds())
|
|
|
|
if total_seconds < 60:
|
|
return f'{total_seconds}s'
|
|
elif total_seconds < 3600:
|
|
minutes = total_seconds // 60
|
|
seconds = total_seconds % 60
|
|
return f'{minutes}m {seconds}s'
|
|
else:
|
|
hours = total_seconds // 3600
|
|
minutes = (total_seconds % 3600) // 60
|
|
return f'{hours}h {minutes}m'
|
|
except Exception:
|
|
return ''
|