9874f79b49
hooks for a module
22 lines
750 B
Python
22 lines
750 B
Python
import socket
|
|
|
|
class Module(object):
|
|
_name = "DNS"
|
|
def __init__(self, bot, events):
|
|
events.on("received").on("command").on("dns").hook(
|
|
self.dns, min_args=1,
|
|
help="Get all addresses for a given hostname (IPv4/IPv6)",
|
|
usage="<hostname>")
|
|
|
|
def dns(self, event):
|
|
hostname = event["args_split"][0]
|
|
try:
|
|
address_info = socket.getaddrinfo(hostname, 1, 0,
|
|
socket.SOCK_DGRAM)
|
|
except socket.gaierror:
|
|
event["stderr"].write("Failed to find hostname")
|
|
return
|
|
ips = []
|
|
for _, _, _, _, address in address_info:
|
|
ips.append(address[0])
|
|
event["stdout"].write("%s: %s" % (hostname, ", ".join(ips)))
|