70 lines
2.9 KiB
Python
70 lines
2.9 KiB
Python
|
import datetime, re, time, traceback
|
||
|
|
import twitter
|
||
|
|
import Utils
|
||
|
|
|
||
|
|
REGEX_TWITTERURL = re.compile(
|
||
|
|
"https?://(?:www\.)?twitter.com/[^/]+/status/(\d+)", re.I)
|
||
|
|
|
||
|
|
class Module(object):
|
||
|
|
def __init__(self, bot):
|
||
|
|
self.bot = bot
|
||
|
|
bot.events.on("received").on("command").on("twitter", "tw"
|
||
|
|
).hook(self.twitter)
|
||
|
|
|
||
|
|
def make_timestamp(self, s):
|
||
|
|
seconds_since = time.time()-datetime.datetime.strptime(s,
|
||
|
|
"%a %b %d %H:%M:%S %z %Y").timestamp()
|
||
|
|
since, unit = Utils.time_unit(seconds_since)
|
||
|
|
return "%s %s ago" % (since, unit)
|
||
|
|
|
||
|
|
def twitter(self, event):
|
||
|
|
api_key = self.bot.config["twitter-api-key"]
|
||
|
|
api_secret = self.bot.config["twitter-api-secret"]
|
||
|
|
access_token = self.bot.config["twitter-access-token"]
|
||
|
|
access_secret = self.bot.config["twitter-access-secret"]
|
||
|
|
|
||
|
|
if event["args"]:
|
||
|
|
target = event["args"]
|
||
|
|
else:
|
||
|
|
target = event["log"].find(REGEX_TWITTERURL)
|
||
|
|
if target:
|
||
|
|
target = target.message
|
||
|
|
if target:
|
||
|
|
twitter_object = twitter.Twitter(auth=twitter.OAuth(
|
||
|
|
access_token, access_secret, api_key, api_secret))
|
||
|
|
url_match = re.search(REGEX_TWITTERURL, target)
|
||
|
|
if url_match or target.isdigit():
|
||
|
|
tweet_id = url_match.group(1) if url_match else target
|
||
|
|
try:
|
||
|
|
tweet = twitter_object.statuses.show(id=tweet_id)
|
||
|
|
except:
|
||
|
|
traceback.print_exc()
|
||
|
|
tweet = None
|
||
|
|
elif target.startswith("@"):
|
||
|
|
try:
|
||
|
|
tweet = twitter_object.statuses.user_timeline(
|
||
|
|
screen_name=target[1:], count=1)[0]
|
||
|
|
except:
|
||
|
|
traceback.print_exc()
|
||
|
|
tweet = None
|
||
|
|
if tweet:
|
||
|
|
username = "@%s" % tweet["user"]["screen_name"]
|
||
|
|
if "retweeted_status" in tweet:
|
||
|
|
original_username = "@%s" % tweet["retweeted_status"
|
||
|
|
]["user"]["screen_name"]
|
||
|
|
original_text = tweet["retweeted_status"]["text"]
|
||
|
|
retweet_timestamp = self.make_timestamp(tweet[
|
||
|
|
"created_at"])
|
||
|
|
original_timestamp = self.make_timestamp(tweet[
|
||
|
|
"retweeted_status"]["created_at"])
|
||
|
|
event["stdout"].write("(%s (%s) retweeted %s (%s)) %s" % (
|
||
|
|
username, retweet_timestamp,
|
||
|
|
original_username, original_timestamp, original_text))
|
||
|
|
else:
|
||
|
|
event["stdout"].write("(%s, %s) %s" % (username,
|
||
|
|
self.make_timestamp(tweet["created_at"]), tweet["text"]))
|
||
|
|
else:
|
||
|
|
event["stderr"].write("Invalid tweet identifiers provided")
|
||
|
|
else:
|
||
|
|
event["stderr"].write("No tweet provided to get information about")
|