Add stdout capture

server/hello_world.py

@@ -4,7 +4,7 @@
 
 import numpy as np
 
-from helper import Api, rpc
+from helper import Api, rpc, forward_stdout
 
 class AlgorithmApi(Api):
     def __init__(self):
@@ -48,6 +48,7 @@ def params(self):
         ]
 
     @rpc('run')
+    @forward_stdout
     def run(self, params):
         input_image = params['input_image']
         labelmap = params['input_labelmap']

server/helper.py

@@ -1,3 +1,4 @@
+import sys
 import weakref
 import uuid
 
@@ -8,6 +9,42 @@
 from serializable import serialize, unserialize
 import transformers # register our serializers/unserializers
 
+class WsOut(object):
+    def __init__(self, protocol):
+        self.protocol = protocol
+
+    def write(self, b):
+        self.protocol.publish('streams.stdout', b)
+
+    def flush(self):
+        pass
+
+class NetworkTee(object):
+    def __init__(self, netout):
+        self.netout = netout
+        self.stdout = sys.stdout
+
+    def write(self, b):
+        self.netout.write(b)
+        self.stdout.write(b)
+
+    def flush(self):
+        self.netout.flush()
+        self.stdout.flush()
+
+
+def forward_stdout(fn):
+    def handler(self, *args, **kwargs):
+        wsout = WsOut(self)
+        tee = NetworkTee(wsout)
+        sys.stdout = tee
+
+        ret = fn(self, *args, **kwargs)
+
+        sys.stdout = tee.stdout
+        return ret
+    return handler
+
 def rpc(name):
     def wrapper(fn):
         def handler(self, *args, **kwargs):

src/components/tools/RemoteProcessing/script.js

@@ -17,6 +17,7 @@ export default {
   computed: {
     ...mapState('remote', {
       processing: 'processing',
+      serverStdout: 'serverStdout',
       parameters: (state) => state.paramOrder.map((name) => state.params[name]),
     }),
   },
@@ -61,6 +62,7 @@ export default {
     ...mapActions('remote', {
       fetchParamList: 'fetchParamList',
       runRemoteAlgorithm: 'runRemoteAlgorithm',
+      clearOutput: 'clearStdout',
       setParameter: (dispatch, name, value) => {
         dispatch('setParameter', { name, value });
       },

src/components/tools/RemoteProcessing/style.css

@@ -5,3 +5,10 @@
 .container * {
   user-select: none;
 }
+
+.serverOutput {
+  width: 100%;
+  white-space: nowrap;
+  border-radius: 3px;
+  border: 1px solid grey;
+}

src/components/tools/RemoteProcessing/template.html

@@ -71,4 +71,13 @@
       {{ error }}
     </v-alert>
   </v-layout>
+  <v-layout wrap align-center class="mt-3">
+    <v-flex xs10><span class="caption">Server output</span></v-flex>
+    <v-flex xs2>
+      <v-btn icon @click="clearOutput"><v-icon>mdi-delete</v-icon></v-btn>
+    </v-flex>
+    <v-flex xs12>
+      <textarea :class="$style.serverOutput">{{ serverStdout }}</textarea>
+    </v-flex>
+  </v-layout>
 </v-container>

src/store/remote.js

@@ -15,6 +15,7 @@ export default ({ proxyManager, remote }) => ({
     processing: false,
     params: {},
     paramOrder: [],
+    serverStdout: '',
   }),
 
   mutations: {
@@ -39,13 +40,24 @@ export default ({ proxyManager, remote }) => ({
     processing(state, flag) {
       state.processing = flag;
     },
+    appendStdout(state, text) {
+      state.serverStdout += text;
+    },
+    clearStdout(state) {
+      state.serverStdout = '';
+    },
   },
 
   actions: {
     connect({ commit }, endpoint) {
       return remote
         .connect(endpoint)
-        .then(() => commit('connected'))
+        .then(() => {
+          commit('connected');
+          remote.session.subscribe('streams.stdout', (stdout) =>
+            commit('appendStdout', stdout)
+          );
+        })
         .catch((error) => commit('connectError', error));
     },
     fetchParamList({ commit }) {
@@ -80,5 +92,6 @@ export default ({ proxyManager, remote }) => ({
       return promise;
     },
     setParameter: wrapMutationAsAction('setParameter'),
+    clearStdout: wrapMutationAsAction('clearStdout'),
   },
 });