← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/plugins/launchpad/lp_registration.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2008-04-07 07:52:50 UTC
  • mfrom: (3340.1.1 208418-1.4)
  • Revision ID: pqm@pqm.ubuntu.com-20080407075250-phs53xnslo8boaeo
Return the correct knit serialisation method in _StreamAccess.
        (Andrew Bennetts, Martin Pool, Robert Collins)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/plugins/launchpad/lp_registration.py
21
21
import urllib
22
22
import xmlrpclib
23
23
 
24
 
import bzrlib.config
25
 
import bzrlib.errors as errors
 
24
from bzrlib import (
 
25
    config,
 
26
    errors,
 
27
    __version__ as _bzrlib_version,
 
28
    )
26
29
 
27
30
# for testing, do
28
31
'''
29
32
export BZR_LP_XMLRPC_URL=http://xmlrpc.staging.launchpad.net/bazaar/
30
33
'''
31
34
 
 
35
class InvalidLaunchpadInstance(errors.BzrError):
 
36
 
 
37
    _fmt = "%(lp_instance)s is not a valid Launchpad instance."
 
38
 
 
39
    def __init__(self, lp_instance):
 
40
        errors.BzrError.__init__(self, lp_instance=lp_instance)
 
41
 
 
42
 
32
43
class LaunchpadService(object):
33
44
    """A service to talk to Launchpad via XMLRPC.
34
 
    
 
45
 
35
46
    See http://bazaar-vcs.org/Specs/LaunchpadRpc for the methods we can call.
36
47
    """
37
48
 
38
 
    # NB: this should always end in a slash to avoid xmlrpclib appending
 
49
    # NB: these should always end in a slash to avoid xmlrpclib appending
39
50
    # '/RPC2'
40
 
    DEFAULT_SERVICE_URL = 'https://xmlrpc.launchpad.net/bazaar/'
 
51
    # We use edge as the default because:
 
52
    # Beta users get redirected to it
 
53
    # All users can use it
 
54
    # There is a bug in the launchpad side where redirection causes an OOPS.
 
55
    LAUNCHPAD_INSTANCE = {
 
56
        'production': 'https://xmlrpc.launchpad.net/bazaar/',
 
57
        'edge': 'https://xmlrpc.edge.launchpad.net/bazaar/',
 
58
        'staging': 'https://xmlrpc.staging.launchpad.net/bazaar/',
 
59
        'demo': 'https://xmlrpc.demo.launchpad.net/bazaar/',
 
60
        'dev': 'http://xmlrpc.launchpad.dev/bazaar/',
 
61
        }
 
62
    DEFAULT_SERVICE_URL = LAUNCHPAD_INSTANCE['edge']
41
63
 
42
64
    transport = None
43
65
    registrant_email = None
44
66
    registrant_password = None
45
67
 
46
68
 
47
 
    def __init__(self, transport=None):
 
69
    def __init__(self, transport=None, lp_instance=None):
48
70
        """Construct a new service talking to the launchpad rpc server"""
 
71
        self._lp_instance = lp_instance
49
72
        if transport is None:
50
73
            uri_type = urllib.splittype(self.service_url)[0]
51
74
            if uri_type == 'https':
53
76
            else:
54
77
                transport = xmlrpclib.Transport()
55
78
            transport.user_agent = 'bzr/%s (xmlrpclib/%s)' \
56
 
                    % (bzrlib.__version__, xmlrpclib.__version__)
 
79
                    % (_bzrlib_version, xmlrpclib.__version__)
57
80
        self.transport = transport
58
81
 
59
82
 
66
89
        key = 'BZR_LP_XMLRPC_URL'
67
90
        if key in os.environ:
68
91
            return os.environ[key]
 
92
        elif self._lp_instance is not None:
 
93
            try:
 
94
                return self.LAUNCHPAD_INSTANCE[self._lp_instance]
 
95
            except KeyError:
 
96
                raise InvalidLaunchpadInstance(self._lp_instance)
69
97
        else:
70
98
            return self.DEFAULT_SERVICE_URL
71
99
 
72
 
    def get_proxy(self):
 
100
    def get_proxy(self, authenticated):
73
101
        """Return the proxy for XMLRPC requests."""
74
 
        # auth info must be in url
75
 
        # TODO: if there's no registrant email perhaps we should just connect
76
 
        # anonymously?
77
 
        scheme, hostinfo, path = urlsplit(self.service_url)[:3]
78
 
        assert '@' not in hostinfo
79
 
        assert self.registrant_email is not None
80
 
        assert self.registrant_password is not None
81
 
        # TODO: perhaps fully quote the password to make it very slightly
82
 
        # obscured
83
 
        # TODO: can we perhaps add extra Authorization headers directly to the 
84
 
        # request, rather than putting this into the url?  perhaps a bit more 
