diff options
Diffstat (limited to 'netlib/http_auth.py')
| -rw-r--r-- | netlib/http_auth.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/netlib/http_auth.py b/netlib/http_auth.py index 71f120d6..69bee5c1 100644 --- a/netlib/http_auth.py +++ b/netlib/http_auth.py @@ -33,6 +33,7 @@ class NullProxyAuth(): class BasicProxyAuth(NullProxyAuth): CHALLENGE_HEADER = 'Proxy-Authenticate' AUTH_HEADER = 'Proxy-Authorization' + def __init__(self, password_manager, realm): NullProxyAuth.__init__(self, password_manager) self.realm = realm @@ -125,8 +126,10 @@ class AuthAction(Action): """ def __call__(self, parser, namespace, values, option_string=None): passman = self.getPasswordManager(values) + if not passman: + raise ArgumentTypeError("Error creating password manager for proxy authentication.") authenticator = BasicProxyAuth(passman, "mitmproxy") - setattr(namespace, "authenticator", authenticator) + setattr(namespace, self.dest, authenticator) def getPasswordManager(self, s): # pragma: nocover raise NotImplementedError() |