Merge pull request #365 from jkroll-deepgram/handle-exit-event

Handle NoneType exit event
deepgram · Apr 16, 2024 · 7461d02 · 7461d02
2 parents 81e2940 + 201843f
commit 7461d02
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/deepgram/clients/live/v1/async_client.py b/deepgram/clients/live/v1/async_client.py
@@ -184,7 +184,7 @@ async def _listening(self) -> None:
 
         while True:
             try:
-                if self._exit_event.is_set():
+                if self._exit_event is not None and self._exit_event.is_set():
                     self.logger.notice("_listening exiting gracefully")
                     self.logger.debug("AsyncLiveClient._listening LEAVE")
                     return
@@ -336,7 +336,7 @@ async def _keep_alive(self) -> None:
                 counter += 1
                 await asyncio.sleep(ONE_SECOND)
 
-                if self._exit_event.is_set():
+                if self._exit_event is not None and self._exit_event.is_set():
                     self.logger.notice("_keep_alive exiting gracefully")
                     self.logger.debug("AsyncLiveClient._keep_alive LEAVE")
                     return
@@ -409,7 +409,7 @@ async def send(self, data: Union[str, bytes]) -> bool:
         """
         self.logger.spam("AsyncLiveClient.send ENTER")
 
-        if self._exit_event.is_set():
+        if self._exit_event is not None and self._exit_event.is_set():
             self.logger.notice("send exiting gracefully")
             self.logger.debug("AsyncLiveClient.send LEAVE")
             return False

diff --git a/deepgram/clients/live/v1/client.py b/deepgram/clients/live/v1/client.py
@@ -184,7 +184,7 @@ def _listening(self) -> None:
 
         while True:
             try:
-                if self._exit_event.is_set():
+                if self._exit_event is not None and self._exit_event.is_set():
                     self.logger.notice("_listening exiting gracefully")
                     self.logger.debug("LiveClient._listening LEAVE")
                     return
@@ -336,7 +336,7 @@ def _keep_alive(self) -> None:
                 counter += 1
 
                 self._exit_event.wait(timeout=ONE_SECOND)
-                if self._exit_event.is_set():
+                if self._exit_event is not None and self._exit_event.is_set():
                     self.logger.notice("_keep_alive exiting gracefully")
                     self.logger.debug("LiveClient._keep_alive LEAVE")
                     return
@@ -407,7 +407,7 @@ def send(self, data: Union[str, bytes]) -> bool:
         """
         self.logger.spam("LiveClient.send ENTER")
 
-        if self._exit_event.is_set():
+        if self._exit_event is not None and self._exit_event.is_set():
             self.logger.notice("send exiting gracefully")
             self.logger.debug("AsyncLiveClient.send LEAVE")
             return False