Fix the updated_tags information by storing it in the session

endpoints/index.py

@@ -302,10 +302,6 @@ def update_images(namespace, repository):
     profile.debug('Parsing image data')
     image_with_checksums = json.loads(request.data.decode('utf8'))
 
-    updated_tags = {}
-    for image in image_with_checksums:
-      updated_tags[image['Tag']] = image['id']
-
     if get_authenticated_user():
       profile.debug('Publishing push event')
       username = get_authenticated_user().username
@@ -326,6 +322,7 @@ def update_images(namespace, repository):
     # Generate a job for each notification that has been added to this repo
     profile.debug('Adding notifications for repository')
 
+    updated_tags = session.get('pushed_tags', {})
     event_data = {
       'updated_tags': updated_tags,
       'pushed_image_count': len(image_with_checksums),

endpoints/tags.py

@@ -2,7 +2,7 @@
 import logging
 import json
 
-from flask import abort, request, jsonify, make_response, Blueprint
+from flask import abort, request, jsonify, make_response, Blueprint, session
 
 from app import app
 from util.names import parse_repository_name
@@ -59,6 +59,12 @@ def put_tag(namespace, repository, tag):
     docker_image_id = json.loads(request.data)
     model.create_or_update_tag(namespace, repository, tag, docker_image_id)
 
+    # Store the updated tag.
+    if not 'pushed_tags' in session:
+      session['pushed_tags'] = {}
+
+    session['pushed_tags'][tag] = docker_image_id
+
     return make_response('Created', 200)
 
   abort(403)