binder.py 9.46 KB
Newer Older
1
# Tweepy
2
3
# Copyright 2009-2010 Joshua Roesslein
# See LICENSE for details.
4

Josh Roesslein's avatar
Josh Roesslein committed
5
import urllib
6
import time
7
import re
8
9

import requests
10

Josh Roesslein's avatar
Josh Roesslein committed
11
from tweepy.error import TweepError
12
from tweepy.utils import convert_to_utf8_str
13
from tweepy.models import Model
14

15

16
17
re_path_template = re.compile('{\w+}')

Josh Roesslein's avatar
Josh Roesslein committed
18

Jordi Riera's avatar
Jordi Riera committed
19
def bind_api(**config):
20
21
22

    class APIMethod(object):

Jordi Riera's avatar
Jordi Riera committed
23
24
25
26
27
28
29
30
31
        api = config['api']
        path = config['path']
        payload_type = config.get('payload_type', None)
        payload_list = config.get('payload_list', False)
        allowed_param = config.get('allowed_param', [])
        method = config.get('method', 'GET')
        require_auth = config.get('require_auth', False)
        search_api = config.get('search_api', False)
        use_cache = config.get('use_cache', True)
Aaron Hill's avatar
Aaron Hill committed
32
        session = requests.Session()
33

34
        def __init__(self, args, kwargs):
Jordi Riera's avatar
Jordi Riera committed
35
            api = self.api
36
37
38
39
40
            # If authentication is required and no credentials
            # are provided, throw an error.
            if self.require_auth and not api.auth:
                raise TweepError('Authentication required!')

41
            self.post_data = kwargs.pop('post_data', None)
Omer Murat Yildirim's avatar
Omer Murat Yildirim committed
42
43
44
45
46
47
48
49
50
51
            self.retry_count = kwargs.pop('retry_count',
                                          api.retry_count)
            self.retry_delay = kwargs.pop('retry_delay',
                                          api.retry_delay)
            self.retry_errors = kwargs.pop('retry_errors',
                                           api.retry_errors)
            self.wait_on_rate_limit = kwargs.pop('wait_on_rate_limit',
                                                 api.wait_on_rate_limit)
            self.wait_on_rate_limit_notify = kwargs.pop('wait_on_rate_limit_notify',
                                                        api.wait_on_rate_limit_notify)
52
53
54
            self.parser = kwargs.pop('parser', api.parser)
            self.session.headers = kwargs.pop('headers', {})
            self.build_parameters(args, kwargs)
55
56
57
58
59
60
61
62
63

            # Pick correct URL root to use
            if self.search_api:
                self.api_root = api.search_root
            else:
                self.api_root = api.api_root

            # Perform any path variable substitution
            self.build_path()
64

65
66
            if self.search_api:
                self.host = api.search_host
67
            else:
68
69
                self.host = api.host

70
71
72
            # Manually set Host header to fix an issue in python 2.5
            # or older where Host is set including the 443 port.
            # This causes Twitter to issue 301 redirect.
Pablo Castellano's avatar
Pablo Castellano committed
73
            # See Issue https://github.com/tweepy/tweepy/issues/12
Aaron Hill's avatar
Aaron Hill committed
74
75

            self.session.headers['Host'] = self.host
76
77
78
            # Monitoring rate limits
            self._remaining_calls = None
            self._reset_time = None
79

80
        def build_parameters(self, args, kwargs):
Aaron Hill's avatar
Aaron Hill committed
81
            self.session.params = {}
82
            for idx, arg in enumerate(args):
83
84
                if arg is None:
                    continue
85
                try:
Aaron Hill's avatar
Aaron Hill committed
86
                    self.session.params[self.allowed_param[idx]] = convert_to_utf8_str(arg)
87
88
89
                except IndexError:
                    raise TweepError('Too many parameters supplied!')

90
            for k, arg in kwargs.items():
91
92
                if arg is None:
                    continue
Aaron Hill's avatar
Aaron Hill committed
93
                if k in self.session.params:
94
95
                    raise TweepError('Multiple values for parameter %s supplied!' % k)

Aaron Hill's avatar
Aaron Hill committed
96
                self.session.params[k] = convert_to_utf8_str(arg)
97
98
99
100
101

        def build_path(self):
            for variable in re_path_template.findall(self.path):
                name = variable.strip('{}')

Aaron Hill's avatar
Aaron Hill committed
102
                if name == 'user' and 'user' not in self.session.params and self.api.auth:
103
104
105
106
                    # No 'user' parameter provided, fetch it from Auth instead.
                    value = self.api.auth.get_username()
                else:
                    try:
Aaron Hill's avatar
Aaron Hill committed
107
                        value = urllib.quote(self.session.params[name])
108
109
                    except KeyError:
                        raise TweepError('No parameter value found for path variable: %s' % name)
Aaron Hill's avatar
Aaron Hill committed
110
                    del self.session.params[name]
111

112
113
                self.path = self.path.replace(variable, value)

114
        def execute(self):
Joshua Roesslein's avatar
Joshua Roesslein committed
115
116
            self.api.cached_result = False

117
118
            # Build the request URL
            url = self.api_root + self.path
Joshua Roesslein's avatar
Joshua Roesslein committed
119
            full_url = 'https://' + self.host + url
120
121
122

            # Query the cache if one is available
            # and this request uses a GET method.
