python实现域名解析和归属地查询

前言
工作中有时要查询域名解析和获取域名相关IP归属地信息
安装依赖
python2:
pip install dnspython
python3:
python3 -m pip install -i https://pypi.douban.com/simple/ dnspython3

脚本内容

#!/usr/bin/env python

#coding:utf-8

import dns.resolver

import urllib

import chardet

import sys

import re

ip_list = []

query_domain = sys.argv[1]

#query_domain = raw_input('Please input a domain: ')

# 先查询是否有CNAME，如果有则通过CNAME获取到A记录；如果没有直接获取A记录，然后返回到一个列表中

def get_iplist(domain):

    try:

        cn = dns.resolver.query(domain,'CNAME')

        for i in cn.response.answer:

            for j in i.items:

                cname_domain = j.to_text()

        print '========================================'

        print '    CNAME  --->', cname_domain

        print '========================================'

    except:

        pass

    else:

        A = dns.resolver.query(cname_domain,'A')

        for i in A.response.answer:

            for j in i.items:

                try:

                    ip_list.append(j.address)

                except:

                    get_iplist(str(j))

    try:

        A = dns.resolver.query(domain,'A')

        for i in A.response.answer:

            for j in i.items:

                ip_list.append(j.address)

    except:

        pass

    return ip_list

# 通过调用ip138.com来获取IP归属地信息

def get_ipbelong(domain_ip):

    url = 'http://ip138.com/ips138.asp?ip=%s&action=2' % domain_ip

    content = urllib.urlopen(url).read()

    typeEncode = sys.getfilesystemencoding()

    infoencode = chardet.detect(content).get('encoding','utf-8')

    html = content.decode(infoencode,'ignore').encode(typeEncode)

    ip = re.findall(r'\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}',html)

    result = re.findall(r'(<li>.*?</li>)',html)

    belong = result[0][14:-5]

    print '%-*s--->%*s' % (18,ip[0],20,belong)

if __name__ == '__main__':

    get_iplist(query_domain)

    for ip in ip_list:

        get_ipbelong(ip)

使用
python domain2ip.py www.baidu.com

秒客网

python实现域名解析和归属地查询

相关文章