From 31875395094e622d19b5300e47dc7ed19bd0ffa8 Mon Sep 17 00:00:00 2001 From: yackob03 Date: Mon, 11 Nov 2013 17:01:21 -0500 Subject: [PATCH] Catch an error when trying to generate an access token. --- endpoints/index.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/endpoints/index.py b/endpoints/index.py index 671f8766f..d6f3ea448 100644 --- a/endpoints/index.py +++ b/endpoints/index.py @@ -33,10 +33,14 @@ def generate_headers(role='read'): if has_token_request: repo = model.get_repository(namespace, repository) - token = model.create_access_token(repo, role) - token_str = 'signature=%s' % token.code - response.headers['WWW-Authenticate'] = token_str - response.headers['X-Docker-Token'] = token_str + if repo: + token = model.create_access_token(repo, role) + token_str = 'signature=%s' % token.code + response.headers['WWW-Authenticate'] = token_str + response.headers['X-Docker-Token'] = token_str + else: + logger.info('Token request in non-existing repo: %s/%s' % + (namespace, repository)) return response return wrapper