import dns.resolver
from Tools.domain_extract import Extracter


class DNSResolver:
    def __init__(self):
        pass

    @staticmethod
    def get_NS(resource_url):
        # domain 是资源二级域
        hostname, domain = Extracter.extract(resource_url)
        nameserver = []
        NS = dns.resolver.resolve(domain, "NS")
        for i in NS.response.answer:
            for j in i.items:
                nameserver.append(str(j))
        return nameserver


if __name__ == "__main__":
    a = DNSResolver()
    print(a.get_NS("https://www.baidu.com"))