← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/transport/http.py

Merged John Meinel's integration

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/transport/http.py
22
22
import urlparse
23
23
from warnings import warn
24
24
 
 
25
import bzrlib
25
26
from bzrlib.transport import Transport, Server
26
27
from bzrlib.errors import (TransportNotPossible, NoSuchFile, 
27
28
                           TransportError, ConnectionError)
28
29
from bzrlib.errors import BzrError, BzrCheckError
29
30
from bzrlib.branch import Branch
30
31
from bzrlib.trace import mutter
 
32
from bzrlib.ui import ui_factory
31
33
 
32
34
 
33
35
def extract_auth(url, password_manager):
36
38
    password manager.  Return the url, minus those auth parameters (which
37
39
    confuse urllib2).
38
40
    """
39
 
    assert url.startswith('http://') or url.startswith('https://')
40
 
    scheme, host = url.split('//', 1)
41
 
    if '/' in host:
42
 
        host, path = host.split('/', 1)
43
 
        path = '/' + path
44
 
    else:
45
 
        path = ''
46
 
    port = ''
47
 
    if '@' in host:
48
 
        auth, host = host.split('@', 1)
 
41
    scheme, netloc, path, query, fragment = urlparse.urlsplit(url)
 
42
    assert (scheme == 'http') or (scheme == 'https')
 
43
    
 
44
    if '@' in netloc:
 
45
        auth, netloc = netloc.split('@', 1)
49
46
        if ':' in auth:
50
47
            username, password = auth.split(':', 1)
51
48
        else:
52
49
            username, password = auth, None
53
 
        if ':' in host:
54
 
            host, port = host.split(':', 1)
55
 
            port = ':' + port
56
 
        # FIXME: if password isn't given, should we ask for it?
 
50
        if ':' in netloc:
 
51
            host = netloc.split(':', 1)[0]
 
52
        else:
 
53
            host = netloc
 
54
        username = urllib.unquote(username)
57
55
        if password is not None:
58
 
            username = urllib.unquote(username)
59
56
            password = urllib.unquote(password)
60
 
            password_manager.add_password(None, host, username, password)
61
 
    url = scheme + '//' + host + port + path
 
57
        else:
 
58
            password = ui_factory.get_password(prompt='HTTP %(user)@%(host) password',
 
59
                                               user=username, host=host)
 
60
        password_manager.add_password(None, host, username, password)
 
61
    url = urlparse.urlunsplit((scheme, netloc, path, query, fragment))
62
62
    return url
 
63
 
63
64
    
64
65
def get_url(url):
65
66
    import urllib2
66
 
    mutter("get_url %s" % url)
 
67
    mutter("get_url %s", url)
67
68
    manager = urllib2.HTTPPasswordMgrWithDefaultRealm()
68
69
    url = extract_auth(url, manager)
69
70
    auth_handler = urllib2.HTTPBasicAuthHandler(manager)
70
71
    opener = urllib2.build_opener(auth_handler)
71
 
    url_f = opener.open(url)
72
 
    return url_f
 
72
 
 
73
    request = urllib2.Request(url)
 
74
    request.add_header('User-Agent', 'bzr/%s' % bzrlib.__version__)
 
75
    response = opener.open(request)
 
76
    return response
73
77
 
74
78
class HttpTransport(Transport):
75
79
    """This is the transport agent for http:// access.