Added websocket bridge use as a template tag

Mark Gibbs · Mark Gibbs · commit ca21f68ba8b1 · 2018-06-13T19:31:18.000-07:00
diff --git a/demo/demo/templates/index.html b/demo/demo/templates/index.html
@@ -1,6 +1,5 @@
 <!DOCTYPE HTML>
 <html>
-{%load staticfiles%}
 {%load plotly_dash%}
 <title>Simple stuff</title>
 <body>
@@ -17,14 +16,5 @@
     {%plotly_app name="Ex2"%}
   </div>
 </body>
-<script src="{%static "channels/js/websocketbridge.js"%}"></script>
-<script>
-  const wsb = new channels.WebSocketBridge();
-  wsb.connect("/ws/channel");
-  wsb.listen(function(action, stream) {
-    console.log("WSB inbound");
-    console.log(action);
-    console.log(stream);
-  });
-</script>
+{%plotly_message_pipe%}
 </html>
diff --git a/django_plotly_dash/templates/django_plotly_dash/plotly_messaging.html b/django_plotly_dash/templates/django_plotly_dash/plotly_messaging.html
@@ -0,0 +1,11 @@
+{%load staticfiles%}
+<script src="{%static "channels/js/websocketbridge.js"%}"></script>
+<script>
+  const wsb = new channels.WebSocketBridge();
+  wsb.connect("{{url}}");
+  wsb.listen(function(action, stream) {
+    console.log("WSB inbound");
+    console.log(action);
+    console.log(stream);
+  });
+</script>
diff --git a/django_plotly_dash/templatetags/plotly_dash.py b/django_plotly_dash/templatetags/plotly_dash.py
@@ -38,3 +38,7 @@ def plotly_app(context, name=None, slug=None, da=None, ratio=0.1, use_frameborde
 
     return locals()
 
+@register.inclusion_tag("django_plotly_dash/plotly_messaging.html", takes_context=True)
+def plotly_message_pipe(context, url=None):
+    url = url and url or '/ws/channel'
+    return locals()
