def __init__(self, reviewboard_url, rb_user=None, rb_pass=None,
api_token=None, auth_callback=None, otp_token_callback=None):
HTTPPasswordMgr.__init__(self)
self.passwd = {}
self.rb_url = reviewboard_url
self.rb_user = rb_user
self.rb_pass = rb_pass
self.api_token = api_token
self.auth_callback = auth_callback
self.otp_token_callback = otp_token_callback
def __init__(self, reviewboard_url, rb_user=None, rb_pass=None,
api_token=None, auth_callback=None, otp_token_callback=None):
HTTPPasswordMgr.__init__(self)
self.passwd = {}
self.rb_url = reviewboard_url
self.rb_user = rb_user
self.rb_pass = rb_pass
self.api_token = api_token
self.auth_callback = auth_callback
self.otp_token_callback = otp_token_callback
def find_user_password(self, realm, uri):
if realm == 'Web API':
if self.auth_callback:
username, password = self.auth_callback(realm, uri,
username=self.rb_user,
password=self.rb_pass)
self.rb_user = username
self.rb_pass = password
return self.rb_user, self.rb_pass
else:
# If this is an auth request for some other domain (since HTTP
# handlers are global), fall back to standard password management.
return HTTPPasswordMgr.find_user_password(self, realm, uri)
def find_user_password(self, realm, uri):
if realm == 'Web API':
if self.auth_callback:
username, password = self.auth_callback(realm, uri,
username=self.rb_user,
password=self.rb_pass)
self.rb_user = username
self.rb_pass = password
return self.rb_user, self.rb_pass
else:
# If this is an auth request for some other domain (since HTTP
# handlers are global), fall back to standard password management.
return HTTPPasswordMgr.find_user_password(self, realm, uri)
