Isolate STOP logic in multiplexer module

Signed-off-by: Aanand Prasad <aanand.prasad@gmail.com>
This commit is contained in:
Aanand Prasad 2015-07-22 14:09:12 +01:00
parent 03c3d4c768
commit 27378704df
2 changed files with 3 additions and 2 deletions

View File

@ -4,7 +4,7 @@ import sys
from itertools import cycle
from .multiplexer import Multiplexer, STOP
from .multiplexer import Multiplexer
from . import colors
from .utils import split_buffer
@ -61,7 +61,6 @@ class LogPrinter(object):
exit_code = container.wait()
yield color_fn("%s exited with code %s\n" % (container.name, exit_code))
yield STOP
def _generate_prefix(self, container):
"""

View File

@ -45,3 +45,5 @@ class Multiplexer(object):
def _enqueue_output(iterator, queue):
for item in iterator:
queue.put(item)
queue.put(STOP)