← 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: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/plugins/launchpad/lp_registration.py
1
 
# Copyright (C) 2006 Canonical Ltd
 
1
# Copyright (C) 2006-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
 
from getpass import getpass
19
18
import os
 
19
import socket
20
20
from urlparse import urlsplit, urlunsplit
21
21
import urllib
22
22
import xmlrpclib
23
23
 
24
 
from bzrlib.lazy_import import lazy_import
25
 
lazy_import(globals(), """
26
 
from bzrlib import urlutils
27
 
""")
28
 
 
29
24
from bzrlib import (
30
25
    config,
31
26
    errors,
 
27
    urlutils,
32
28
    __version__ as _bzrlib_version,
33
29
    )
 
30
from bzrlib.transport.http import _urllib2_wrappers
 
31
 
34
32
 
35
33
# for testing, do
36
34
'''
53
51
        errors.BzrError.__init__(self, url=url)
54
52
 
55
53
 
 
54
class XMLRPCTransport(xmlrpclib.Transport):
 
55
 
 
56
    def __init__(self, scheme):
 
57
        # In python2.4 xmlrpclib.Transport is a old-style class, and does not
 
58
        # define __init__, so we check first
 
59
        init = getattr(xmlrpclib.Transport, '__init__', None)
 
60
        if init is not None:
 
61
            init(self)
 
62
        self._scheme = scheme
 
63
        self._opener = _urllib2_wrappers.Opener()
 
64
        self.verbose = 0
 
65
 
 
66
    def request(self, host, handler, request_body, verbose=0):
 
67
        self.verbose = verbose
 
68
        url = self._scheme + "://" + host + handler
 
69
        request = _urllib2_wrappers.Request("POST", url, request_body)
 
70
        # FIXME: _urllib2_wrappers will override user-agent with its own
 
71
        # request.add_header("User-Agent", self.user_agent)
 
72
        request.add_header("Content-Type", "text/xml")
 
73
 
 
74
        response = self._opener.open(request)
 
75
        if response.code != 200:
 
76
            raise xmlrpclib.ProtocolError(host + handler, response.code,
 
77
                                          response.msg, response.info())
 
78
        return self.parse_response(response)
 
79
 
 
80
 
56
81
class LaunchpadService(object):
57
82
    """A service to talk to Launchpad via XMLRPC.
58
83
 
59
 
    See http://bazaar-vcs.org/Specs/LaunchpadRpc for the methods we can call.
 
84
    See http://wiki.bazaar.canonical.com/Specs/LaunchpadRpc for the methods we can call.
60
85
    """
61
86
 
62
87
    LAUNCHPAD_DOMAINS = {
73
98
    for instance, domain in LAUNCHPAD_DOMAINS.iteritems():
74
99
        LAUNCHPAD_INSTANCE[instance] = 'https://xmlrpc.%s/bazaar/' % domain
75
100
 
76
 
    # We use edge as the default because:
77
 
    # Beta users get redirected to it
78
 
    # All users can use it
79
 
    # There is a bug in the launchpad side where redirection causes an OOPS.
80
 
    DEFAULT_INSTANCE = 'edge'
 
101
    # Previously 'edge' was used to avoid a launchpad bug with redirection
 
102
    DEFAULT_INSTANCE = 'production'
81
103
    DEFAULT_SERVICE_URL = LAUNCHPAD_INSTANCE[DEFAULT_INSTANCE]
82
104
 
83
105
    transport = None
90
112
        self._lp_instance = lp_instance
91
113
        if transport is None:
92
114
            uri_type = urllib.splittype(self.service_url)[0]
93
 
            if uri_type == 'https':
94
 
                transport = xmlrpclib.SafeTransport()
95
 
            else:
96
 
                transport = xmlrpclib.Transport()
 
115
            transport = XMLRPCTransport(uri_type)
97
116
            transport.user_agent = 'bzr/%s (xmlrpclib/%s)' \
98
117
                    % (_bzrlib_version, xmlrpclib.__version__)
99
118
        self.transport = transport
100
119
 
101
 
 
102
120
    @property
103
121
    def service_url(self):
104
122
        """Return the http or https url for the xmlrpc server.
116
134
        else:
117
135
            return self.DEFAULT_SERVICE_URL
118
136
 
 
137
    @classmethod
 
138
    def for_url(cls, url, **kwargs):
 
139
        """Return the Launchpad service corresponding to the given URL."""
 
140
        result = urlsplit(url)
 
141
        lp_instance = result[1]
 
142
        if lp_instance == '':
 
143
            lp_instance = None
 
144
        elif lp_instance not in cls.LAUNCHPAD_INSTANCE:
 
145
            raise errors.InvalidURL(path=url)
 
146
        return cls(lp_instance=lp_instance, **kwargs)
 
147
 
119
148
    def get_proxy(self, authenticated):
120
149
        """Return the proxy for XMLRPC requests."""
121
150
        if authenticated:
177
206
                # TODO: print more headers to help in tracking down failures
178
207
                raise errors.BzrError("xmlrpc protocol error connecting to %s: %s %s"
179
208
                        % (self.service_url, e.errcode, e.errmsg))
 
209
        except socket.gaierror, e:
 
210
            raise errors.ConnectionError(
 
211
                "Could not resolve '%s'" % self.domain,
 
212
                orig_error=e)
180
213
        return result
181
214
 
182
215
    @property
187
220
            instance = self._lp_instance
188
221
        return self.LAUNCHPAD_DOMAINS[instance]
189
222
 
190
 
    def get_web_url_from_branch_url(self, branch_url, _request_factory=None):
191
 
        """Get the Launchpad web URL for the given branch URL.
192
 
 
193
 
        :raise errors.InvalidURL: if 'branch_url' cannot be identified as a
194
 
            Launchpad branch URL.
195
 
        :return: The URL of the branch on Launchpad.
196
 
        """
 
223
    def _guess_branch_path(self, branch_url, _request_factory=None):
197
224
        scheme, hostinfo, path = urlsplit(branch_url)[:3]
198
225
        if _request_factory is None:
199
226
            _request_factory = ResolveLaunchpadPathRequest
211
238
                for domain in self.LAUNCHPAD_DOMAINS.itervalues())
212
239
            if hostinfo not in domains:
213
240
                raise NotLaunchpadBranch(branch_url)
 
241
        return path.lstrip('/')
 
242
 
 
243
    def get_web_url_from_branch_url(self, branch_url, _request_factory=None):
 
244
        """Get the Launchpad web URL for the given branch URL.
 
245
 
 
246
        :raise errors.InvalidURL: if 'branch_url' cannot be identified as a
 
247
            Launchpad branch URL.
 
248
        :return: The URL of the branch on Launchpad.
 
249
        """
 
250
        path = self._guess_branch_path(branch_url, _request_factory)
214
251
        return urlutils.join('https://code.%s' % self.domain, path)
215
252
 
216
253
 
314
351
    def __init__(self, path):
315
352
        if not path:
316
353
            raise errors.InvalidURL(path=path,
317
 
                                    extra="You must specify a product.")
 
354
                                    extra="You must specify a project.")
318
355
        self.path = path
319
356
 
320
357
    def _request_params(self):