Make sure attach() is called as soon as LogPrinter is initialized

Fixes #35.
This commit is contained in:
Aanand Prasad 2014-01-21 17:58:04 +00:00 committed by Ben Firshman
parent 4ee87a7029
commit 65071aafb0
1 changed files with 3 additions and 2 deletions

View File

@ -31,8 +31,9 @@ class LogPrinter(object):
def _make_log_generator(self, container, color_fn):
prefix = color_fn(container.name + " | ")
for line in split_buffer(self._attach(container), '\n'):
yield prefix + line
# Attach to container before log printer starts running
line_generator = split_buffer(self._attach(container), '\n')
return (prefix + line for line in line_generator)
def _attach(self, container):
params = {