Add wait_for_service

imager/image.py

@@ -9,7 +9,7 @@ import tempfile
 import time
 
 
-parser = argparse.ArgumentParser(description='iconograph build_image')
+parser = argparse.ArgumentParser(description='iconograph image')
 parser.add_argument(
     '--base-url',
     dest='base_url',

imager/wait_for_service.py

@@ -0,0 +1,36 @@
+#!/usr/bin/python3
+
+import argparse
+import socket
+import time
+
+
+parser = argparse.ArgumentParser(description='iconograph wait_for_service')
+parser.add_argument(
+    '--host',
+    dest='host',
+    action='store',
+    required=True)
+parser.add_argument(
+    '--service',
+    dest='service',
+    action='store',
+    required=True)
+FLAGS = parser.parse_args()
+
+
+def main():
+  conn = (FLAGS.host, FLAGS.service)
+  print('Trying to connect to %s:%s' % conn)
+  while True:
+    try:
+      socket.create_connection(conn, timeout=5)
+      break
+    except Exception:
+      time.sleep(1)
+      continue
+  print('Connection successful')
+
+
+if __name__ == '__main__':
+  main()