批量域名dig---使用协程

#!/usr/bin/env python
# _*_coding:utf-8_*_

from gevent import monkey
import gevent
import time
import dns.resolver
monkey.patch_socket()

class MyGeven:

    def __init__(self):
        self.data = {}

    def get_dig_domain(self, domain):
                res = dns.resolver.query(domain, 1)
        for i in res.response.answer:
                for j in i.items:
                 if j.rdtype == 5:
               find_res = j.to_text().find('***dns.com')
                         self.data[domain] = 0 if find_res != -1 else -1
                 else:
                         self.data[domain] = -1
                break

    def gprobe(self, domain_list):
        jobs = []
        for domain in domain_list:
            jobs.append(gevent.spawn(self.get_dig_domain, domain))
        gevent.joinall(jobs)

if __name__ == '__main__':
    start_ts = time.time()
    print "start....."
    domain_list = [
            "www.baidu.com",
                "www.qq.com",
                "www.163.com",
            "dl.3366.com"
    ]
    obj = MyGeven()
    obj.gprobe(domain_list)
    print "请求数:", len(domain_list), "耗时:", time.time() - start_ts, "s"

    print "域名dig:", obj.data

本文转自cloves 51CTO博客，原文链接:http://blog.51cto.com/yeqing/1925168