bitbot-3.11-fork/modules/dns.py

import socket


class Module(object):
    _name = "DNS"

    def __init__(self, bot):
        bot.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)))