From 9bb63160cc48111b0a57acaed85956cb9b82c59d Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Sun, 27 May 2018 04:33:08 +0200
Subject: stop a running process, if you have the uuid

---
 rpc/rpc.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'rpc/rpc.py')

diff --git a/rpc/rpc.py b/rpc/rpc.py
index 4b634a9..010715e 100644
--- a/rpc/rpc.py
+++ b/rpc/rpc.py
@@ -3,6 +3,7 @@ import os
 import zerorpc
 import gevent
 import time
+import sys
 from io import BytesIO
 import base64
 from PIL import Image
@@ -34,12 +35,16 @@ class CortexRPC(object):
     val = None
     try:
       val = self.relay.connected('rpc')
-      atexit.register(lambda: self.relay.disconnecting())
+      atexit.register(self.disconnecting)
       print('RPC connected...'.format(val))
     except zerorpc.TimeoutExpired:
       print('RPC timeout :(')
     return val
 
+  def disconnecting(self):
+    print('Disconnecting from relay', flush=True, file=sys.stderr)
+    self.relay.disconnecting()
+
   def ping(self):
     return 'pong'
 
-- 
cgit v1.2.3-70-g09d2