Merge pull request #264 from oliverlee/fix-server-shutdown-error

Don't assume user enters input in server shutdown
pydy · Oct 14, 2015 · d57dac0 · d57dac0
2 parents e43a79c + 4270cf6
commit d57dac0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/pydy/viz/server.py b/pydy/viz/server.py
@@ -115,8 +115,7 @@ def _stop_server(self, signal, frame):
 
         """
         res = raw_input("Shutdown this visualization server ([y]/n)? ")
-        res = res.lower()[0]
-        if res == '' or res == 'y':
+        if not res or res[0].lower() == 'y':
             print("Shutdown confirmed")
             print("Shutting down server...")
             self.httpd.stop()