mirror of
https://github.com/oddluck/limnoria-plugins.git
synced 2025-04-27 05:21:13 -05:00
547 lines
22 KiB
Python
547 lines
22 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) 2013, spline
|
|
###
|
|
|
|
# my libs
|
|
import urllib2
|
|
import json
|
|
import string
|
|
# libraries for time_created_at
|
|
import time
|
|
from datetime import tzinfo, datetime, timedelta
|
|
# for unescape
|
|
import re
|
|
import htmlentitydefs
|
|
# reencode
|
|
import unicodedata
|
|
# oauthtwitter
|
|
import oauth2 as oauth
|
|
# supybot libs
|
|
import supybot.utils as utils
|
|
from supybot.commands import *
|
|
import supybot.plugins as plugins
|
|
import supybot.ircutils as ircutils
|
|
import supybot.callbacks as callbacks
|
|
|
|
|
|
class OAuthApi:
|
|
""" OAuth class to work with Twitter v1.1 API."""
|
|
def __init__(self, consumer_key, consumer_secret, token, token_secret):
|
|
token = oauth.Token(token, token_secret)
|
|
self._Consumer = oauth.Consumer(consumer_key, consumer_secret)
|
|
self._signature_method = oauth.SignatureMethod_HMAC_SHA1()
|
|
self._access_token = token
|
|
|
|
def _FetchUrl(self,url, parameters=None):
|
|
extra_params = {}
|
|
if parameters:
|
|
extra_params.update(parameters)
|
|
|
|
req = self._makeOAuthRequest(url, params=extra_params)
|
|
opener = urllib2.build_opener(urllib2.HTTPHandler(debuglevel=1))
|
|
url = req.to_url()
|
|
url_data = opener.open(url)
|
|
opener.close()
|
|
return url_data
|
|
|
|
def _makeOAuthRequest(self, url, token=None, params=None):
|
|
oauth_base_params = {
|
|
'oauth_version': "1.0",
|
|
'oauth_nonce': oauth.generate_nonce(),
|
|
'oauth_timestamp': int(time.time())
|
|
}
|
|
|
|
if params:
|
|
params.update(oauth_base_params)
|
|
else:
|
|
params = oauth_base_params
|
|
|
|
if not token:
|
|
token = self._access_token
|
|
request = oauth.Request(method="GET", url=url, parameters=params)
|
|
request.sign_request(self._signature_method, self._Consumer, token)
|
|
return request
|
|
|
|
def ApiCall(self, call, parameters={}):
|
|
return_value = []
|
|
try:
|
|
data = self._FetchUrl("https://api.twitter.com/1.1/" + call + ".json", parameters)
|
|
except urllib2.HTTPError, e:
|
|
return e
|
|
except urllib2.URLError, e:
|
|
return e
|
|
else:
|
|
return data
|
|
|
|
|
|
class Tweety(callbacks.Plugin):
|
|
"""Public Twitter class for working with the API."""
|
|
threaded = True
|
|
|
|
def __init__(self, irc):
|
|
self.__parent = super(Tweety, self)
|
|
self.__parent.__init__(irc)
|
|
self.twitterApi = False
|
|
if not self.twitterApi:
|
|
self._checkAuthorization()
|
|
|
|
def _checkAuthorization(self):
|
|
""" Check if we have our keys and can auth."""
|
|
if not self.twitterApi:
|
|
failTest = False
|
|
for checkKey in ('consumerKey', 'consumerSecret', 'accessKey', 'accessSecret'):
|
|
try:
|
|
testKey = self.registryValue(checkKey)
|
|
except:
|
|
self.log.debug("Failed checking keys. We're missing the config value for: {0}. Please set this and try again.".format(checkKey))
|
|
failTest = True
|
|
break
|
|
|
|
if failTest:
|
|
self.log.error('Failed getting keys. You must set all 4 keys in config variables.')
|
|
return False
|
|
|
|
self.log.info("Got all 4 keys. Now trying to auth up with Twitter.")
|
|
twitterApi = OAuthApi(self.registryValue('consumerKey'), self.registryValue('consumerSecret'), self.registryValue('accessKey'), self.registryValue('accessSecret'))
|
|
data = twitterApi.ApiCall('account/verify_credentials')
|
|
|
|
if data.getcode() == "401":
|
|
self.log.error("ERROR: I could not log in using your credentials. Message: %s" % data.read())
|
|
return False
|
|
else:
|
|
self.log.info("I have successfully authorized and logged in to Twitter using your credentials.")
|
|
self.twitterApi = OAuthApi(self.registryValue('consumerKey'), self.registryValue('consumerSecret'), self.registryValue('accessKey'), self.registryValue('accessSecret'))
|
|
else:
|
|
pass
|
|
|
|
########################
|
|
# COLOR AND FORMATTING #
|
|
########################
|
|
|
|
def _red(self, string):
|
|
"""Returns a red string."""
|
|
return ircutils.mircColor(string, 'red')
|
|
|
|
def _yellow(self, string):
|
|
"""Returns a yellow string."""
|
|
return ircutils.mircColor(string, 'yellow')
|
|
|
|
def _green(self, string):
|
|
"""Returns a green string."""
|
|
return ircutils.mircColor(string, 'green')
|
|
|
|
def _teal(self, string):
|
|
"""Returns a teal string."""
|
|
return ircutils.mircColor(string, 'teal')
|
|
|
|
def _blue(self, string):
|
|
"""Returns a blue string."""
|
|
return ircutils.mircColor(string, 'blue')
|
|
|
|
def _orange(self, string):
|
|
"""Returns an orange string."""
|
|
return ircutils.mircColor(string, 'orange')
|
|
|
|
def _bold(self, string):
|
|
"""Returns a bold string."""
|
|
return ircutils.bold(string)
|
|
|
|
def _ul(self, string):
|
|
"""Returns an underline string."""
|
|
return ircutils.underline(string)
|
|
|
|
def _bu(self, string):
|
|
"""Returns a bold/underline string."""
|
|
return ircutils.bold(ircutils.underline(string))
|
|
|
|
def _unescape(self, text):
|
|
"""Created by Fredrik Lundh (http://effbot.org/zone/re-sub.htm#unescape-html)"""
|
|
text = text.replace("\n", " ")
|
|
def fixup(m):
|
|
text = m.group(0)
|
|
if text[:2] == "&#":
|
|
# character reference
|
|
try:
|
|
if text[:3] == "&#x":
|
|
return unichr(int(text[3:-1], 16))
|
|
else:
|
|
return unichr(int(text[2:-1]))
|
|
except (ValueError, OverflowError):
|
|
pass
|
|
else:
|
|
# named entity
|
|
try:
|
|
text = unichr(htmlentitydefs.name2codepoint[text[1:-1]])
|
|
except KeyError:
|
|
pass
|
|
return text # leave as is
|
|
return re.sub("&#?\w+;", fixup, text)
|
|
|
|
def _time_created_at(self, s):
|
|
"""
|
|
Return relative time delta. Ex: 3m ago.
|
|
"""
|
|
|
|
try: # timeline's created_at Tue May 08 10:58:49 +0000 2012
|
|
ddate = time.strptime(s, "%a %b %d %H:%M:%S +0000 %Y")[:-2]
|
|
except ValueError:
|
|
try: # search's created_at Thu, 06 Oct 2011 19:41:12 +0000
|
|
ddate = time.strptime(s, "%a, %d %b %Y %H:%M:%S +0000")[:-2]
|
|
except ValueError:
|
|
return s
|
|
|
|
# do the math
|
|
d = datetime.utcnow() - datetime(*ddate, tzinfo=None)
|
|
|
|
# now parse and return.
|
|
if d.days:
|
|
rel_time = "%sd ago" % d.days
|
|
elif d.seconds > 3600:
|
|
rel_time = "%sh ago" % (d.seconds / 3600)
|
|
elif 60 <= d.seconds < 3600:
|
|
rel_time = "%sm ago" % (d.seconds / 60)
|
|
else:
|
|
rel_time = "%ss ago" % (d.seconds)
|
|
return rel_time
|
|
|
|
def _outputTweet(self, irc, msg, nick, name, text, time, tweetid):
|
|
"""
|
|
Takes a group of strings and outputs a Tweet to IRC. Used for tsearch and twitter.
|
|
"""
|
|
|
|
outputColorTweets = self.registryValue('outputColorTweets', msg.args[0])
|
|
|
|
if outputColorTweets:
|
|
ret = ircutils.underline(ircutils.mircColor(("@" + nick), 'blue'))
|
|
else:
|
|
ret = ircutils.underline(ircutils.bold("@" + nick))
|
|
|
|
if not self.registryValue('hideRealName', msg.args[0]): # show realname in tweet output?
|
|
ret += " ({0})".format(name)
|
|
|
|
# add in the end with the text + tape
|
|
if outputColorTweets:
|
|
text = re.sub(r'(http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+)', ircutils.mircColor(r'\1', 'red'), text) # color urls.
|
|
ret += ": {0} ({1})".format(text, ircutils.mircColor(time, 'yellow'))
|
|
else:
|
|
ret += ": {0} ({1})".format(text, ircutils.bold(time))
|
|
|
|
if self.registryValue('addShortUrl', msg.args[0]):
|
|
if self._createShortUrl(nick, tweetid):
|
|
ret += " {0}".format(url)
|
|
|
|
irc.reply(ret)
|
|
|
|
def _createShortUrl(self, nick, tweetid):
|
|
"""
|
|
Takes a nick and tweetid and returns a shortened URL via is.gd service.
|
|
"""
|
|
|
|
longurl = "https://twitter.com/#!/{0}/status/{1}".format(nick, tweetid)
|
|
try:
|
|
req = urllib2.Request("http://is.gd/api.php?longurl=" + utils.web.urlquote(longurl))
|
|
f = urllib2.urlopen(req)
|
|
shorturl = f.read()
|
|
return shorturl
|
|
except:
|
|
return False
|
|
|
|
def _woeid_lookup(self, lookup):
|
|
"""<location>
|
|
Use Yahoo's API to look-up a WOEID.
|
|
"""
|
|
|
|
query = "select * from geo.places where text='%s'" % lookup
|
|
params = {"q": query,
|
|
"format":"json",
|
|
"diagnostics":"false",
|
|
"env":"store://datatables.org/alltableswithkeys"}
|
|
|
|
try:
|
|
response = urllib2.urlopen("http://query.yahooapis.com/v1/public/yql", utils.web.urlencode(params))
|
|
data = json.loads(response.read())
|
|
|
|
if data['query']['count'] > 1:
|
|
woeid = data['query']['results']['place'][0]['woeid']
|
|
else:
|
|
woeid = data['query']['results']['place']['woeid']
|
|
return woeid
|
|
except Exception, err:
|
|
self.log.error("Error looking up %s :: %s" % (lookup,err))
|
|
return None
|
|
|
|
##########################
|
|
### PUBLIC FUNCTIONS #####
|
|
##########################
|
|
|
|
def woeidlookup(self, irc, msg, args, lookup):
|
|
"""<location>
|
|
Search Yahoo's WOEID DB for a location. Useful for the trends variable.
|
|
"""
|
|
|
|
woeid = self._woeid_lookup(lookup)
|
|
if woeid:
|
|
irc.reply(("I found WOEID: %s while searching for: '%s'") % (self._bold(woeid), lookup))
|
|
else:
|
|
irc.reply(("Something broke while looking up: '%s'") % (lookup))
|
|
|
|
woeidlookup = wrap(woeidlookup, ['text'])
|
|
|
|
def ratelimits(self, irc, msg, args):
|
|
"""
|
|
Display current rate limits for your twitter API account.
|
|
"""
|
|
|
|
# before we do anything, make sure we have a twitterApi object.
|
|
if not self.twitterApi:
|
|
irc.reply("ERROR: Twitter is not authorized. Please check logs before running this command.")
|
|
return
|
|
|
|
data = self.twitterApi.ApiCall('application/rate_limit_status')
|
|
try:
|
|
data = json.loads(data.read())
|
|
except:
|
|
irc.reply("Failed to lookup ratelimit data: %s" % data)
|
|
return
|
|
|
|
data = data.get('resources', None)
|
|
if not data: # simple check if we have part of the json dict.
|
|
irc.reply("Failed to fetch application rate limit status. Something could be wrong with Twitter.")
|
|
self.log.error(data)
|
|
return
|
|
|
|
# we only have resources needed in here. def below works with each entry properly.
|
|
resourcelist = ['trends/place', 'search/tweets', 'users/show/:id', 'statuses/show/:id', 'statuses/user_timeline']
|
|
|
|
for resource in resourcelist:
|
|
family, endpoint = resource.split('/', 1) # need to split each entry on /, resource family is [0], append / to entry.
|
|
resourcedict = data.get(family, None)
|
|
endpoint = resourcedict.get("/"+resource, None)
|
|
minutes = "%sm%ss" % divmod(int(endpoint['reset'])-int(time.time()), 60)
|
|
output = "Reset in: {0} Remaining: {1}".format(minutes, endpoint['remaining'])
|
|
irc.reply("{0} :: {1}".format(self._bold(resource), output))
|
|
|
|
ratelimits = wrap(ratelimits)
|
|
|
|
def trends(self, irc, msg, args, getopts, optwoeid):
|
|
"""[--exclude] <location>
|
|
|
|
Returns the Top 10 Twitter trends for a specific location.
|
|
Use optional argument location for trends. Ex: trends Boston
|
|
Use --exclude to not include #hashtags in trends data.
|
|
"""
|
|
|
|
# before we do anything, make sure we have a twitterApi object.
|
|
if not self.twitterApi:
|
|
irc.reply("ERROR: Twitter is not authorized. Please check logs before running this command.")
|
|
return
|
|
|
|
args = {'id': self.registryValue('woeid', msg.args[0]),'exclude': self.registryValue('hideHashtagsTrends', msg.args[0])}
|
|
if getopts:
|
|
for (key, value) in getopts:
|
|
if key == 'exclude': # remove hashtags from trends.
|
|
args['exclude'] = 'hashtags'
|
|
|
|
# work with woeid. 1 is world, the default. can be set via input or via config.
|
|
if optwoeid:
|
|
woeid = self._woeid_lookup(optwoeid)
|
|
if woeid:
|
|
args['id'] = woeid
|
|
|
|
data = self.twitterApi.ApiCall('trends/place', parameters=args)
|
|
try:
|
|
data = json.loads(data.read())
|
|
except:
|
|
irc.reply("Error: failed to lookup Twitter trends: %s" % data)
|
|
return
|
|
|
|
try:
|
|
location = data[0]['locations'][0]['name']
|
|
except:
|
|
irc.reply("ERROR: Cannot load trends: {0}".format(data)) # error also throws 404.
|
|
return
|
|
|
|
# package together in object and output.
|
|
ttrends = string.join([trend['name'].encode('utf-8') for trend in data[0]['trends']], " | ")
|
|
irc.reply("Top 10 Twitter Trends in {0} :: {1}".format(self._bold(location), ttrends))
|
|
|
|
trends = wrap(trends, [getopts({'exclude':''}), optional('text')])
|
|
|
|
def tsearch(self, irc, msg, args, optlist, optterm):
|
|
"""[--num number] [--searchtype mixed,recent,popular] [--lang xx] <term>
|
|
|
|
Searches Twitter for the <term> and returns the most recent results.
|
|
Number is number of results. Must be a number higher than 0 and max 10.
|
|
searchtype being recent, popular or mixed. Popular is the default.
|
|
"""
|
|
|
|
# before we do anything, make sure we have a twitterApi object.
|
|
if not self.twitterApi:
|
|
irc.reply("ERROR: Twitter is not authorized. Please check logs before running this command.")
|
|
return
|
|
|
|
tsearchArgs = {'include_entities':'false', 'count': self.registryValue('defaultSearchResults', msg.args[0]), 'lang':'en', 'q':utils.web.urlquote(optterm)}
|
|
|
|
if optlist:
|
|
for (key, value) in optlist:
|
|
if key == 'num':
|
|
max = self.registryValue('maxSearchResults', msg.args[0])
|
|
if value > max or value <= 0:
|
|
irc.reply("Error: '{0}' is not a valid number of tweets. Range is above 0 and max {1}.".format(value, max))
|
|
return
|
|
else:
|
|
tsearchArgs['count'] = value
|
|
if key == 'searchtype':
|
|
tsearchArgs['result_type'] = value # limited by getopts to valid values.
|
|
if key == 'lang': # lang . Uses ISO-639 codes like 'en' http://en.wikipedia.org/wiki/ISO_639-1
|
|
tsearchArgs['lang'] = value
|
|
|
|
data = self.twitterApi.ApiCall('search/tweets', parameters=tsearchArgs)
|
|
try:
|
|
data = json.loads(data.read())
|
|
except:
|
|
irc.reply("Error: %s trying to search Twitter." % data)
|
|
return
|
|
|
|
results = data.get('statuses', None) # data returned as a dict
|
|
|
|
if not results or len(results) == 0:
|
|
irc.reply("ERROR: No Twitter Search results found for '{0}'".format(optterm))
|
|
return
|
|
else:
|
|
for result in results:
|
|
nick = result['user'].get('screen_name', None)
|
|
name = result["user"].get('name', None)
|
|
text = self._unescape(result.get('text', None)) # look also at the unicode strip here.
|
|
date = self._time_created_at(result.get('created_at', None))
|
|
tweetid = result.get('id_str', None)
|
|
self._outputTweet(irc, msg, nick.encode('utf-8'), name.encode('utf-8'), text.encode('utf-8'), date, tweetid)
|
|
|
|
tsearch = wrap(tsearch, [getopts({'num':('int'), 'searchtype':('literal', ('popular', 'mixed', 'recent')), 'lang':('somethingWithoutSpaces')}), ('text')])
|
|
|
|
def twitter(self, irc, msg, args, optlist, optnick):
|
|
"""[--noreply] [--nort] [--num number] <nick> | <--id id> | [--info nick]
|
|
|
|
Returns last tweet or 'number' tweets (max 10). Shows all tweets, including rt and reply.
|
|
To not display replies or RT's, use --noreply or --nort, respectively.
|
|
Or returns tweet with id 'id'.
|
|
Or returns information on user with --info.
|
|
"""
|
|
|
|
# before we do anything, make sure we have a twitterApi object.
|
|
if not self.twitterApi:
|
|
irc.reply("ERROR: Twitter is not authorized. Please check logs before running this command.")
|
|
return
|
|
|
|
optnick = optnick.replace('@','') # strip @ from input if given.
|
|
|
|
args = {'id': False, 'nort': False, 'noreply': False, 'num': self.registryValue('defaultResults', msg.args[0]), 'info': False}
|
|
|
|
# handle input optlist.
|
|
if optlist:
|
|
for (key, value) in optlist:
|
|
if key == 'id':
|
|
args['id'] = True
|
|
if key == 'nort':
|
|
args['nort'] = True
|
|
if key == 'noreply':
|
|
args['noreply'] = True
|
|
if key == 'num':
|
|
if value > self.registryValue('maxResults', msg.args[0]) or value <= 0:
|
|
irc.reply("Error: '{0}' is not a valid number of tweets. Range is above 0 and max {1}.".format(value, max))
|
|
return
|
|
else:
|
|
args['num'] = value
|
|
if key == 'info':
|
|
args['info'] = True
|
|
|
|
# handle the three different rest api endpoint urls + twitterArgs dict for options.
|
|
if args['id']: # -id #.
|
|
apiUrl = 'statuses/show'
|
|
twitterArgs = {'id': optnick, 'include_entities':'false'}
|
|
elif args['info']: # --info.
|
|
apiUrl = 'users/show'
|
|
twitterArgs = {'screen_name': optnick, 'include_entities':'false'}
|
|
else: # if not an --id or --info, we're printing from their timeline.
|
|
apiUrl = 'statuses/user_timeline'
|
|
twitterArgs = {'screen_name': optnick, 'count': args['num']}
|
|
if args['nort']: # When set to false, the timeline will strip any native retweets.
|
|
twitterArgs['include_rts'] = 'false'
|
|
else:
|
|
twitterArgs['include_rts'] = 'true'
|
|
|
|
if args['noreply']: # This parameter will prevent replies from appearing in the returned timeline.
|
|
twitterArgs['exclude_replies'] = 'true'
|
|
else:
|
|
twitterArgs['exclude_replies'] = 'false'
|
|
|
|
# now with and call the api.
|
|
data = self.twitterApi.ApiCall(apiUrl, parameters=twitterArgs)
|
|
try:
|
|
data = json.loads(data.read())
|
|
except:
|
|
irc.reply("ERROR: Failed to lookup Twitter account for '{0}' ({1}) ".format(optnick, data))
|
|
return
|
|
|
|
# check for errors
|
|
if 'errors' in data:
|
|
errmsg = "" # prep string for output
|
|
if data['errors'][0]['code']:
|
|
errmsg += "{0} ".format(data['errors'][0]['code'])
|
|
if data['errors'][0]['message']:
|
|
errmsg += " {0}".format(data['errors'][0]['message'])
|
|
irc.reply("ERROR: {0}".format(errmsg))
|
|
return
|
|
|
|
# process the data.
|
|
if args['id']: # If --id was given for a single tweet.
|
|
text = self._unescape(data.get('text', None))
|
|
nick = data["user"].get('screen_name', None)
|
|
name = data["user"].get('name', None)
|
|
relativeTime = self._time_created_at(data.get('created_at', None))
|
|
tweetid = data.get('id', None)
|
|
self._outputTweet(irc, msg, nick.encode('utf-8'), name.encode('utf-8'), text.encode('utf-8'), relativeTime, tweetid)
|
|
return
|
|
elif args['info']: # Works with --info to return info on a Twitter user.
|
|
location = data.get('location', None)
|
|
followers = data.get('followers_count', None)
|
|
friends = data.get('friends_count', None)
|
|
description = data.get('description', None)
|
|
screen_name = data.get('screen_name', None)
|
|
name = data.get('name', None)
|
|
url = data.get('url', None)
|
|
|
|
# build output string conditionally.
|
|
ret = self._bu("@" + screen_name.encode('utf-8'))
|
|
ret += " ({0}):".format(name.encode('utf-8'))
|
|
if url:
|
|
ret += " {0}".format(self._ul(url.encode('utf-8')))
|
|
if description:
|
|
ret += " {0}".format(description.encode('utf-8'))
|
|
ret += " [{0} friends,".format(self._bold(friends))
|
|
ret += " {0} followers.".format(self._bold(followers))
|
|
if location:
|
|
ret += " Location: {0}]".format(location.encode('utf-8'))
|
|
else:
|
|
ret += "]"
|
|
irc.reply(ret) # output info.
|
|
return
|
|
else: # this will display tweets/a user's timeline.
|
|
if len(data) == 0: # If we have no data, user has not tweeted.
|
|
irc.reply("ERROR: '{0}' has not tweeted yet.".format(optnick))
|
|
return
|
|
for tweet in data: # iterate through each tweet.
|
|
text = self._unescape(tweet.get('text', None))
|
|
nick = tweet["user"].get('screen_name', None)
|
|
name = tweet["user"].get('name', None)
|
|
tweetid = tweet.get('id', None)
|
|
relativeTime = self._time_created_at(tweet.get('created_at', None))
|
|
self._outputTweet(irc, msg, nick.encode('utf-8'), name.encode('utf-8'), text.encode('utf-8'), relativeTime, tweetid)
|
|
|
|
twitter = wrap(twitter, [getopts({'noreply':'', 'nort':'', 'info':'', 'id':'', 'num':('int')}), ('somethingWithoutSpaces')])
|
|
|
|
Class = Tweety
|
|
|
|
|
|
# vim:set shiftwidth=4 softtabstop=4 expandtab textwidth=279:
|