bzr branch
http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
1 |
# Copyright (C) 2007-2010 Canonical Ltd
|
2898.3.5
by James Henstridge
Add copyright headers to new files |
2 |
#
|
3 |
# This program is free software; you can redistribute it and/or modify
|
|
4 |
# it under the terms of the GNU General Public License as published by
|
|
5 |
# the Free Software Foundation; either version 2 of the License, or
|
|
6 |
# (at your option) any later version.
|
|
7 |
#
|
|
8 |
# This program is distributed in the hope that it will be useful,
|
|
9 |
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
10 |
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
11 |
# GNU General Public License for more details.
|
|
12 |
#
|
|
13 |
# You should have received a copy of the GNU General Public License
|
|
14 |
# along with this program; if not, write to the Free Software
|
|
4183.7.1
by Sabin Iacob
update FSF mailing address |
15 |
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
2898.3.5
by James Henstridge
Add copyright headers to new files |
16 |
|
17 |
"""Functions to manage the user's Launchpad user ID.
|
|
18 |
||
19 |
This allows the user to configure their Launchpad user ID once, rather
|
|
20 |
than once for each place that needs to take it into account.
|
|
21 |
"""
|
|
22 |
||
6624
by Jelmer Vernooij
Merge Python3 porting work ('py3 pokes') |
23 |
from ... import ( |
5273.1.7
by Vincent Ladeuil
No more use of the get_transport imported *symbol*, all uses are through |
24 |
errors, |
25 |
trace, |
|
26 |
transport, |
|
27 |
)
|
|
6624
by Jelmer Vernooij
Merge Python3 porting work ('py3 pokes') |
28 |
from ...config import AuthenticationConfig, GlobalStack |
29 |
from ...i18n import gettext |
|
2898.3.1
by James Henstridge
API for storing and retrieving the Launchpad user ID that will be used |
30 |
|
31 |
LAUNCHPAD_BASE = 'https://launchpad.net/' |
|
32 |
||
33 |
||
34 |
class UnknownLaunchpadUsername(errors.BzrError): |
|
35 |
_fmt = "The user name %(user)s is not registered on Launchpad." |
|
36 |
||
37 |
||
38 |
class NoRegisteredSSHKeys(errors.BzrError): |
|
3834.1.1
by Martin Pool
Better message when you have no Launchpad SSH keys (#289148) |
39 |
_fmt = "The user %(user)s has not registered any SSH keys with Launchpad.\n" \ |
40 |
"See <https://launchpad.net/people/+me>"
|
|
2898.3.1
by James Henstridge
API for storing and retrieving the Launchpad user ID that will be used |
41 |
|
42 |
||
3777.1.12
by Aaron Bentley
Enable configuring ssh auth from launchpad-login, with auto-upgrade. |
43 |
class MismatchedUsernames(errors.BzrError): |
44 |
||
6740.1.1
by Jelmer Vernooij
Rename bazaar.conf to breezy.conf. |
45 |
_fmt = ('breezy.conf and authentication.conf disagree about launchpad' |
3777.1.12
by Aaron Bentley
Enable configuring ssh auth from launchpad-login, with auto-upgrade. |
46 |
' account name. Please re-run launchpad-login.') |
47 |
||
48 |
||
2898.3.6
by James Henstridge
Rename function arguments that are only intended for use by the tests. |
49 |
def get_lp_login(_config=None): |
3777.1.15
by Aaron Bentley
Update docs |
50 |
"""Return the user's Launchpad username. |
51 |
||
6740.1.1
by Jelmer Vernooij
Rename bazaar.conf to breezy.conf. |
52 |
:raises: MismatchedUsername if authentication.conf and breezy.conf
|
3777.1.15
by Aaron Bentley
Update docs |
53 |
disagree about username.
|
54 |
"""
|
|
2898.3.6
by James Henstridge
Rename function arguments that are only intended for use by the tests. |
55 |
if _config is None: |
6464.1.1
by Jelmer Vernooij
Merge launchpad configuration to config stacks. |
56 |
_config = GlobalStack() |
2898.3.6
by James Henstridge
Rename function arguments that are only intended for use by the tests. |
57 |
|
6464.1.1
by Jelmer Vernooij
Merge launchpad configuration to config stacks. |
58 |
username = _config.get('launchpad_username') |
3777.1.18
by Aaron Bentley
Fix None handling wrt auth upgrades |
59 |
if username is not None: |
60 |
auth = AuthenticationConfig() |
|
3777.1.24
by Aaron Bentley
Use .launchpad.net instead of bazaar. and bazaar.staging. |
61 |
auth_username = _get_auth_user(auth) |
3777.1.18
by Aaron Bentley
Fix None handling wrt auth upgrades |
62 |
# Auto-upgrading
|
3777.1.24
by Aaron Bentley
Use .launchpad.net instead of bazaar. and bazaar.staging. |
63 |
if auth_username is None: |
6150.3.4
by Jonathan Riddell
more launchpad plugin gettext()ing |
64 |
trace.note(gettext('Setting ssh/sftp usernames for launchpad.net.')) |
3777.1.18
by Aaron Bentley
Fix None handling wrt auth upgrades |
65 |
_set_auth_user(username, auth) |
3777.1.24
by Aaron Bentley
Use .launchpad.net instead of bazaar. and bazaar.staging. |
66 |
elif auth_username != username: |
67 |
raise MismatchedUsernames() |
|
3777.1.12
by Aaron Bentley
Enable configuring ssh auth from launchpad-login, with auto-upgrade. |
68 |
return username |
69 |
||
70 |
||
71 |
def _set_global_option(username, _config=None): |
|
72 |
if _config is None: |
|
6464.1.1
by Jelmer Vernooij
Merge launchpad configuration to config stacks. |
73 |
_config = GlobalStack() |
6852.1.1
by Jelmer Vernooij
Add lp-logout command. |
74 |
if username is None: |
75 |
_config.remove('launchpad_username') |
|
76 |
else: |
|
77 |
_config.set('launchpad_username', username) |
|
2898.3.6
by James Henstridge
Rename function arguments that are only intended for use by the tests. |
78 |
|
79 |
||
80 |
def set_lp_login(username, _config=None): |
|
2898.3.1
by James Henstridge
API for storing and retrieving the Launchpad user ID that will be used |
81 |
"""Set the user's Launchpad username""" |
3777.1.12
by Aaron Bentley
Enable configuring ssh auth from launchpad-login, with auto-upgrade. |
82 |
_set_global_option(username, _config) |
6852.1.1
by Jelmer Vernooij
Add lp-logout command. |
83 |
if username is not None: |
84 |
_set_auth_user(username) |
|
3777.1.12
by Aaron Bentley
Enable configuring ssh auth from launchpad-login, with auto-upgrade. |
85 |
|
86 |
||
87 |
def _get_auth_user(auth=None): |
|
88 |
if auth is None: |
|
89 |
auth = AuthenticationConfig() |
|
4304.2.1
by Vincent Ladeuil
Fix bug #367726 by reverting some default user handling introduced |
90 |
username = auth.get_user('ssh', '.launchpad.net') |
4222.3.15
by Jelmer Vernooij
Fix launchpad username prompting in a similar way to smtp. |
91 |
return username |
3777.1.12
by Aaron Bentley
Enable configuring ssh auth from launchpad-login, with auto-upgrade. |
92 |
|
7143.15.2
by Jelmer Vernooij
Run autopep8. |
93 |
|
3777.1.12
by Aaron Bentley
Enable configuring ssh auth from launchpad-login, with auto-upgrade. |
94 |
def _set_auth_user(username, auth=None): |
95 |
if auth is None: |
|
96 |
auth = AuthenticationConfig() |
|
97 |
auth.set_credentials( |
|
3777.1.24
by Aaron Bentley
Use .launchpad.net instead of bazaar. and bazaar.staging. |
98 |
'Launchpad', '.launchpad.net', username, 'ssh') |
2898.3.6
by James Henstridge
Rename function arguments that are only intended for use by the tests. |
99 |
|
100 |
||
101 |
def check_lp_login(username, _transport=None): |
|
2898.3.1
by James Henstridge
API for storing and retrieving the Launchpad user ID that will be used |
102 |
"""Check whether the given Launchpad username is okay. |
103 |
||
2934.1.1
by Ian Clatworthy
(James Henstridge) add a command for managing the Launchpad user ID |
104 |
This will check for both existence and whether the user has
|
2898.3.1
by James Henstridge
API for storing and retrieving the Launchpad user ID that will be used |
105 |
uploaded SSH keys.
|
106 |
"""
|
|
2898.3.6
by James Henstridge
Rename function arguments that are only intended for use by the tests. |
107 |
if _transport is None: |
6039.1.5
by Jelmer Vernooij
Add get_transport_from_url and get_transport_from_path functions. |
108 |
_transport = transport.get_transport_from_url(LAUNCHPAD_BASE) |
2898.3.1
by James Henstridge
API for storing and retrieving the Launchpad user ID that will be used |
109 |
|
110 |
try: |
|
2898.3.6
by James Henstridge
Rename function arguments that are only intended for use by the tests. |
111 |
data = _transport.get_bytes('~%s/+sshkeys' % username) |
2898.3.1
by James Henstridge
API for storing and retrieving the Launchpad user ID that will be used |
112 |
except errors.NoSuchFile: |
113 |
raise UnknownLaunchpadUsername(user=username) |
|
114 |
||
115 |
if not data: |
|
116 |
raise NoRegisteredSSHKeys(user=username) |