bitbot-3.11-fork/modules/isgd.py

32 lines
988 B
Python
Raw Normal View History

2016-03-29 15:20:50 +00:00
import re
from src import ModuleManager, utils
2016-03-29 15:20:50 +00:00
ISGD_API_URL = "https://is.gd/create.php"
2016-03-29 15:20:50 +00:00
REGEX_URL = re.compile("https?://", re.I)
class Module(ModuleManager.BaseModule):
@utils.hook("get.shortlink")
2016-03-29 15:20:50 +00:00
def shortlink(self, event):
2018-09-09 12:20:59 +00:00
url = event["url"]
2016-03-29 15:20:50 +00:00
if not re.match(REGEX_URL, url):
url = "http://%s" % url
page = utils.http.request(ISGD_API_URL, get_params=
{"format": "json", "url": url}, json=True)
if page and page.data["shorturl"]:
return page.data["shorturl"]
2016-03-29 15:20:50 +00:00
@utils.hook("received.command.shorten", min_args=1)
2016-03-29 15:20:50 +00:00
def shorten(self, event):
"""
:help: Shorten a given URL using the is.gd service
:usage: <url>
"""
link = self.events.on("get.shortlink").call_for_result(
url=event["args"])
2016-03-29 15:20:50 +00:00
if link:
event["stdout"].write("Shortened URL: %s" % link)
2016-03-29 15:20:50 +00:00
else:
event["stderr"].write("Unable to shorten that URL.")