X-Git-Url: https://jfr.im/git/z_archive/twitter.git/blobdiff_plain/94803fc949c8a894e7fd04a507dbd795d2632640..eeec9b0087ff5219ab5d3d15fe7241ded89de4c0:/twitter/api.py diff --git a/twitter/api.py b/twitter/api.py index 30ccb76..8c4379d 100644 --- a/twitter/api.py +++ b/twitter/api.py @@ -5,9 +5,17 @@ except ImportError: import urllib2 as urllib_request import urllib2 as urllib_error +try: + from cStringIO import StringIO +except ImportError: + from io import BytesIO as StringIO + from twitter.twitter_globals import POST_ACTIONS from twitter.auth import NoAuth +import re +import gzip + try: import json except ImportError: @@ -33,7 +41,12 @@ class TwitterHTTPError(TwitterError): self.uri = uri self.format = format self.uriparts = uriparts - self.response_data = self.e.fp.read() + if self.e.headers['Content-Encoding'] == 'gzip': + buf = StringIO(self.e.fp.read()) + f = gzip.GzipFile(fileobj=buf) + self.response_data = f.read() + else: + self.response_data = self.e.fp.read() def __str__(self): fmt = ("." + self.format) if self.format else "" @@ -51,7 +64,7 @@ class TwitterResponse(object): `headers` gives you access to the response headers as an httplib.HTTPHeaders instance. You can do - `response.headers.getheader('h')` to retrieve a header. + `response.headers.get('h')` to retrieve a header. """ def __init__(self, headers): self.headers = headers @@ -61,14 +74,21 @@ class TwitterResponse(object): """ Remaining requests in the current rate-limit. """ - return int(self.headers.getheader('X-RateLimit-Remaining')) + return int(self.headers.get('X-Rate-Limit-Remaining', "0")) + + @property + def rate_limit_limit(self): + """ + Remaining requests in the current rate-limit. + """ + return int(self.headers.get('X-Rate-Limit-Limit', "0")) @property def rate_limit_reset(self): """ Time in UTC epoch seconds when the rate limit will reset. """ - return int(self.headers.getheader('X-RateLimit-Reset')) + return int(self.headers.get('X-Rate-Limit-Reset', "0")) def wrap_response(response, headers): @@ -132,7 +152,7 @@ class TwitterCall(object): if not method: method = "GET" for action in POST_ACTIONS: - if uri.endswith(action): + if re.search("%s(/\d+)?$" % action, uri): method = "POST" break @@ -142,6 +162,12 @@ class TwitterCall(object): if id: uri += "/%s" %(id) + # If an _id kwarg is present, this is treated as id as a CGI + # param. + _id = kwargs.pop('_id', None) + if _id: + kwargs['id'] = _id + secure_str = '' if self.secure: secure_str = 's' @@ -151,7 +177,7 @@ class TwitterCall(object): uriBase = "http%s://%s/%s%s%s" %( secure_str, self.domain, uri, dot, self.format) - headers = {} + headers = {'Accept-Encoding': 'gzip'} if self.auth: headers.update(self.auth.generate_headers()) arg_data = self.auth.encode_params(uriBase, method, kwargs) @@ -160,19 +186,28 @@ class TwitterCall(object): body = None else: body = arg_data.encode('utf8') - req = urllib_request.Request(uriBase, body, headers) return self._handle_response(req, uri, arg_data) def _handle_response(self, req, uri, arg_data): try: handle = urllib_request.urlopen(req) + if handle.headers['Content-Type'] in ['image/jpeg', 'image/png']: + return handle + elif handle.info().get('Content-Encoding') == 'gzip': + # Handle gzip decompression + buf = StringIO(handle.read()) + f = gzip.GzipFile(fileobj=buf) + data = f.read() + else: + data = handle.read() + if "json" == self.format: - res = json.loads(handle.read().decode('utf8')) + res = json.loads(data.decode('utf8')) return wrap_response(res, handle.headers) else: return wrap_response( - handle.read().decode('utf8'), handle.headers) + data.decode('utf8'), handle.headers) except urllib_error.HTTPError as e: if (e.code == 304): return [] @@ -186,43 +221,51 @@ class Twitter(TwitterCall): Get RESTful data by accessing members of this class. The result is decoded python objects (lists and dicts). - The Twitter API is documented here: + The Twitter API is documented at: http://dev.twitter.com/doc Examples:: - twitter = Twitter( - auth=OAuth(token, token_key, con_secret, con_secret_key))) + t = Twitter( + auth=OAuth(token, token_key, con_secret, con_secret_key))) + + # Get the public timeline + t.statuses.public_timeline() + + # Get a particular friend's timeline + t.statuses.friends_timeline(id="billybob") - # Get the public timeline - twitter.statuses.public_timeline() + # Also supported (but totally weird) + t.statuses.friends_timeline.billybob() - # Get a particular friend's timeline - twitter.statuses.friends_timeline(id="billybob") + # Update your status + t.statuses.update( + status="Using @sixohsix's sweet Python Twitter Tools.") - # Also supported (but totally weird) - twitter.statuses.friends_timeline.billybob() + # Send a direct message + t.direct_messages.new( + user="billybob", + text="I think yer swell!") - # Send a direct message - twitter.direct_messages.new( - user="billybob", - text="I think yer swell!") + # Get the members of tamtar's list "Things That Are Rad" + t._("tamtar")._("things-that-are-rad").members() - # Get the members of a particular list of a particular friend - twitter.user.listname.members(user="billybob", listname="billysbuds") + # Note how the magic `_` method can be used to insert data + # into the middle of a call. You can also use replacement: + t.user.list.members(user="tamtar", list="things-that-are-rad") Searching Twitter:: - twitter_search = Twitter(domain="search.twitter.com") + twitter_search = Twitter(domain="search.twitter.com") - # Find the latest search trends - twitter_search.trends() + # Find the latest search trends + twitter_search.trends() - # Search for the latest News on #gaza - twitter_search.search(q="#gaza") + # Search for the latest News on #gaza + twitter_search.search(q="#gaza") Using the data returned @@ -231,13 +274,13 @@ class Twitter(TwitterCall): Twitter API calls return decoded JSON. This is converted into a bunch of Python lists, dicts, ints, and strings. For example:: - x = twitter.statuses.public_timeline() + x = twitter.statuses.public_timeline() - # The first 'tweet' in the timeline - x[0] + # The first 'tweet' in the timeline + x[0] - # The screen name of the user who wrote the first 'tweet' - x[0]['user']['screen_name'] + # The screen name of the user who wrote the first 'tweet' + x[0]['user']['screen_name'] Getting raw XML data @@ -246,10 +289,10 @@ class Twitter(TwitterCall): If you prefer to get your Twitter data in XML format, pass format="xml" to the Twitter object when you instantiate it:: - twitter = Twitter(format="xml") + twitter = Twitter(format="xml") - The output will not be parsed in any way. It will be a raw string - of XML. + The output will not be parsed in any way. It will be a raw string + of XML. """ def __init__( @@ -285,7 +328,7 @@ class Twitter(TwitterCall): if api_version is _DEFAULT: if domain == 'api.twitter.com': - api_version = '1' + api_version = '1.1' else: api_version = None