dnspython读取dnszone所有记录

需求

有时需要从zone文件里解析DNS的数据，做一些分析。可以利用dnspython库做zone的解析处理。
简单写一个如下

#!/usr/bin/env python  
import dns.zone  
import dns.ipv4  
import os.path  
import sys  
import os  
import string  
import re  
zonefile=sys.argv[1]  
name=sys.argv[2]  
class LOAD_ZONES:  
def  __init__(self,zonefile,name):  
self.types=("A","CNAME","TXT","SOA","MX","NS","PTR")  
self.record=dict()  
self.zonefile=zonefile  
self.name=name  
for type in self.types:  
self.record[type]=dict()  
try:  
zoneload= dns.zone.from_file(f=self.zonefile,origin=self.name,check_origin=False,relativize=False)  
for type in self.types:  
for (name, ttl, rdata) in zoneload.iterate_rdatas(type):  
dname=str(name)  
l = self.record[type].get(dname)  
if l is None:  
self.record[type][dname] = set()  
if type == "A":  
self.record[type][dname].add( ("%s-%s") % (ttl,rdata.address ) )  
else:  
self.record[type][dname].add( ("%s-%s") % (ttl,rdata.to_text().lower()) )  
except dns.zone.NoSOA:  
print "%s no soa" % self.zonefile  
except dns.zone.NoNS:  
print "%s no ns" % self.zonefile  
def dump_record(self):  
for type in self.types:  
for k in self.record[type].keys():  
print k,type,self.record[type][k]  
def __del__(self):  
pass  
  
dns=LOAD_ZONES(zonefile,name)  
dns.dump_record()

输入

./load_zone.py test.org.zone test.org