1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
"""
urllib2.HTTPPasswordMgr object using the keyring, for use with the
urllib2.HTTPBasicAuthHandler.
usage:
import urllib2
handlers = [urllib2.HTTPBasicAuthHandler(PasswordMgr())]
urllib2.install_opener(handlers)
urllib2.urlopen(...)
This will prompt for a password if one is required and isn't already
in the keyring. Then, it adds it to the keyring for subsequent use.
"""
import keyring
import getpass
class PasswordMgr(object):
def get_username(self, realm, authuri):
return getpass.getuser()
def add_password(self, realm, authuri, password):
keyring.set_password(realm, user, passwd)
def find_user_password(self, realm, authuri):
user = self.get_username(realm, authuri)
passwd = keyring.get_password(realm, user)
if passwd is None:
prompt = 'password for %(user)s@%(realm)s for '\
'%(authuri)s: ' % vars()
passwd = getpass.getpass(prompt)
keyring.set_password(realm, user, passwd)
return user, passwd
|
