From c6f93bd3c609bd4cb87e2966549b74c628a952a0 Mon Sep 17 00:00:00 2001
From: Athanasius <github@miggy.org>
Date: Mon, 23 Aug 2021 15:36:21 +0100
Subject: [PATCH] companion: Update "close down" check for using EDMCAPIRequest

---
 companion.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/companion.py b/companion.py
index a1c9b401..a689e957 100644
--- a/companion.py
+++ b/companion.py
@@ -803,11 +803,15 @@ class Session(object):
             querytime: int
             play_sound: bool
             auto_update: bool
-            # endpoint, querytime, play_sound, auto_update = self.capi_query_queue.get()
             query = self.capi_query_queue.get()
-            if not query.endpoint:
-                logger.info('Empty queue message, exiting...')
-                break
+            if not isinstance(query, EDMCCAPIRequest):
+                if query is not None:
+                    logger.error("Item from queue wasn't an EDMCCAPIRequest")
+                    break
+
+                else:
+                    logger.info('Empty queue message, exiting...')
+                    break
 
             logger.trace_if('capi.worker', f'Processing query: {query.endpoint}')
             data: CAPIData