85
 
        # secure against accidentally revealing it.  std66 s3.2.1 discourages putting
86
 
        # the password in the url.
87
 
        hostinfo = '%s:%s@%s' % (urllib.quote(self.registrant_email),
88
 
                                 urllib.quote(self.registrant_password),
89
 
                                 hostinfo)
90
 
        url = urlunsplit((scheme, hostinfo, path, '', ''))
 
102
        if authenticated:
 
103
            # auth info must be in url
 
104
            # TODO: if there's no registrant email perhaps we should
 
105
            # just connect anonymously?
 
106
            scheme, hostinfo, path = urlsplit(self.service_url)[:3]
 
107
            assert '@' not in hostinfo
 
108
            assert self.registrant_email is not None
 
109
            assert self.registrant_password is not None
 
110
            # TODO: perhaps fully quote the password to make it very slightly
 
111
            # obscured
 
112
            # TODO: can we perhaps add extra Authorization headers
 
113
            # directly to the request, rather than putting this into
 
114
            # the url?  perhaps a bit more secure against accidentally
 
115
            # revealing it.  std66 s3.2.1 discourages putting the
 
116
            # password in the url.
 
117
            hostinfo = '%s:%s@%s' % (urllib.quote(self.registrant_email),
 
118
                                     urllib.quote(self.registrant_password),
 
119
                                     hostinfo)
 
120
            url = urlunsplit((scheme, hostinfo, path, '', ''))
 
121
        else:
 
122
            url = self.service_url
91
123
        return xmlrpclib.ServerProxy(url, transport=self.transport)
92
124
 
93
125
    def gather_user_credentials(self):
94
126
        """Get the password from the user."""
95
 
        config = bzrlib.config.GlobalConfig()
96
 
        self.registrant_email = config.user_email()
 
127
        the_config = config.GlobalConfig()
 
128
        self.registrant_email = the_config.user_email()
97
129
        if self.registrant_password is None:
 
130
            auth = config.AuthenticationConfig()
 
131
            scheme, hostinfo = urlsplit(self.service_url)[:2]
98
132
            prompt = 'launchpad.net password for %s: ' % \
99
133
                    self.registrant_email
100
 
            self.registrant_password = getpass(prompt)
 
134
            # We will reuse http[s] credentials if we can, prompt user
 
135
            # otherwise
 
136
            self.registrant_password = auth.get_password(scheme, hostinfo,
 
137
                                                         self.registrant_email,
 
138
                                                         prompt=prompt)
101
139
 
102
 
    def send_request(self, method_name, method_params):
103
 
        proxy = self.get_proxy()
 
140
    def send_request(self, method_name, method_params, authenticated):
 
141
        proxy = self.get_proxy(authenticated)
104
142
        assert method_name
105
143
        method = getattr(proxy, method_name)
106
144
        try:
126
164
 
127
165
    # Set this to the XMLRPC method name.
128
166
    _methodname = None
 
167
    _authenticated = True
129
168
 
130
169
    def _request_params(self):
131
170
        """Return the arguments to pass to the method"""
137
176
        :param service: LaunchpadService indicating where to send
138
177
            the request and the authentication credentials.
139
178
        """
140
 
        return service.send_request(self._methodname, self._request_params())
 
179
        return service.send_request(self._methodname, self._request_params(),
 
180
                                    self._authenticated)
141
181
 
142
182
 
143
183
class DryRunLaunchpadService(LaunchpadService):
146
186
    The dummy service does not need authentication.
147
187
    """
148
188
 
149
 
    def send_request(self, method_name, method_params):
 
189
    def send_request(self, method_name, method_params, authenticated):
150
190
        pass
151
191
 
152
192
    def gather_user_credentials(self):
165
205
                 author_email='',
166
206
                 product_name='',
167
207
                 ):
168
 
        assert branch_url
 
208
        if not branch_url:
 
209
            raise errors.InvalidURL(branch_url, "You need to specify a non-empty branch URL.")
169
210
        self.branch_url = branch_url
170
211
        if branch_name:
171
212
            self.branch_name = branch_name
208
249
        # This must match the parameter tuple expected by Launchpad for this
209
250
        # method
210
251
        return (self.branch_url, self.bug_id, '')
 
252
 
 
253
 
 
254
class ResolveLaunchpadPathRequest(BaseRequest):
 
255
    """Request to resolve the path component of an lp: URL."""
 
256
 
 
257
    _methodname = 'resolve_lp_path'
 
258
    _authenticated = False
 
259
 
 
260
    def __init__(self, path):
 
261
        if not path:
 
262
            raise errors.InvalidURL(path=path,
 
263
                                    extra="You must specify a product.")
 
264
        self.path = path
 
265
 
 
266
    def _request_params(self):
 
267
        """Return xmlrpc request parameters"""
 
268
        return (self.path,)