set changed = True when we pop a module with no dependencies left
This commit is contained in:
jesopo
parent
ce3a0668c6
commit
786f04ea16
2 changed files with 2 additions and 123 deletions
|
|
@ -1,122 +0,0 @@
|
||||||
#--depends-on commands
|
|
||||||
#--depends-on config
|
|
||||||
|
|
||||||
#--require-config twitter-api-key
|
|
||||||
#--require-config twitter-api-secret
|
|
||||||
#--require-config twitter-access-token
|
|
||||||
#--require-config twitter-access-secret
|
|
||||||
|
|
||||||
import datetime, html, re, time, traceback
|
|
||||||
import twitter
|
|
||||||
from src import ModuleManager, utils
|
|
||||||
|
|
||||||
REGEX_TWITTERURL = re.compile(
|
|
||||||
"https?://(?:www\.)?twitter.com/[^/]+/status/(\d+)", re.I)
|
|
||||||
|
|
||||||
@utils.export("channelset", {"setting": "auto-tweet",
|
|
||||||
"help": "Enable/disable automatically getting tweet info",
|
|
||||||
"validate": utils.bool_or_none, "example": "on"})
|
|
||||||
class Module(ModuleManager.BaseModule):
|
|
||||||
_name = "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 _get_api(self):
|
|
||||||
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"]
|
|
||||||
return twitter.Twitter(auth=twitter.OAuth(
|
|
||||||
access_token, access_secret, api_key, api_secret))
|
|
||||||
|
|
||||||
def _from_id(self, tweet_id):
|
|
||||||
api = self._get_api()
|
|
||||||
try:
|
|
||||||
return api.statuses.show(id=tweet_id)
|
|
||||||
except:
|
|
||||||
traceback.print_exc()
|
|
||||||
|
|
||||||
def _format_tweet(self, tweet):
|
|
||||||
linked_id = tweet["id"]
|
|
||||||
username = tweet["user"]["screen_name"]
|
|
||||||
|
|
||||||
verified = ""
|
|
||||||
if tweet["user"]["verified"]:
|
|
||||||
verified = " %s" % utils.irc.color("✓", utils.consts.LIGHTBLUE)
|
|
||||||
|
|
||||||
tweet_link = "https://twitter.com/%s/status/%s" % (username,
|
|
||||||
linked_id)
|
|
||||||
|
|
||||||
short_url = self.exports.get_one("shortlink")(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"])
|
|
||||||
return "(@%s%s (%s) retweeted %s (%s)) %s%s" % (
|
|
||||||
username, verified, retweet_timestamp, original_username,
|
|
||||||
original_timestamp, html.unescape(original_text),
|
|
||||||
short_url)
|
|
||||||
else:
|
|
||||||
return "(@%s%s, %s) %s%s" % (username, verified,
|
|
||||||
self.make_timestamp(tweet["created_at"]),
|
|
||||||
html.unescape(tweet["text"]), short_url)
|
|
||||||
|
|
||||||
@utils.hook("received.command.tw", alias_of="tweet")
|
|
||||||
@utils.hook("received.command.tweet")
|
|
||||||
def tweet(self, event):
|
|
||||||
"""
|
|
||||||
:help: Get/find a tweet
|
|
||||||
:usage: [@username/URL/ID]
|
|
||||||
"""
|
|
||||||
|
|
||||||
if event["args"]:
|
|
||||||
target = event["args"]
|
|
||||||
else:
|
|
||||||
target = event["target"].buffer.find(REGEX_TWITTERURL)
|
|
||||||
if target:
|
|
||||||
target = target.message
|
|
||||||
if target:
|
|
||||||
url_match = re.search(REGEX_TWITTERURL, target)
|
|
||||||
if url_match or target.isdigit():
|
|
||||||
tweet_id = url_match.group(1) if url_match else target
|
|
||||||
tweet = self._from_id(tweet_id)
|
|
||||||
else:
|
|
||||||
if target.startswith("@"):
|
|
||||||
target = target[1:]
|
|
||||||
api = self._get_api()
|
|
||||||
try:
|
|
||||||
tweet = api.statuses.user_timeline(
|
|
||||||
screen_name=target, count=1)[0]
|
|
||||||
except:
|
|
||||||
traceback.print_exc()
|
|
||||||
tweet = None
|
|
||||||
if tweet:
|
|
||||||
tweet_str = self._format_tweet(tweet)
|
|
||||||
event["stdout"].write(tweet_str)
|
|
||||||
else:
|
|
||||||
event["stderr"].write("Invalid tweet identifiers provided")
|
|
||||||
else:
|
|
||||||
event["stderr"].write("No tweet provided to get information about")
|
|
||||||
|
|
||||||
@utils.hook("command.regex", pattern=REGEX_TWITTERURL)
|
|
||||||
def regex(self, event):
|
|
||||||
"""
|
|
||||||
:command: tweet
|
|
||||||
"""
|
|
||||||
if event["target"].get_setting("auto-tweet", False):
|
|
||||||
event.eat()
|
|
||||||
tweet_id = event["match"].group(1)
|
|
||||||
tweet = self._from_id(tweet_id)
|
|
||||||
if tweet:
|
|
||||||
tweet_str = self._format_tweet(tweet)
|
|
||||||
event["stdout"].write(tweet_str)
|
|
||||||
|
|
@ -250,6 +250,7 @@ class ModuleManager(object):
|
||||||
to_remove = []
|
to_remove = []
|
||||||
for name, dependencies in definition_dependencies.items():
|
for name, dependencies in definition_dependencies.items():
|
||||||
if not dependencies:
|
if not dependencies:
|
||||||
|
changed = True
|
||||||
# pop things with no unfufilled dependencies
|
# pop things with no unfufilled dependencies
|
||||||
to_remove.append(name)
|
to_remove.append(name)
|
||||||
for name in to_remove:
|
for name in to_remove:
|
||||||
|
|
@ -257,8 +258,8 @@ class ModuleManager(object):
|
||||||
del definition_dependencies[name]
|
del definition_dependencies[name]
|
||||||
for deps in definition_dependencies.values():
|
for deps in definition_dependencies.values():
|
||||||
if name in deps:
|
if name in deps:
|
||||||
# fulfill dependencies for things we just popped
|
|
||||||
changed = True
|
changed = True
|
||||||
|
# fulfill dependencies for things we just popped
|
||||||
deps.remove(name)
|
deps.remove(name)
|
||||||
|
|
||||||
if not changed:
|
if not changed:
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue