exit after command executed

diff --git a/sztool b/sztool
index 82f547d..6ff1a9e 100755 (executable)
--- a/sztool
+++ b/sztool
@@ -3,6 +3,7 @@
 from subprocess import call, Popen, PIPE, STDOUT
 import paho.mqtt.client as mqtt
 import argparse
+import sys
 
 MQTT_SERVER = 'mqtt.suanzi.ai'
 MQTT_PORT = 1883
@@ -20,6 +21,7 @@ def on_message(client, userdata, msg):
     if msg.topic == id + '-response':
         port = msg.payload
         print 'run "ssh ' + URI + ' -p ' + str(port) + '" to connect to device (%s)' % id
+        sys.exit(0)
 
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='Choose an avaiable port and run ssh.')