From 949b7072cabce0daed6c94993ad44c8ea8648dbd Mon Sep 17 00:00:00 2001 From: Colin Watson Date: Fri, 10 Nov 2017 15:50:40 +0000 Subject: Import py-macaroon-bakery_0.0.6.orig.tar.gz --- macaroonbakery/httpbakery/agent/agent.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) (limited to 'macaroonbakery/httpbakery') diff --git a/macaroonbakery/httpbakery/agent/agent.py b/macaroonbakery/httpbakery/agent/agent.py index 862f00e..ad56015 100644 --- a/macaroonbakery/httpbakery/agent/agent.py +++ b/macaroonbakery/httpbakery/agent/agent.py @@ -6,6 +6,7 @@ import json import nacl.public import nacl.encoding +import nacl.exceptions import requests.cookies import six from six.moves.urllib.parse import urlparse @@ -35,8 +36,10 @@ def load_agent_file(filename, cookies=None): with open(filename) as f: data = json.load(f) try: - key = nacl.public.PrivateKey(data['key']['private'], - nacl.encoding.Base64Encoder) + key = nacl.public.PrivateKey( + data['key']['private'], + nacl.encoding.Base64Encoder, + ) if cookies is None: cookies = requests.cookies.RequestsCookieJar() for agent in data['agents']: @@ -54,7 +57,7 @@ def load_agent_file(filename, cookies=None): path=u.path) cookies.set_cookie(cookie) return cookies, key - except (KeyError, ValueError) as e: + except (KeyError, ValueError, nacl.exceptions.TypeError) as e: raise AgentFileFormatError('invalid agent file', e) -- cgit v1.2.3