import random import re from time import strptime, strftime from urllib import quote from util import hook, http @hook.api_key('twitter') @hook.command def twitter(inp, api_key=None): ".twitter / //#/# -- " \ "get 's last/th tweet/get tweet /do /get th result" if not isinstance(api_key, dict) or any(key not in api_key for key in ('consumer', 'consumer_secret', 'access', 'access_secret')): return "error: api keys not set" getting_id = False doing_search = False index_specified = False if re.match(r'^\d+$', inp): getting_id = True request_url = "https://api.twitter.com/1.1/statuses/show.json?id=%s" % inp else: try: inp, index = re.split('\s+', inp, 1) index = int(index) index_specified = True except ValueError: index = 0 if index < 0: index = 0 if index >= 20: return 'error: only supports up to the 20th tweet' if re.match(r'^#', inp): doing_search = True request_url = "https://api.twitter.com/1.1/search/tweets.json?q=%s" % quote(inp) else: request_url = "https://api.twitter.com/1.1/statuses/user_timeline.json?screen_name=%s" % inp try: tweet = http.get_json(request_url, oauth=True, oauth_keys=api_key) except http.HTTPError, e: errors = {400: 'bad request (ratelimited?)', 401: 'unauthorized', 403: 'forbidden', 404: 'invalid user/id', 500: 'twitter is broken', 502: 'twitter is down ("getting upgraded")', 503: 'twitter is overloaded (lol, RoR)', 410: 'twitter shut off api v1.'} if e.code == 404: return 'error: invalid ' + ['username', 'tweet id'][getting_id] if e.code in errors: return 'error: ' + errors[e.code] return 'error: unknown %s' % e.code if doing_search: try: tweet = tweet["statuses"] if not index_specified: index = random.randint(0, len(tweet) - 1) except KeyError: return 'error: no results' if not getting_id: try: tweet = tweet[index] except IndexError: return 'error: not that many tweets found' if 'retweeted_status' in tweet: rt = tweet["retweeted_status"] rt_text = http.unescape(rt["text"]).replace('\n', ' ') text = "RT @%s %s" % (rt["user"]["screen_name"], rt_text) else: text = http.unescape(tweet["text"]).replace('\n', ' ') screen_name = tweet["user"]["screen_name"] time = tweet["created_at"] time = strftime('%Y-%m-%d %H:%M:%S', strptime(time, '%a %b %d %H:%M:%S +0000 %Y')) return "%s \x02%s\x02: %s" % (time, screen_name, text) @hook.api_key('twitter') @hook.regex(r'https?://twitter.com/(#!/)?([_0-9a-zA-Z]+)/status/(\d+)') def show_tweet(match, api_key=None): return twitter(match.group(3), api_key)