123
            if self.use_cache and self.api.cache and self.method == 'GET':
124
                cache_result = self.api.cache.get(url)
125
126
127
128
129
                # if cache result found and not expired, return it
                if cache_result:
                    # must restore api reference
                    if isinstance(cache_result, list):
                        for result in cache_result:
130
131
                            if isinstance(result, Model):
                                result._api = self.api
132
                    else:
133
134
                        if isinstance(cache_result, Model):
                            cache_result._api = self.api
Joshua Roesslein's avatar
Joshua Roesslein committed
135
                    self.api.cached_result = True
136
137
138
139
140
141
                    return cache_result

            # Continue attempting request until successful
            # or maximum number of retries is reached.
            retries_performed = 0
            while retries_performed < self.retry_count + 1:
142
                # handle running out of api calls
Omer Murat Yildirim's avatar
Omer Murat Yildirim committed
143
144
145
146
147
148
149
150
151
                if self.wait_on_rate_limit:
                    if self._reset_time is not None:
                        if self._remaining_calls is not None:
                            if self._remaining_calls < 1:
                                sleep_time = self._reset_time - int(time.time())
                                if sleep_time > 0:
                                    if self.wait_on_rate_limit_notify:
                                        print "Rate limit reached. Sleeping for: " + str(sleep_time)
                                    time.sleep(sleep_time + 5)  # sleep for few extra sec
152

153
154
                # Apply authentication
                if self.api.auth:
Aaron Hill's avatar
Aaron Hill committed
155
                    auth = self.api.auth.apply_auth()
156

Mike's avatar
Mike committed
157
158
                # Request compression if configured
                if self.api.compression:
Aaron Hill's avatar
Aaron Hill committed
159
                    self.session.headers['Accept-encoding'] = 'gzip'
Mike's avatar
Mike committed
160

161
162
                # Execute request
                try:
Omer Murat Yildirim's avatar
Omer Murat Yildirim committed
163
164
165
166
167
168
                    resp = self.session.request(self.method,
                                                full_url,
                                                data=self.post_data,
                                                timeout=self.api.timeout,
                                                auth=auth,
                                                proxies=self.api.proxy)
Aaron Hill's avatar
Aaron Hill committed
169
                except Exception, e:
170
                    raise TweepError('Failed to send request: %s' % e)
171
                rem_calls = resp.headers.get('x-rate-limit-remaining')
172
                if rem_calls is not None:
173
                    self._remaining_calls = int(rem_calls)
174
175
                elif isinstance(self._remaining_calls, int):
                    self._remaining_calls -= 1
176
                reset_time = resp.headers.get('x-rate-limit-reset')
177
                if reset_time is not None:
178
                    self._reset_time = int(reset_time)
179
                if self.wait_on_rate_limit and self._remaining_calls == 0 and (
Omer Murat Yildirim's avatar
Omer Murat Yildirim committed
180
181
                        # if ran out of calls before waiting switching retry last call
                        resp.status == 429 or resp.status == 420):
182
                    continue
183
                retry_delay = self.retry_delay
184
                # Exit request loop if non-retry error code
Aaron Hill's avatar
Aaron Hill committed
185
                if resp.status_code == 200:
186
                    break
Aaron Hill's avatar
Aaron Hill committed
187
188
189
190
                elif (resp.status_code == 429 or resp.status_code == 420) and self.wait_on_rate_limit:
                    if 'retry-after' in resp.headers:
                        retry_delay = float(resp.headers['retry-after'])
                elif self.retry_errors and resp.status_code not in self.retry_errors:
191
                    break
192

193
                # Sleep before retrying request again
194
                time.sleep(retry_delay)
195
                retries_performed += 1
Josh Roesslein's avatar
Josh Roesslein committed
196

197
198
            # If an error was returned, throw an exception
            self.api.last_response = resp
199
            if resp.status_code and not 200 <= resp.status_code < 300:
200
                try:
Aaron Hill's avatar
Aaron Hill committed
201
                    error_msg = self.parser.parse_error(resp.text)
202
                except Exception:
Aaron Hill's avatar
Aaron Hill committed
203
                    error_msg = "Twitter error response: status code = %s" % resp.status_code
204
                raise TweepError(error_msg, resp)
205
206

            # Parse the response payload
207
            result = self.parser.parse(self, resp.text)
208
209

            # Store result into cache if one is available.
210
            if self.use_cache and self.api.cache and self.method == 'GET' and result:
211
                self.api.cache.store(url, result)
212

213
            return result
Josh Roesslein's avatar
Josh Roesslein committed
214

215
216
217
    def _call(*args, **kwargs):
        method = APIMethod(args, kwargs)
        if kwargs.get('create'):
218
219
220
            return method
        else:
            return method.execute()
Josh Roesslein's avatar
Josh Roesslein committed
221

222
    # Set pagination mode
223
    if 'cursor' in APIMethod.allowed_param:
224
        _call.pagination_mode = 'cursor'
Omer Murat Yildirim's avatar
Omer Murat Yildirim committed
225
226
227
    elif 'max_id' in APIMethod.allowed_param:
        if 'since_id' in APIMethod.allowed_param:
            _call.pagination_mode = 'id'
228
    elif 'page' in APIMethod.allowed_param:
229
        _call.pagination_mode = 'page'
230

Josh Roesslein's avatar
Josh Roesslein committed
231
    return _call