From 90591ec5b4e010b14428c9eef33f1f1a5aa9a262 Mon Sep 17 00:00:00 2001 From: jesopo Date: Thu, 7 Feb 2019 10:07:11 +0000 Subject: Change isgd.py to use Exports to offer url shortening, not events --- modules/isgd.py | 10 +++++----- modules/tweets.py | 3 +-- 2 files changed, 6 insertions(+), 7 deletions(-) (limited to 'modules') 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: -- cgit v1.3.1-10-gc9f91