Gecko [ c.stage.kornherr.net ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2024-07-10 10:57
cli	[ DIR ]	drwxr-xr-x	2024-07-10 10:57
config	[ DIR ]	drwxr-xr-x	2024-07-10 10:57
metrics	[ DIR ]	drwxr-xr-x	2024-07-10 10:57
__init__.py	23 B	-rw-r--r--	2021-05-10 10:30
__main__.py	42 B	-rw-r--r--	2021-05-10 10:30
const.py	1.25 KB	-rw-r--r--	2021-05-10 10:30
container.py	9.13 KB	-rw-r--r--	2021-05-10 10:30
errors.py	909 B	-rw-r--r--	2021-05-10 10:30
network.py	11.46 KB	-rw-r--r--	2021-05-10 10:30
parallel.py	11.29 KB	-rw-r--r--	2021-05-10 10:30
progress_stream.py	3.49 KB	-rw-r--r--	2021-05-10 10:30
project.py	41.37 KB	-rw-r--r--	2021-05-10 10:30
service.py	67.91 KB	-rw-r--r--	2021-05-10 10:30
timeparse.py	2.71 KB	-rw-r--r--	2021-05-10 10:30
utils.py	5.01 KB	-rw-r--r--	2021-05-10 10:30
version.py	177 B	-rw-r--r--	2021-05-10 10:30
volume.py	7.21 KB	-rw-r--r--	2021-05-10 10:30

Rename

from compose import utils

class StreamOutputError(Exception):
    pass

def write_to_stream(s, stream):
    try:
        stream.write(s)
    except UnicodeEncodeError:
        encoding = getattr(stream, 'encoding', 'ascii')
        stream.write(s.encode(encoding, errors='replace').decode(encoding))

def stream_output(output, stream):
    is_terminal = hasattr(stream, 'isatty') and stream.isatty()
    stream = stream
    lines = {}
    diff = 0

for event in utils.json_stream(output):
        yield event
        is_progress_event = 'progress' in event or 'progressDetail' in event

if not is_progress_event:
            print_output_event(event, stream, is_terminal)
            stream.flush()
            continue

if not is_terminal:
            continue

# if it's a progress event and we have a terminal, then display the progress bars
        image_id = event.get('id')
        if not image_id:
            continue

if image_id not in lines:
            lines[image_id] = len(lines)
            write_to_stream("\n", stream)

diff = len(lines) - lines[image_id]

# move cursor up `diff` rows
        write_to_stream("%c[%dA" % (27, diff), stream)

print_output_event(event, stream, is_terminal)

if 'id' in event:
            # move cursor back down
            write_to_stream("%c[%dB" % (27, diff), stream)

stream.flush()

def print_output_event(event, stream, is_terminal):
    if 'errorDetail' in event:
        raise StreamOutputError(event['errorDetail']['message'])

terminator = ''

if is_terminal and 'stream' not in event:
        # erase current line
        write_to_stream("%c[2K\r" % 27, stream)
        terminator = "\r"
    elif 'progressDetail' in event:
        return

if 'time' in event:
        write_to_stream("[%s] " % event['time'], stream)

if 'id' in event:
        write_to_stream("%s: " % event['id'], stream)

if 'from' in event:
        write_to_stream("(from %s) " % event['from'], stream)

status = event.get('status', '')

if 'progress' in event:
        write_to_stream("{} {}{}".format(status, event['progress'], terminator), stream)
    elif 'progressDetail' in event:
        detail = event['progressDetail']
        total = detail.get('total')
        if 'current' in detail and total:
            percentage = float(detail['current']) / float(total) * 100
            write_to_stream('{} ({:.1f}%){}'.format(status, percentage, terminator), stream)
        else:
            write_to_stream('{}{}'.format(status, terminator), stream)
    elif 'stream' in event:
        write_to_stream("{}{}".format(event['stream'], terminator), stream)
    else:
        write_to_stream("{}{}\n".format(status, terminator), stream)

def get_digest_from_pull(events):
    digest = None
    for event in events:
        status = event.get('status')
        if not status or 'Digest' not in status:
            continue
        else:
            digest = status.split(':', 1)[1].strip()
    return digest

def get_digest_from_push(events):
    for event in events:
        digest = event.get('aux', {}).get('Digest')
        if digest:
            return digest
    return None

def read_status(event):
    status = event['status'].lower()
    if 'progressDetail' in event:
        detail = event['progressDetail']
        if 'current' in detail and 'total' in detail:
            percentage = float(detail['current']) / float(detail['total'])
            status = '{} ({:.1%})'.format(status, percentage)
    return status