diff --git a/modules/isgd.py b/modules/isgd.py index 470c0dda..7674439c 100644 --- a/modules/isgd.py +++ b/modules/isgd.py @@ -5,9 +5,10 @@ ISGD_API_URL = "https://is.gd/create.php" REGEX_URL = re.compile("https?://", re.I) class Module(ModuleManager.BaseModule): - @utils.hook("get.shortlink") - def shortlink(self, event): - url = event["url"] + def on_load(self): + self.exports.add("shortlink", self._shortlink) + + def _shortlink(self, url): if not re.match(REGEX_URL, url): url = "http://%s" % url @@ -23,8 +24,7 @@ class Module(ModuleManager.BaseModule): :help: Shorten a given URL using the is.gd service :usage: """ - link = self.events.on("get.shortlink").call_for_result( - url=event["args"]) + link = self._shortlink(event["args"]) if link: event["stdout"].write("Shortened URL: %s" % link) else: diff --git a/modules/tweets.py b/modules/tweets.py index 73091171..55cf28e4 100644 --- a/modules/tweets.py +++ b/modules/tweets.py @@ -64,8 +64,7 @@ class Module(ModuleManager.BaseModule): 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 = self.exports.get_one("shortlink")(tweet_link) short_url = " - %s" % short_url if short_url else "" if "retweeted_status" in tweet: