Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Support galleries in modules/imgur.py

Browse source
This commit is contained in:
jesopo 2018-10-31 17:41:12 +00:00
parent 60151dc1a0
commit ff1d4da362
1 changed files with 30 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

34
modules/imgur.py
View file

@ -4,9 +4,17 @@ import re
from src import ModuleManager, utils from src import ModuleManager, utils
REGEX_IMAGE = re.compile("https?://(?:i\.)?imgur.com/(\w+)") REGEX_IMAGE = re.compile("https?://(?:i\.)?imgur.com/(\w+)")
REGEX_GALLERY = re.compile("https?://imgur.com/gallery/(\w+)")
URL_IMAGE = "https://api.imgur.com/3/image/%s" URL_IMAGE = "https://api.imgur.com/3/image/%s"
class Module(ModuleManager.BaseModule): class Module(ModuleManager.BaseModule):
def _prefix(self, data):
text = "%s: " % data["id"]
if data["nsfw"]:
text += "[NSFW] "
if data["account_url"]:
text += "%s " % data["account_url"]
def _image_info(self, hash): def _image_info(self, hash):
api_key = self.bot.config["imgur-api-key"] api_key = self.bot.config["imgur-api-key"]
result = utils.http.get_url(URL_IMAGE % hash, result = utils.http.get_url(URL_IMAGE % hash,
@ -15,14 +23,29 @@ class Module(ModuleManager.BaseModule):
if result and result["success"]: if result and result["success"]:
data = result["data"] data = result["data"]
text = "%s: " % data["id"] text = self._prefix(data)
if data["nsfw"]:
text += "[NSFW] "
text += "(%s %dx%d, %d views) " % (data["type"], data["width"], text += "(%s %dx%d, %d views) " % (data["type"], data["width"],
data["height"], data["views"]) data["height"], data["views"])
if data["title"]: if data["title"]:
text += " %s" % data["title"] text += data["title"]
return text
else:
raise utils.EventsResultsError()
def _gallery_info(self, hash):
api_key = self.bot.config["imgur-api-key"]
result = utils.http.get_url(URL_IMAGE % hash,
headers={"Authorization": "Client-ID %s" % api_key},
json=True)
if result and result["success"]:
data = result["data"]
text = self._prefix(data)
text += "(%d views, %d▲▼%d)" % (event["views"],
event["ups"], event["downs"])
if data["title"]:
text += data["title"]
return text return text
else: else:
raise utils.EventsResultsError() raise utils.EventsResultsError()
@ -36,3 +59,6 @@ class Module(ModuleManager.BaseModule):
match = REGEX_IMAGE.match(event["args_split"][0]) match = REGEX_IMAGE.match(event["args_split"][0])
if match: if match:
event["stdout"].write(self._image_info(match.group(1))) event["stdout"].write(self._image_info(match.group(1)))
match = REGEX_GALLERY.match(event["args_split"][0])
if match:
event["stdout"].write(self._gallery_info(match.group(1)))