moto/wait_for.py

32 lines
692 B
Python
Raw Normal View History

2018-12-21 11:28:56 +00:00
import time
try:
# py2
import urllib2 as urllib
from urllib2 import URLError
import socket
import httplib
EXCEPTIONS = (URLError, socket.error, httplib.BadStatusLine)
except ImportError:
# py3
import urllib.request as urllib
from urllib.error import URLError
EXCEPTIONS = (URLError, ConnectionResetError)
start_ts = time.time()
print("Waiting for service to come up")
while True:
try:
urllib.urlopen('http://localhost:5000/', timeout=1)
break
except EXCEPTIONS:
elapsed_s = time.time() - start_ts
if elapsed_s > 60:
raise
print('.')
time.sleep(1)