From 86b61a8fa35e5fe40af1ef01df9dcbd11f2f3cec Mon Sep 17 00:00:00 2001
From: Dan Jones <danjon@noc.ac.uk>
Date: Tue, 11 Feb 2025 13:36:55 +0000
Subject: [PATCH] fix: remove unused imports

---
 api_wrap_socket.py | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/api_wrap_socket.py b/api_wrap_socket.py
index da51098..640d5ed 100644
--- a/api_wrap_socket.py
+++ b/api_wrap_socket.py
@@ -18,17 +18,16 @@
 # On disconnect the queue consumer is cancelled so that new 
 # messages are left in the queue
 
-import concurrent.futures
 import json
 import uuid
 import time
-from twisted.internet import reactor, threads
+from twisted.internet import reactor
 from twisted.web.server import Site
 from twisted.web.resource import Resource
 from twisted.web.wsgi import WSGIResource
 from autobahn.exception import Disconnected
 from autobahn.twisted.resource import WebSocketResource
-from autobahn.twisted.websocket import WebSocketServerFactory, WebSocketServerProtocol, listenWS
+from autobahn.twisted.websocket import WebSocketServerFactory, WebSocketServerProtocol
 from models.client_model import ClientModel
 from models.token_model import TokenModel
 from api_rest import create_app
@@ -68,16 +67,12 @@ class SubscriptionServerProtocol(WebSocketServerProtocol):
             if valid:
                 try:
                     client_id = parsed.get('client_id')
-                    client = self.apiClients.get(parsed.get('client_id'))
+                    client = self.apiClients.get(client_id)
                     if client: 
                         self.client = client
                         allow = True
                 except Exception: 
                     allow = False
-        
-        if allow:
-            # setup queue consumer
-            SubscriptionListener.createListener(self)
                 
         return allow
         
@@ -92,6 +87,9 @@ class SubscriptionServerProtocol(WebSocketServerProtocol):
         Try authentication
         """
         allow = self.auth(request.headers)
+        if allow:
+            # setup queue consumer
+            SubscriptionListener.createListener(self)
              
     def onMessage(self, payload, isBinary):
         if not isBinary:
-- 
GitLab