def get_ns_history_record(self): try: db = MySQL(SOURCE_CONFIG) except: pass full_list = [] sql = 'SELECT domain,ns,ns_ip FROM domain_ns WHERE ns!=""' try: db.query(sql) data = db.fetch_all_rows() except Exception as e: return "read tld Error" + str(e) for i in data: tld = str(i['domain']) ns = str(i['ns']) ns_list = ns.split(';') ns = ','.join(ns_list) ns_ip = str(i['ns_ip']) flag = '0' full_list.append((tld,ns,ns_ip,flag)) taskid = str(random.randint(10000000, 99999999)) self.write_tld(taskid) new_file_name = 'full_' + taskid with open("domain_dns_data/sec_compared/"+new_file_name, "w") as f: for t in full_list: f.write(t[0]+'\t'+'NS'+'\t'+t[1]+'\t'+t[3]+'\n') ns_list = t[1].split(',') ip_list = t[2].split(';') for i in range(0,len(ns_list)): f.write(ns_list[i]+'\t'+'A'+'\t'+ip_list[i]+'\t'+'0'+'\n') self.produce(new_file_name) self.produce('tld_'+taskid)
def fetch_tld_ns(): """ 获取顶级域名的权威服务器(ns)IP地址 """ tld_ns = defaultdict(set) try: db = MySQL(SOURCE_CONFIG) sql = 'SELECT tld,server_ipv4 from tld_ns_zone' db.query(sql) tld_ns_query = db.fetch_all_rows() # 获取已存储的顶级域名的权威服务器信息 except Exception, e: logger.logger.error("获取顶级域名异常:", e) return tld_ns