From fa1fc7f9a2def79b1944e673c7362ebb46822378 Mon Sep 17 00:00:00 2001 From: yackob03 Date: Thu, 26 Sep 2013 17:58:41 -0400 Subject: [PATCH] Fix the tag image fetch model method. --- data/model.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/data/model.py b/data/model.py index cfaaec44e..2a1bef234 100644 --- a/data/model.py +++ b/data/model.py @@ -144,9 +144,14 @@ def list_repository_tags(namespace_name, repository_name): def get_tag_image(namespace_name, repository_name, tag_name): joined = Image.select().join(RepositoryTag).join(Repository) - return joined.where(Repository.name == repository_name, - Repository.namespace == namespace_name, - RepositoryTag.name == tag_name) + fetched = list(joined.where(Repository.name == repository_name, + Repository.namespace == namespace_name, + RepositoryTag.name == tag_name)) + + if not fetched: + raise Image.DoesNotExist('Unable to find image for tag.') + + return fetched[0] def create_or_update_tag(namespace_name, repository_name, tag_name,