OLD | NEW |
1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 # Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 # Use of this source code is governed by a BSD-style license that can be | 2 # Use of this source code is governed by a BSD-style license that can be |
3 # found in the LICENSE file. | 3 # found in the LICENSE file. |
4 | 4 |
5 import base64 | 5 import base64 |
6 import posixpath | 6 import posixpath |
7 | 7 |
8 from appengine_wrappers import urlfetch | 8 from appengine_wrappers import GetAppVersion, urlfetch |
9 from future import Future | 9 from future import Future |
10 | 10 |
11 | 11 |
12 class _AsyncFetchDelegate(object): | 12 class _AsyncFetchDelegate(object): |
13 def __init__(self, rpc): | 13 def __init__(self, rpc): |
14 self._rpc = rpc | 14 self._rpc = rpc |
15 | 15 |
16 def Get(self): | 16 def Get(self): |
17 return self._rpc.get_result() | 17 return self._rpc.get_result() |
18 | 18 |
19 | 19 |
20 def _MakeHeaders(username, password): | 20 def _MakeHeaders(username, password): |
21 headers = { 'Cache-Control': 'max-age=0' } | 21 headers = { |
| 22 'User-Agent': 'Chromium docserver %s' % GetAppVersion(), |
| 23 'Cache-Control': 'max-age=0', |
| 24 } |
22 if username is not None and password is not None: | 25 if username is not None and password is not None: |
23 headers['Authorization'] = 'Basic %s' % base64.encodestring( | 26 headers['Authorization'] = 'Basic %s' % base64.b64encode( |
24 '%s:%s' % (username, password)) | 27 '%s:%s' % (username, password)) |
25 return headers | 28 return headers |
26 | 29 |
27 | 30 |
28 class AppEngineUrlFetcher(object): | 31 class AppEngineUrlFetcher(object): |
29 """A wrapper around the App Engine urlfetch module that allows for easy | 32 """A wrapper around the App Engine urlfetch module that allows for easy |
30 async fetches. | 33 async fetches. |
31 """ | 34 """ |
32 def __init__(self, base_path=None): | 35 def __init__(self, base_path=None): |
33 assert base_path is None or not base_path.endswith('/') | 36 assert base_path is None or not base_path.endswith('/') |
(...skipping 11 matching lines...) Expand all Loading... |
45 rpc = urlfetch.create_rpc() | 48 rpc = urlfetch.create_rpc() |
46 urlfetch.make_fetch_call(rpc, | 49 urlfetch.make_fetch_call(rpc, |
47 self._FromBasePath(url), | 50 self._FromBasePath(url), |
48 headers=_MakeHeaders(username, password)) | 51 headers=_MakeHeaders(username, password)) |
49 return Future(delegate=_AsyncFetchDelegate(rpc)) | 52 return Future(delegate=_AsyncFetchDelegate(rpc)) |
50 | 53 |
51 def _FromBasePath(self, url): | 54 def _FromBasePath(self, url): |
52 if self._base_path is not None: | 55 if self._base_path is not None: |
53 url = posixpath.join(self._base_path, url) if url else self._base_path | 56 url = posixpath.join(self._base_path, url) if url else self._base_path |
54 return url | 57 return url |
OLD | NEW |