70 lines
1.6 KiB
Python
70 lines
1.6 KiB
Python
# analyzeutl 是分析器工具集合
|
||
import awdb
|
||
|
||
|
||
# 过滤所有空值
|
||
def filterNull(s):
|
||
if s == "":
|
||
return "UnKnown"
|
||
return s
|
||
|
||
|
||
def IP46(IP: str):
|
||
if ':' in IP:
|
||
return "v6"
|
||
if '.' in IP:
|
||
return "v4"
|
||
return "Unknown"
|
||
|
||
|
||
path_ip4app = "./data/IP_scene_all_cn.awdb"
|
||
path_ip6 = "./data/IP_city_single_BD09_WGS84_ipv6.awdb"
|
||
path_ip4qvxian = "./data/IP_basic_single_WGS84.awdb"
|
||
|
||
|
||
# 实例化数据读取器
|
||
def makereader(arg=0):
|
||
# 默认加载所有离线数据
|
||
dloader_ip4app = awdb.open_database(path_ip4app)
|
||
dloader_ip6 = awdb.open_database(path_ip6)
|
||
dloader_ip4qx = awdb.open_database(path_ip4qvxian)
|
||
return dloader_ip4app, dloader_ip4qx, dloader_ip6
|
||
|
||
|
||
reader_ip4app, reader_ip4qx, reader_ip6 = makereader()
|
||
|
||
|
||
# # 传入一个ip地址，根据对象中的数据进行IP关联分析
|
||
# def make_IPinfo(ip):
|
||
# record=getrecord(ip)
|
||
# cou = record.get('areacode', b'').decode("utf-8")
|
||
# multiarea = record.get('multiAreas', {})
|
||
# if multiarea:
|
||
# print("有多组记录")
|
||
# else:
|
||
# pass
|
||
# return record
|
||
|
||
# 返回IP离线库中与ip相关的记录，记录中不包含应用场景
|
||
def getrecord(ip):
|
||
if (IP46(ip) == "v4"):
|
||
return IP4_info(ip)
|
||
elif (IP46(ip) == "v6"):
|
||
return IP6_info(ip)
|
||
else:
|
||
print("地址存在问题")
|
||
print(ip)
|
||
return 1
|
||
|
||
|
||
# 返回IPv4记录
|
||
def IP4_info(ip):
|
||
(record, prefix_len) = reader_ip4qx.get_with_prefix_len(ip)
|
||
return record
|
||
|
||
|
||
# 返回IPv6记录
|
||
def IP6_info(ip):
|
||
(record, prefix_len) = reader_ip6.get_with_prefix_len(ip)
|
||
return record
|