#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2016/10/13 10:30
# @Author  : Beam
# @Site    : 實(shí)現(xiàn)查詢域名的A記錄
# @File    : demo_dns.py
# @Software: PyCharm
 
import dns.resolver
 
def getArember(domain):
    """
    僅查詢二級域名，頂級域名則出錯(cuò)
    :param domain:
    :return:打印該域名下的所有A記錄
    """
    A = dns.resolver.query(domain,'A')   #指定查詢A記錄
    print '-----------------------A記錄-----------------------'
    for i in A.response.answer:     #通過response.answer方法獲取查詢回應(yīng)的信息
        for j in i:
            print j
    print '-----------------------分割線-----------------------'
 
def getMXrember(domain):
    """
    僅限輸入一級域名（頂級域名），否則出錯(cuò)
    :param domain:
    :return:打印該域名下的所有MX記錄
    """
    MX = dns.resolver.query(domain,'MX')   #指定查詢MX記錄
    print '-----------------------MX記錄-----------------------'
    for i in MX:     #遍歷回應(yīng)結(jié)果，輸出MX記錄的preference以及exchanger信息
        print "MX preference =", i.preference,"MX exchanger =",i.exchange
    print '-----------------------分割線-----------------------'
 
def getNsrember(domain):
    """
    僅限輸入一級域名（頂級域名），否則出錯(cuò)
    :param domain:
    :return: 打印該域名的NS記錄值
    """
    ns = dns.resolver.query(domain,'NS')
    print '-----------------------NS記錄-----------------------'
    for i in ns.response.answer:
        for j in i.items:
            print j.to_text(),
    print ''
    print '-----------------------分割線-----------------------'
 
def getCnamerember(domain):
    """
    僅限輸入二級域名，否則出錯(cuò)
    :param domain:
    :return: 打印該域名的cname記錄值
    """
    cname = dns.resolver.query(domain,'CNAME')
    print '-----------------------CNAME記錄-----------------------'
    for i in cname.response.answer:
        for j in i.items:
            print j.to_text()
    print '-----------------------分割線-----------------------'
 
print '查詢頂級域名輸出MX、NS記錄，查詢耳機(jī)域名輸出A、CNAME記錄'
domain = raw_input('Please input an domain:')
if len(domain.strip().split('.')) == 2:
    getMXrember(domain)
    getNsrember(domain)
else:
    getArember(domain)
    getCnamerember(domain)