Update ListenerLayer.py

3mora2 · Apr 27, 2024 · 955fbd6 · 955fbd6
1 parent 57e8f58
commit 955fbd6
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/WPP_Whatsapp/api/layers/ListenerLayer.py b/WPP_Whatsapp/api/layers/ListenerLayer.py
@@ -1,4 +1,6 @@
 from event_emitter import EventEmitter
+from playwright.async_api import Error
+
 from WPP_Whatsapp.api.layers.ProfileLayer import ProfileLayer
 
 OnMessage = 'onMessage'
@@ -66,11 +68,16 @@ async def _afterPageScriptInjectedListener(self):
         ]
 
         for func in functions:
-            has = await self.ThreadsafeBrowser.page_evaluate("(func) => typeof window[func] === 'function'", func, page=self.page)
+            has = await self.ThreadsafeBrowser.page_evaluate("(func) => typeof window[func] === 'function'", func,
+                                                             page=self.page)
             if not has:
                 self.logger.debug(f'{self.session}: Exposing {func} function')
                 handel_func = HandelFunc(func, self.session, self.logger, self.__listenerEmitter).handel_func
-                await self.ThreadsafeBrowser.expose_function(func, handel_func, page=self.page)
+                try:
+                    await self.ThreadsafeBrowser.expose_function(func, handel_func, page=self.page)
+                except Error as e:
+                    if "has been already registered" not in e.message:
+                        raise e
 
         await self.ThreadsafeBrowser.page_evaluate("""() => {
         try {