Fix browsertrix WebSocket proxy

VIDA-NYU · Aug 21, 2023 · dcf0bf8 · dcf0bf8
1 parent 30e71af
commit dcf0bf8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/reproserver/web/webcapture.py b/reproserver/web/webcapture.py
@@ -515,6 +515,8 @@ def get(self, upload_short_id, run_short_id):
 
 
 class CrawlStatusWebsocket(WebSocketHandler, BaseHandler):
+    upstream_ws = None
+
     def get(self, upload_short_id, run_short_id):
         # Decode info from URL
         try:
@@ -566,7 +568,8 @@ def on_upstream_message(self, message):
             return self.write_message(message, isinstance(message, bytes))
 
     def on_ws_connection_close(self, close_code=None, close_reason=None):
-        self.upstream_ws.close(close_code, close_reason)
+        if self.upstream_ws is not None:
+            self.upstream_ws.close(close_code, close_reason)
 
 
 class UploadWacz(BaseHandler):