Browse Source

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

Fixes #35.
Aanand Prasad 12 years ago
parent
commit
65071aafb0
1 changed files with 3 additions and 2 deletions
  1. 3 2
      fig/cli/log_printer.py

+ 3 - 2
fig/cli/log_printer.py

@@ -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 = {