1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
# --require-config twitter-api-key
# --require-config twitter-api-secret
# --require-config twitter-access-token
# --require-config twitter-access-secret
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, events, exports):
self.bot = bot
self.url_shortener_enabled = "bitly" in self.bot.modules.modules
self.events = events
events.on("received").on("command").on("tweet", "tw"
).hook(self.tweet,
help="Find a tweet",
usage="[@username/URL/ID]")
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 tweet(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["buffer"].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
else:
if target.startswith("@"):
taret = target[1:]
try:
tweet = twitter_object.statuses.user_timeline(
screen_name=target, count=1)[0]
except:
traceback.print_exc()
tweet = None
if tweet:
linked_id = tweet["id"]
username = "@%s" % tweet["user"]["screen_name"]
url_shortener_link = ""
if self.url_shortener_enabled:
chopped_uname = username[1:]
tweet_link = "https://twitter.com/%s/status/%s" % (
chopped_uname, linked_id)
url_shortener_link = " -- " + self.events.on("get").on(
"shortlink").call(
url=tweet_link)[0]
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 %s" % (
username, retweet_timestamp,
original_username, original_timestamp,
original_text,
url_shortener_link))
else:
event["stdout"].write("(%s, %s) %s %s" % (username,
self.make_timestamp(
tweet[
"created_at"]),
tweet["text"],
url_shortener_link)
)
else:
event["stderr"].write("Invalid tweet identifiers provided")
else:
event["stderr"].write("No tweet provided to get information about")
|
