Continue developing

bazarr/check_update.py

@@ -12,7 +12,7 @@ import json
 
 from get_args import args
 from config import settings, bazarr_url
-from queueconfig import q4ws
+from queueconfig import q4ws, q4ws_updater
 
 
 def check_releases():
@@ -137,7 +137,7 @@ def check_updates():
         update()
     
     elif commits_behind == 0:
-        q4ws.append('BAZZAR is up to date')
+        q4ws_updater.append('BAZZAR is up to date')
         logging.info('BAZZAR is up to date')
     
     return latest_version

bazarr/main.py

@@ -1886,6 +1886,26 @@ def handle_websocket():
             break
 
 
+@route(base_url + 'websocket_updater')
+@custom_auth_basic(check_credentials)
+def handle_websocket():
+    wsock = request.environ.get('wsgi.websocket')
+    if not wsock:
+        abort(400, 'Expected WebSocket request.')
+    
+    queueconfig.q4ws_updater.clear()
+    
+    while True:
+        try:
+            if queueconfig.q4ws_updater:
+                wsock.send(queueconfig.q4ws_updater.popleft())
+                gevent.sleep(0.1)
+            else:
+                gevent.sleep(0.5)
+        except WebSocketError:
+            break
+
+
 # Mute DeprecationWarning
 warnings.simplefilter("ignore", DeprecationWarning)
 server = WSGIServer((str(settings.general.ip), (int(args.port) if args.port else int(settings.general.port))), app, handler_class=WebSocketHandler)

bazarr/queueconfig.py

@@ -1,4 +1,6 @@
 from collections import deque
 
 global q4ws
+global q4ws_updater
 q4ws = deque(maxlen=10)
+q4ws_updater = deque(maxlen=2)

views/menu.tpl

@@ -225,8 +225,10 @@
 	if (location.protocol != 'https:')
 	{
 		var ws = new WebSocket("ws://" + window.location.host + "{{base_url}}websocket");
+        var wsupdater = new WebSocket("ws://" + window.location.host + "{{base_url}}websocket_updater");
 	} else {
 		var ws = new WebSocket("wss://" + window.location.host + "{{base_url}}websocket");
+        var wsupdater = new WebSocket("wss://" + window.location.host + "{{base_url}}websocket_updater");
 	}
 
     ws.onmessage = function (evt) {
@@ -244,4 +246,21 @@
             theme: 'semanticui'
         }).show();
     };
+
+    wsupdater.onmessage = function (evt) {
+        new Noty({
+            text: evt.data,
+            timeout: false,
+			progressBar: false,
+			animation: {
+				open: null,
+				close: null
+			},
+			killer: true,
+    		type: 'info',
+            layout: 'bottomLeft',
+            theme: 'semanticui',
+            visibilityControl: true
+		}).show();
+    };
 </script>