Skip to content

Commit 0dafd8b

Browse files
committed
Add sb.cdp.reset_permissions()
1 parent 3f94a55 commit 0dafd8b

File tree

4 files changed

+13
-0
lines changed

4 files changed

+13
-0
lines changed

examples/cdp_mode/ReadMe.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -386,6 +386,7 @@ sb.cdp.get_navigation_history()
386386
sb.cdp.tile_windows(windows=None, max_columns=0)
387387
sb.cdp.grant_permissions(permissions, origin=None)
388388
sb.cdp.grant_all_permissions()
389+
sb.cdp.reset_permissions()
389390
sb.cdp.get_all_cookies(*args, **kwargs)
390391
sb.cdp.set_all_cookies(*args, **kwargs)
391392
sb.cdp.save_cookies(*args, **kwargs)

seleniumbase/core/browser_launcher.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -682,6 +682,7 @@ def uc_open_with_cdp_mode(driver, url=None, **kwargs):
682682
cdp.tile_windows = CDPM.tile_windows
683683
cdp.grant_permissions = CDPM.grant_permissions
684684
cdp.grant_all_permissions = CDPM.grant_all_permissions
685+
cdp.reset_permissions = CDPM.reset_permissions
685686
cdp.get_all_cookies = CDPM.get_all_cookies
686687
cdp.set_all_cookies = CDPM.set_all_cookies
687688
cdp.save_cookies = CDPM.save_cookies

seleniumbase/core/sb_cdp.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -662,6 +662,13 @@ def grant_all_permissions(self):
662662
driver = driver.cdp_base
663663
return self.loop.run_until_complete(driver.grant_all_permissions())
664664

665+
def reset_permissions(self):
666+
"""Reset permissions for all origins on the current window."""
667+
driver = self.driver
668+
if hasattr(driver, "cdp_base"):
669+
driver = driver.cdp_base
670+
return self.loop.run_until_complete(driver.reset_permissions())
671+
665672
def get_all_cookies(self, *args, **kwargs):
666673
driver = self.driver
667674
if hasattr(driver, "cdp_base"):

seleniumbase/undetected/cdp_driver/browser.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -571,6 +571,10 @@ async def grant_all_permissions(self):
571571
]
572572
await self.connection.send(cdp.browser.grant_permissions(permissions))
573573

574+
async def reset_permissions(self):
575+
"""Reset permissions for all origins on the current window."""
576+
await self.connection.send(cdp.browser.reset_permissions())
577+
574578
async def tile_windows(self, windows=None, max_columns: int = 0):
575579
import math
576580
try:

0 commit comments

Comments
 (0)