使用python判断网站是否存在CDN

有些网站会做cdn，当你测试一个网站时，测了半天，发现你跑偏了，这时想死的心都有了。。。。。

可以通过dns的cname记录和http返回头来判断是否使用了cdn

CNAME：
我做的百度云加速，cname配置


这时你可以nslookup，得到结果如下：
C:\Users\WYB_9>nslookup 
服务器:  
Address:  114.114.114.114

非权威应答:
名称:    
Addresses:  162.159.210.12
          162.159.211.12
Aliases:  

HTTP返回包：
Http返回头有类似下面的信息也可判断为使用了cdn
X-Via: 1.1 zhshx14:3 (Cdn Cache Server V2.0), 1.1 oudxin55:6 (Cdn Cache Server V2.0)

首先安装dnspython模块：pip install dnspython

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

import sys
import dns.resolver
import urllib2
import urlparse

class CdnCheck(object):
    def __init__(self, url):
        super(CdnCheck, self).__init__()
        self.cdninfo()
        self.url = url
         = []
        self.headers = []

    def get_cnames(self): # get all cname
        furl = urlparse.urlparse(self.url)
        url = 
        # print url

        rsv = dns.resolver.Resolver()
        # rsv.nameservers = ['114.114.114.114']
        try:
            answer = dns.resolver.query(url,'CNAME')
        except Exception as e:
             = None
            # print "ERROR: %s" % e
        else:
            cname = [_.to_text() for _ in answer][0]
            
            self.get_cname(cname)

    def get_cname(self,cname): # get cname
        try:
            answer = dns.resolver.query(cname,'CNAME')
            cname = [_.to_text() for _ in answer][0]
            
            self.get_cname(cname)
        except dns.resolver.NoAnswer:
            pass

    def get_headers(self): # get header
        try:
            resp = urllib2.urlopen(self.url)
        except Exception as e:
            self.headers = None
            # print "ERROR: %s" % e
        else:
            headers = str(resp.headers).lower()
            self.headers = headers

    def matched(self, context, *args): # Matching string
        if not isinstance(context, basestring):
            context = str(context)

        func = lambda x, y: y in x
        for pattern in args:
            if func(context,pattern):
                return pattern
        return False

    def check(self):
        flag = None
        self.get_cnames()
        self.get_headers()
        if 
            # print 
            flag = 
            if flag:
                return {'Status':True, 'CDN':self.cdn['cname'].get(flag)}
        if not flag and self.headers:
            flag = self.matched(self.headers,*self.cdn['headers'])
            if flag:
                return {'Status':True, 'CDN':'unknown'}
        return {'Status':False,  'Headers':self.headers}

    def cdninfo(self):
        self.cdn = {
            'headers': set([
                'via',
                'x-via',
                'by-360wzb',
                'by-anquanbao',
                'cc_cache',
                'cdn cache server',
                'cf-ray',
                'chinacache',
                'verycdn'
                'webcache',
                'x-cacheable',
                'x-fastly',
                'yunjiasu',
            ]),
            'cname': {
                 # 应该是淘宝自己的。。。。
                 # 应该是腾讯的。。。
                'yunjiasu-cdn':u'Baiduyun', # 百度云加速
                 # 阿里云
                 # 阿里云
                 # KeyCDN
                 # Leaseweb
                 # 网宿科技
                 # 网宿科技
                # 其余的特征可以自己找一下
            }
        }

if __name__ == '__main__':
    # url = 
    url = sys.argv[1]
    cdn = CdnCheck(url)
    print cdn.check()

结果如下：
C:\Users\WYB_9\Desktop>python check_cdn.py 
{'Status': True, 'CDN': u'Baiduyun'}

C:\Users\WYB_9\Desktop>python check_cdn.py 
{'Status': True, 'CDN': u'ChinaCache'}