Added timeout param to Client.wait

This commit is contained in:
Joffrey F 2014-12-15 17:32:42 -08:00
parent 7fcddc22ce
commit 51c6b34859
1 changed files with 2 additions and 2 deletions

View File

@ -1038,11 +1038,11 @@ class Client(requests.Session):
res = self._post(url)
self._raise_for_status(res)
def wait(self, container):
def wait(self, container, timeout=None):
if isinstance(container, dict):
container = container.get('Id')
url = self._url("/containers/{0}/wait".format(container))
res = self._post(url, timeout=None)
res = self._post(url, timeout=timeout)
self._raise_for_status(res)
json_ = res.json()
if 'StatusCode' in json_: