Fix cleanup of connections.

Author: thatnerdjosh

shared/wait-for-psql.py

@@ -16,16 +16,18 @@
 
     args = arg_parser.parse_args()
 
+    conn = None
     start_time = time.time()
+    error = ''
+    
     while (time.time() - start_time) < args.timeout:
         try:
             conn = psycopg2.connect(user=args.db_user, host=args.db_host, port=args.db_port, password=args.db_password, dbname=args.db_name)
-            error = ''
+            conn.close()
             break
         except psycopg2.OperationalError as e:
             error = e
-        else:
-            conn.close()
+
         time.sleep(1)
 
     if error: