python dig

import subprocess

def dig(domain, record_type='A'):
    try:
        result = subprocess.run(['dig', '+noall', '+answer', f'{domain} {record_type}'], capture_output=True, text=True, check=True)
        return result.stdout
    except subprocess.CalledProcessError as e:
        return f"Error executing dig: {e}"

# 使用示例
print(dig('www.example.com'))

import dns.resolver

def query_dns(domain, record_type='A'):
    try:
        answers = dns.resolver.resolve(domain, record_type)
        return [rdata.to_text() for rdata in answers]
    except dns.resolver.NXDOMAIN:
        return f"The domain {domain} does not exist."
    except dns.resolver.NoAnswer:
        return f"The domain {domain} has no {record_type} records."
    except dns.resolver.Timeout:
        return f"Timed out while querying DNS for {domain}."
    except Exception as e:
        return f"An error occurred: {e}"

# 使用示例
print(query_dns('www.example.com'))