Implement 'docker-compose push' and 'docker-compose bundle'

Signed-off-by: Aanand Prasad <aanand.prasad@gmail.com>

1 files changed, 19 insertions, 0 deletions

diff --git a/compose/progress_stream.py b/compose/progress_stream.py
index 1f873d1d..a0f5601f 100644
--- a/compose/progress_stream.py
+++ b/compose/progress_stream.py
@@ -91,3 +91,22 @@ def print_output_event(event, stream, is_terminal):
         stream.write("%s%s" % (event['stream'], terminator))
     else:
         stream.write("%s%s\n" % (status, terminator))
+
+
+def get_digest_from_pull(events):
+    for event in events:
+        status = event.get('status')
+        if not status or 'Digest' not in status:
+            continue
+
+        _, digest = status.split(':', 1)
+        return digest.strip()
+    return None
+
+
+def get_digest_from_push(events):
+    for event in events:
+        digest = event.get('aux', {}).get('Digest')
+        if digest:
+            return digest
+    return None