← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

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

(vila) Rename assertWarns in test_config to avoid clashing with unittest2
 assertWarns. (Vincent Ladeuil)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/plugins/launchpad/lp_api.py
1
 
# Copyright (C) 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2009-2012 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
23
23
# needed by a command that uses it.
24
24
 
25
25
 
 
26
import httplib2
26
27
import os
27
28
import re
28
29
import urlparse
49
50
    STAGING_SERVICE_ROOT,
50
51
    Launchpad,
51
52
    )
52
 
 
 
53
from launchpadlib import uris
53
54
 
54
55
# Declare the minimum version of launchpadlib that we need in order to work.
55
 
# 1.5.1 is the version of launchpadlib packaged in Ubuntu 9.10, the most
56
 
# recent Ubuntu release at the time of writing.
57
 
MINIMUM_LAUNCHPADLIB_VERSION = (1, 5, 1)
 
56
# 1.6.0 is the version of launchpadlib packaged in Ubuntu 10.04, the most
 
57
# recent Ubuntu LTS release supported on the desktop at the time of writing.
 
58
MINIMUM_LAUNCHPADLIB_VERSION = (1, 6, 0)
58
59
 
59
60
 
60
61
def get_cache_directory():
76
77
            installed_version, installed_version)
77
78
 
78
79
 
79
 
# The older versions of launchpadlib only provided service root constants for
80
 
# edge and staging, whilst newer versions drop edge. Therefore service root
81
 
# URIs for which we do not always have constants are derived from the staging
82
 
# one, which does always exist.
83
 
#
84
 
# It is necessary to derive, rather than use hardcoded URIs because
85
 
# launchpadlib <= 1.5.4 requires service root URIs that end in a path of
86
 
# /beta/, whilst launchpadlib >= 1.5.5 requires service root URIs with no path
87
 
# info.
88
 
#
89
 
# Once we have a hard dependency on launchpadlib >= 1.5.4 we can replace all of
90
 
# bzr's local knowledge of individual Launchpad instances with use of the
91
 
# launchpadlib.uris module.
92
 
LAUNCHPAD_API_URLS = {
93
 
    'production': STAGING_SERVICE_ROOT.replace('api.staging.launchpad.net',
94
 
        'api.launchpad.net'),
95
 
    'qastaging': STAGING_SERVICE_ROOT.replace('api.staging.launchpad.net',
96
 
        'api.qastaging.launchpad.net'),
97
 
    'staging': STAGING_SERVICE_ROOT,
98
 
    'dev': STAGING_SERVICE_ROOT.replace('api.staging.launchpad.net',
99
 
        'api.launchpad.dev'),
100
 
    }
 
80
def lookup_service_root(service_root):
 
81
    try:
 
82
        return uris.lookup_service_root(service_root)
 
83
    except ValueError:
 
84
        if service_root != 'qastaging':
 
85
            raise
 
86
        staging_root = uris.lookup_service_root('staging')
 
87
        return staging_root.replace('staging', 'qastaging')
101
88
 
102
89
 
103
90
def _get_api_url(service):
114
101
    else:
115
102
        lp_instance = service._lp_instance
116
103
    try:
117
 
        return LAUNCHPAD_API_URLS[lp_instance]
118
 
    except KeyError:
 
104
        return lookup_service_root(lp_instance)
 
105
    except ValueError:
119
106
        raise InvalidLaunchpadInstance(lp_instance)
120
107
 
121
108
 
126
113
        errors.BzrError.__init__(self, branch=branch, url=branch.base)
127
114
 
128
115
 
129
 
def login(service, timeout=None, proxy_info=None):
 
116
def login(service, timeout=None, proxy_info=None,
 
117
          version=Launchpad.DEFAULT_VERSION):
130
118
    """Log in to the Launchpad API.
131
119
 
132
120
    :return: The root `Launchpad` object from launchpadlib.
133
121
    """
 
122
    if proxy_info is None:
 
123
        proxy_info = httplib2.proxy_info_from_environment('https')
134
124
    cache_directory = get_cache_directory()
135
125
    launchpad = Launchpad.login_with(
136
126
        'bzr', _get_api_url(service), cache_directory, timeout=timeout,
137
 
        proxy_info=proxy_info)
138
 
    # XXX: Work-around a minor security bug in launchpadlib 1.5.1, which would
139
 
    # create this directory with default umask.
 
127
        proxy_info=proxy_info, version=version)
 
128
    # XXX: Work-around a minor security bug in launchpadlib < 1.6.3, which
 
129
    # would create this directory with default umask.
140
130
    osutils.chmod_if_possible(cache_directory, 0700)
141
131
    return launchpad
142
132
 
210
200
        if str(launchpad._root_uri) == STAGING_SERVICE_ROOT:
211
201
            return url.replace('bazaar.launchpad.net',
212
202
                               'bazaar.staging.launchpad.net')
213
 
        elif str(launchpad._root_uri) == LAUNCHPAD_API_URLS['qastaging']:
 
203
        elif str(launchpad._root_uri) == lookup_service_root('qastaging'):
214
204
            return url.replace('bazaar.launchpad.net',
215
205
                               'bazaar.qastaging.launchpad.net')
216
206
        return url