Improvements for psql wait script.

* support: Move to shared directory to avoid code drift.
* feat: Make DB name configurable

Author: thatnerdjosh

15.0/wait-for-psql.py

@@ -10,21 +10,24 @@
     arg_parser.add_argument('--db_host', required=True)
     arg_parser.add_argument('--db_port', required=True)
     arg_parser.add_argument('--db_user', required=True)
+    arg_parser.add_argument('--db_name', required=True)
     arg_parser.add_argument('--db_password', required=True)
     arg_parser.add_argument('--timeout', type=int, default=5)
 
     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='postgres')
-            error = ''
+            conn = psycopg2.connect(user=args.db_user, host=args.db_host, port=args.db_port, password=args.db_password, dbname=args.db_name)
+            conn.close()
             break
         except psycopg2.OperationalError as e:
             error = e
-        else:
-            conn.close()
+
         time.sleep(1)
 
     if error:

16.0/wait-for-psql.py

@@ -10,21 +10,24 @@
     arg_parser.add_argument('--db_host', required=True)
     arg_parser.add_argument('--db_port', required=True)
     arg_parser.add_argument('--db_user', required=True)
+    arg_parser.add_argument('--db_name', required=True)
     arg_parser.add_argument('--db_password', required=True)
     arg_parser.add_argument('--timeout', type=int, default=5)
 
     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='postgres')
-            error = ''
+            conn = psycopg2.connect(user=args.db_user, host=args.db_host, port=args.db_port, password=args.db_password, dbname=args.db_name)
+            conn.close()
             break
         except psycopg2.OperationalError as e:
             error = e
-        else:
-            conn.close()
+
         time.sleep(1)
 
     if error:

17.0/wait-for-psql.py

@@ -10,21 +10,24 @@
     arg_parser.add_argument('--db_host', required=True)
     arg_parser.add_argument('--db_port', required=True)
     arg_parser.add_argument('--db_user', required=True)
+    arg_parser.add_argument('--db_name', required=True)
     arg_parser.add_argument('--db_password', required=True)
     arg_parser.add_argument('--timeout', type=int, default=5)
 
     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='postgres')
-            error = ''
+            conn = psycopg2.connect(user=args.db_user, host=args.db_host, port=args.db_port, password=args.db_password, dbname=args.db_name)
+            conn.close()
             break
         except psycopg2.OperationalError as e:
             error = e
-        else:
-            conn.close()
+
         time.sleep(1)
 
     if error: