Skip to content

Commit 33bbd91

Browse files
committed
feat: add get current URL + blocked domains
1 parent 3495cfe commit 33bbd91

2 files changed

Lines changed: 39 additions & 0 deletions

File tree

src/scrapybara/client.py

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@
3131
AuthStateResponse,
3232
BrowserAuthenticateResponse,
3333
BrowserGetCdpUrlResponse,
34+
BrowserGetCurrentUrlResponse,
3435
Button,
3536
ClickMouseActionClickType,
3637
ComputerResponse,
@@ -136,6 +137,13 @@ def get_cdp_url(
136137
return self._client.browser.get_cdp_url(
137138
self.instance_id, request_options=request_options
138139
)
140+
141+
def get_current_url(
142+
self, request_options: Optional[RequestOptions] = None
143+
) -> BrowserGetCurrentUrlResponse:
144+
return self._client.browser.get_current_url(
145+
self.instance_id, request_options=request_options
146+
)
139147

140148
def save_auth(
141149
self,
@@ -196,6 +204,13 @@ async def get_cdp_url(
196204
return await self._client.browser.get_cdp_url(
197205
self.instance_id, request_options=request_options
198206
)
207+
208+
async def get_current_url(
209+
self, request_options: Optional[RequestOptions] = None
210+
) -> BrowserGetCurrentUrlResponse:
211+
return await self._client.browser.get_current_url(
212+
self.instance_id, request_options=request_options
213+
)
199214

200215
async def save_auth(
201216
self,
@@ -1020,6 +1035,13 @@ def get_cdp_url(
10201035
return self._client.browser.get_cdp_url(
10211036
self.id, request_options=request_options
10221037
)
1038+
1039+
def get_current_url(
1040+
self, request_options: Optional[RequestOptions] = None
1041+
) -> BrowserGetCurrentUrlResponse:
1042+
return self._client.browser.get_current_url(
1043+
self.id, request_options=request_options
1044+
)
10231045

10241046
def save_auth(
10251047
self,
@@ -1460,6 +1482,13 @@ async def get_cdp_url(
14601482
return await self._client.browser.get_cdp_url(
14611483
self.id, request_options=request_options
14621484
)
1485+
1486+
async def get_current_url(
1487+
self, request_options: Optional[RequestOptions] = None
1488+
) -> BrowserGetCurrentUrlResponse:
1489+
return await self._client.browser.get_current_url(
1490+
self.id, request_options=request_options
1491+
)
14631492

14641493
async def save_auth(
14651494
self,
@@ -1534,11 +1563,13 @@ def start_ubuntu(
15341563
self,
15351564
*,
15361565
timeout_hours: Optional[float] = OMIT,
1566+
blocked_domains: Optional[Sequence[str]] = OMIT,
15371567
request_options: Optional[RequestOptions] = None,
15381568
) -> UbuntuInstance:
15391569
response = self._base_client.start(
15401570
instance_type="ubuntu",
15411571
timeout_hours=timeout_hours,
1572+
blocked_domains=blocked_domains,
15421573
request_options=request_options,
15431574
)
15441575
return UbuntuInstance(
@@ -1552,11 +1583,13 @@ def start_browser(
15521583
self,
15531584
*,
15541585
timeout_hours: Optional[float] = OMIT,
1586+
blocked_domains: Optional[Sequence[str]] = OMIT,
15551587
request_options: Optional[RequestOptions] = None,
15561588
) -> BrowserInstance:
15571589
response = self._base_client.start(
15581590
instance_type="browser",
15591591
timeout_hours=timeout_hours,
1592+
blocked_domains=blocked_domains,
15601593
request_options=request_options,
15611594
)
15621595
return BrowserInstance(
@@ -1946,11 +1979,13 @@ async def start_ubuntu(
19461979
self,
19471980
*,
19481981
timeout_hours: Optional[float] = OMIT,
1982+
blocked_domains: Optional[Sequence[str]] = OMIT,
19491983
request_options: Optional[RequestOptions] = None,
19501984
) -> AsyncUbuntuInstance:
19511985
response = await self._base_client.start(
19521986
instance_type="ubuntu",
19531987
timeout_hours=timeout_hours,
1988+
blocked_domains=blocked_domains,
19541989
request_options=request_options,
19551990
)
19561991
return AsyncUbuntuInstance(
@@ -1964,11 +1999,13 @@ async def start_browser(
19641999
self,
19652000
*,
19662001
timeout_hours: Optional[float] = OMIT,
2002+
blocked_domains: Optional[Sequence[str]] = OMIT,
19672003
request_options: Optional[RequestOptions] = None,
19682004
) -> AsyncBrowserInstance:
19692005
response = await self._base_client.start(
19702006
instance_type="browser",
19712007
timeout_hours=timeout_hours,
2008+
blocked_domains=blocked_domains,
19722009
request_options=request_options,
19732010
)
19742011
return AsyncBrowserInstance(

src/scrapybara/types/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from .bash_response import BashResponse
44
from .browser_authenticate_response import BrowserAuthenticateResponse
55
from .browser_get_cdp_url_response import BrowserGetCdpUrlResponse
6+
from .browser_get_current_url_response import BrowserGetCurrentUrlResponse
67
from .button import Button
78
from .cell_type import CellType
89
from .click_mouse_action import ClickMouseAction
@@ -78,6 +79,7 @@
7879
"BashResponse",
7980
"BrowserAuthenticateResponse",
8081
"BrowserGetCdpUrlResponse",
82+
"BrowserGetCurrentUrlResponse",
8183
"Button",
8284
"CellType",
8385
"ClickMouseAction",

0 commit comments

Comments
 (0)