python实现besttrace 归属地功能

N1盒子安装的armbian,之前一直使用besttrace来测试路由的,因为有归属地,所以比较直观,结果没找到arm版本的besttrace,随手搜了下,有些类似的脚本,不过,基本都失效了,自己就简单的搞个,内部使用了自己的ip归属地接口,使用的纯真ip库,效果还行,就是接口速度因为网络原因,稍慢,可以自己改成别的接口的.

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

import sys
import os
import re
import urllib2
import subprocess
import platform

header={
          'User-Agent': 'Dalvik/1.6.0 (Linux; U; Android 4.4.4; HM NOTE 1S MIUI/V6.6.1.0.KHKCNCF)',
          }

def iploc(ipaddr):
    try:
        url = "http://www.112114.xyz/iploc/{}".format(ipaddr)
        request = urllib2.Request(url,headers=header)
        u = urllib2.urlopen(request)
        s = u.read()
    except Exception,e:
        s = ""
    return s

def trpy(iphost):
    system_name = platform.dist()[0]
    if system_name == 'redhat':
        trace_cmd = '/bin/traceroute'
    elif system_name == 'Ubuntu':
        trace_cmd = '/usr/sbin/traceroute'
    else:
        trace_cmd = '/bin/traceroute'
    try:
        p = subprocess.Popen([trace_cmd,iphost],stdout=subprocess.PIPE)
        while True:
            line = p.stdout.readline()
            if not line:
                break
            ips = re.findall(r"\((\b(?:[0-9]{1,3}\.){3}[0-9]{1,3}\b)\)", line)
            if ips:
                try:
                    ip = ips[0]
                    print(line.replace('\n', '').replace('\r', '') + ", " + iploc(ip))
                except IndexError,e:
                    print(line)
            else:
                print(line)
    except (KeyboardInterrupt,SystemExit):
        sys.exit()


if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("Usage: {} <hostname|ip>".format(sys.argv[0]))
        sys.exit()
    else:
        iphost = sys.argv[1]
    trpy(iphost)

本地ip库速度比较快,不过只能本机使用,最好自己写个接口来调用.
这里使用的是纯真ip库,下载ip库后,需要安装纯正ip库的模块
pip install qqwry-py3

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

import sys
import os
import re
import subprocess
import platform
from qqwry import QQwry


q = QQwry()
dbfile = "/home/sign/czip/qqwry.dat"
header={
          'User-Agent': 'Dalvik/1.6.0 (Linux; U; Android 4.4.4; HM NOTE 1S MIUI/V6.6.1.0.KHKCNCF)',
          }

def lookup(ip):
    q.load_file(dbfile)
    ipinfo = q.lookup(ip)
    return ipinfo


def trpy(iphost):
    system_name = platform.dist()[0]
    if system_name == 'redhat':
        trace_cmd = '/bin/traceroute'
    elif system_name == 'Ubuntu':
        trace_cmd = '/usr/sbin/traceroute'
    else:
        trace_cmd = '/bin/traceroute'
    try:
        p = subprocess.Popen([trace_cmd,iphost],stdout=subprocess.PIPE)
        while True:
            line = p.stdout.readline()
            if not line:
                break
            line = line.decode().replace('\n', '').replace('\r', '')
            ips = re.findall(r"\(((?:[0-9]{1,3}\.){3}[0-9]{1,3})\)", line)
            if ips:
                try:
                    ip = ips[0]
                    print(line + ", " + ' '.join(lookup(ip)))
                except IndexError as e:
                    print()
            else:
                print(line)
    except (KeyboardInterrupt,SystemExit):
        sys.exit()


if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("Usage: {} <hostname|ip>".format(sys.argv[0]))
        sys.exit()
    else:
        iphost = sys.argv[1]
    trpy(iphost)