import dns.resolver

from pprint import pprint


class DNS():
    def __init__(self, domain):
        self.domain = domain
        self.a = []
        self.aaaa = []
        self.mx = []
        self.txt = []
        self.ns = []
        self.soa = []
        if domain != 'blank':
            self.getRecords()

    def getRecords(self):
        try:
            for t in dns.resolver.query(self.domain, 'A'):
                self.a.append(str(t))
        except Exception as E:
            print(E)
        try:
            for t in dns.resolver.query(self.domain, 'AAAA'):
                self.aaaa.append(str(t))
        except Exception as E:
            print(E)
        try:
            for t in dns.resolver.query(self.domain, 'MX'):
                self.mx.append(str(t).split())
        except Exception as E:
            print(E)
        try:
            for t in dns.resolver.query(self.domain, 'TXT'):
                self.txt.append(str(t))
        except Exception as E:
            print(E)
        try:
            for t in dns.resolver.query(self.domain, 'NS'):
                self.ns.append(str(t))
        except Exception as E:
            print(E)
        try:
            for t in dns.resolver.query(self.domain, 'SOA'):
                self.soa.append(str(t))
        except Exception as E:
            print(E)