Add first version of modules/imgur.py
This commit is contained in:
parent
a4d8d1f855
commit
f073613848
2 changed files with 39 additions and 0 deletions
|
@ -40,3 +40,5 @@ pushbullet-channel-tag =
|
||||||
omdbapi-api-key =
|
omdbapi-api-key =
|
||||||
# https://www.virustotal.com/en/documentation/public-api/
|
# https://www.virustotal.com/en/documentation/public-api/
|
||||||
virustotal-api-key =
|
virustotal-api-key =
|
||||||
|
# https://apidocs.imgur.com/
|
||||||
|
imgur-api-key =
|
||||||
|
|
37
modules/imgur.py
Normal file
37
modules/imgur.py
Normal file
|
@ -0,0 +1,37 @@
|
||||||
|
#--require-config imgur-api-key
|
||||||
|
|
||||||
|
from src import ModuleManager, utils
|
||||||
|
|
||||||
|
REGEX_IMAGE = "https?://(?:i\.)?imgur.com/(\w{7})"
|
||||||
|
URL_IMAGE = "https://api.imgur.com/3/image/%s"
|
||||||
|
|
||||||
|
class Module(ModuleManager.BaseModule):
|
||||||
|
def _image_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 = "%s: " % data["id"]
|
||||||
|
if data["nsfw"]:
|
||||||
|
text += "[NSFW] "
|
||||||
|
|
||||||
|
text += "(%s %dx%d, %d views) " % (data["type"], data["width"],
|
||||||
|
data["height"], data["views"])
|
||||||
|
if data["title"]:
|
||||||
|
text += " %s" % data["title"]
|
||||||
|
return text
|
||||||
|
else:
|
||||||
|
raise utils.EventsResultsError()
|
||||||
|
|
||||||
|
@utils.hook("received.command.imgur", min_args=1)
|
||||||
|
def imgur(self, event):
|
||||||
|
"""
|
||||||
|
:help: Get information about a given imgur image URL
|
||||||
|
:usage: <url>
|
||||||
|
"""
|
||||||
|
match = REGEX_IMAGE.match(event["args_split"][0])
|
||||||
|
if match:
|
||||||
|
event["stdout"].write(self._image_info(match.group(1)))
|
Loading…
Reference in a new issue