X-Git-Url: https://jfr.im/git/z_archive/twitter.git/blobdiff_plain/dd48ea532f2a0d71d3a97d779b4e5c06d1b0762e..7fe9aab661b649adc5023afdc648414b53b23319:/twitter/api.py diff --git a/twitter/api.py b/twitter/api.py index 141f077..daa3a2b 100644 --- a/twitter/api.py +++ b/twitter/api.py @@ -15,6 +15,7 @@ from twitter.auth import NoAuth import re import gzip +import httplib try: import json @@ -41,12 +42,18 @@ class TwitterHTTPError(TwitterError): self.uri = uri self.format = format self.uriparts = uriparts + try: + data = self.e.fp.read() + except httplib.IncompleteRead, e: + # can't read the error text + # let's try some of it + data = e.partial if self.e.headers['Content-Encoding'] == 'gzip': - buf = StringIO(self.e.fp.read()) + buf = StringIO(data) f = gzip.GzipFile(fileobj=buf) self.response_data = f.read() else: - self.response_data = self.e.fp.read() + self.response_data = data def __str__(self): fmt = ("." + self.format) if self.format else "" @@ -201,14 +208,17 @@ class TwitterCall(object): handle = urllib_request.urlopen(req, **kwargs) if handle.headers['Content-Type'] in ['image/jpeg', 'image/png']: return handle - elif handle.info().get('Content-Encoding') == 'gzip': + try: + data = handle.read() + except httplib.IncompleteRead, e: + # Even if we don't get all the bytes we should have there + # may be a complete response in e.partial + data = e.partial + if handle.info().get('Content-Encoding') == 'gzip': # Handle gzip decompression - buf = StringIO(handle.read()) + buf = StringIO(data) f = gzip.GzipFile(fileobj=buf) data = f.read() - else: - data = handle.read() - if "json" == self.format: res = json.loads(data.decode('utf8')) return wrap_response(res, handle.headers) @@ -238,8 +248,8 @@ class Twitter(TwitterCall): t = Twitter( auth=OAuth(token, token_key, con_secret, con_secret_key))) - # Get the public timeline - t.statuses.public_timeline() + # Get your "home" timeline + t.statuses.home_timeline() # Get a particular friend's timeline t.statuses.friends_timeline(id="billybob") @@ -274,13 +284,8 @@ class Twitter(TwitterCall): Searching Twitter:: - twitter_search = Twitter(domain="search.twitter.com") - - # Find the latest search trends - twitter_search.trends() - - # Search for the latest News on #gaza - twitter_search.search(q="#gaza") + # Search for the latest tweets about #pycon + t.search.tweets(q="#pycon") Using the data returned @@ -289,7 +294,7 @@ 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.home_timeline() # The first 'tweet' in the timeline x[0]