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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
'''
Task Coach - Your friendly task manager
Copyright (C) 2011 Task Coach developers <developers@taskcoach.org>
Task Coach is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
Task Coach is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
'''
import wx
from taskcoachlib.i18n import _
class KeychainPasswordWidget(wx.Dialog):
def __init__(self, domain, username, *args, **kwargs):
super(KeychainPasswordWidget, self).__init__(*args, **kwargs)
self.domain = domain.encode('UTF-8')
self.username = username.encode('UTF-8')
pnl = wx.Panel(self, wx.ID_ANY)
hsz = wx.BoxSizer(wx.HORIZONTAL)
hsz.Add(wx.StaticText(pnl, wx.ID_ANY, _('Password:')), 0, wx.ALL, 3)
from keyring import get_password
password = get_password(self.domain, self.username)
self.password = (password or '').decode('UTF-8')
self.passwordField = wx.TextCtrl(pnl, wx.ID_ANY, self.password, style=wx.TE_PASSWORD)
hsz.Add(self.passwordField, 1, wx.ALL, 3)
vsz = wx.BoxSizer(wx.VERTICAL)
vsz.Add(hsz, 0, wx.ALL|wx.EXPAND, 3)
self.keepInKeychain = wx.CheckBox(pnl, wx.ID_ANY, _('Store in keychain'))
self.keepInKeychain.SetValue(bool(password))
vsz.Add(self.keepInKeychain, 0, wx.ALL|wx.EXPAND, 3)
hsz = wx.BoxSizer(wx.HORIZONTAL)
btnOK = wx.Button(pnl, wx.ID_ANY, _('OK'))
hsz.Add(btnOK, 0, wx.ALL, 3)
btnCancel = wx.Button(pnl, wx.ID_ANY, _('Cancel'))
hsz.Add(btnCancel, 0, wx.ALL, 3)
vsz.Add(hsz, 0, wx.ALL|wx.ALIGN_CENTRE, 3)
pnl.SetSizer(vsz)
sz = wx.BoxSizer(wx.HORIZONTAL)
sz.Add(pnl, 1, wx.EXPAND|wx.ALL, 3)
self.SetSizer(sz)
self.Fit()
wx.EVT_BUTTON(btnOK, wx.ID_ANY, self.OnOK)
wx.EVT_BUTTON(btnCancel, wx.ID_ANY, self.OnCancel)
self.SetDefaultItem(btnOK)
wx.CallAfter(self.RequestUserAttention)
def OnOK(self, event):
self.password = self.passwordField.GetValue()
from keyring import set_password
if self.keepInKeychain.GetValue():
set_password(self.domain, self.username, self.password.encode('UTF-8'))
else:
set_password(self.domain, self.username, '')
self.EndModal(wx.ID_OK)
def OnCancel(self, event):
self.EndModal(wx.ID_CANCEL)
_PASSWORDCACHE = None
def _GetCachedPassword(domain, username, reset):
global _PASSWORDCACHE
if _PASSWORDCACHE is None:
import StringIO, traceback
bf = StringIO.StringIO()
traceback.print_exc(file=bf)
wx.MessageBox(_('There was a problem trying to find out your system\'s keychain.\nPlease file a bug report (see the Help menu) and attach a screenshot of this message.\nError was:\n\n%s') % bf.getvalue(), _('Error'), wx.OK)
_PASSWORDCACHE = dict()
if (domain, username) in _PASSWORDCACHE and reset:
del _PASSWORDCACHE[(domain, username)]
if (domain, username) not in _PASSWORDCACHE:
pwd = wx.GetPasswordFromUser(_('Please enter your password.'), domain)
if not pwd:
return None
_PASSWORDCACHE[(domain, username)] = pwd
return _PASSWORDCACHE[(domain, username)]
def GetPassword(domain, username, reset=False):
try:
from keyring import set_password, get_password
except:
# Keychain unavailable.
return _GetCachedPassword(domain, username, reset)
try:
if reset:
set_password(domain.encode('UTF-8'), username.encode('UTF-8'), '')
else:
pwd = get_password(domain.encode('UTF-8'), username.encode('UTF-8'))
if pwd:
return pwd.decode('UTF-8')
except ImportError:
# Bug seen on Ubuntu 13.10: secretstorage cannot import ._gi
return _GetCachedPassword(domain, username, reset)
dlg = KeychainPasswordWidget(domain, username, None, wx.ID_ANY, _('Please enter your password'), style=wx.DEFAULT_DIALOG_STYLE|wx.STAY_ON_TOP)
try:
dlg.CentreOnScreen()
if dlg.ShowModal() == wx.ID_OK:
return dlg.password
finally:
dlg.Destroy()
|
