Validate connection before enter the main loop

chanfung032 · chanfung032 · commit f9d9ac1ff156 · 2013-06-20T13:41:51.000+08:00
diff --git a/dev_server/cloudsql.py b/dev_server/cloudsql.py
@@ -91,6 +91,11 @@ def do_desc(self, args):
     def do_load(self, args):
         self.do_run(args)
 
+    def preloop(self, *args, **kwargs):
+        sqlcmd.SQLCmd.preloop(self, *args, **kwargs)
+        # Just exit if the connect failed
+        if self._SQLCmd__db is None: sys.exit(1)
+
     def __init__(self, *args, **kwargs):
         sqlcmd.SQLCmd.__init__(self, *args, **kwargs)
         self.prompt = sqlcmd.SQLCmd.MAIN_PROMPT
