Recreate tweets.py because i'm lazy.

2018-09-09 16:11:02 +01:00 · 2018-09-09 16:11:02 +01:00 · 165efe20b9
commit 165efe20b9
parent d8c4293dbc
1 changed files with 89 additions and 0 deletions
--- a/modules/tweets.py
+++ b/modules/tweets.py
@ -0,0 +1,89 @@
+# --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.events = events
+
+        events.on("received.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 = tweet["user"]["screen_name"]
+
+                tweet_link = "https://twitter.com/%s/status/%s" % (username,
+                    linked_id)
+
+                short_url = self.events.on("get.shortlink").call_for_result(
+                    url=tweet_link)
+                short_url = " -- %s" % short_url if short_url else ""
+
+                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, short_url))
+                else:
+                    event["stdout"].write("(@%s, %s) %s%s" % (username,
+                        self.make_timestamp(tweet["created_at"]),
+                        tweet["text"], short_url))
+            else:
+                event["stderr"].write("Invalid tweet identifiers provided")
+        else:
+            event["stderr"].write("No tweet provided to get information about")