def compare(self, all_ip=IPSet(), input_ip=IPSet()) -> None:
'''将输入的 IP 范围与完整的 IP 范围做比较，输出缺少的部分'''
t0 = datetime.now()
self.result = '\n'.join([x.strCompressed()
for x in (all_ip - input_ip)])
t1 = (datetime.now() - t0).total_seconds()
pft(str_blue_green('运行用时：', '{:4.2f} s'.format(t1)))
def incomplete(self) -> None:
pft(str_blue('存在漏报的 IP：'))
if argv[-1] == '--cli':
print('\n'.join(self.result))
else:
result_file = '遗漏.-{}.txt'.format(self.now())
_write_file(result_file, self.result)
pft(str_blue_green('结果已输出到：', '{}'.format(result_file)))
def ip_from_last(_ip: str, ip_type: bool = False) -> list:
'''
ip_start - ip_end 转换为 ip / mask
ip_type == Ture: return [IP()] else : [ip_str]
'''
_start, _end = '', ''
if '\t' in _ip:
_start, _end = _ip.split('\t')
elif '-' in _ip:
_start, _end = _ip.split('-')
else:
try:
return ip_to_last(_ip)
except:
pft(str_red('IP 格式不对，请检查。'), str_blue_green('如果是文件记得加', ' -f'))
exit('>>运行结束。')
_start_int = IPint(_start).ip
_end_int = IPint(_end).ip
ip_len = _end_int-_start_int+1
# IP 范围个数转成二进制文本
ip_len_bin_str = '{:b}'.format(ip_len)
# 计算二进制文本的长度
_len = len(ip_len_bin_str)
# 判断 ipv4 or ipv6
_ver_len = 128 if _ip.find(':') != -1 else 32
# 计算最小掩码
_mask_min = _ver_len-_len+1
# 构建计算结果
def ip_handler(arg: IP):
return arg if ip_type else arg.strCompressed()
_res = []
_res_1 = IP(_start).make_net(_mask_min)
if _res_1.ip < _start_int:
_res_1 -= IP(_start_int-1)
for x in _res_1:
if x.ip >= _start_int:
_res.append(ip_handler(x))
else:
_res.append(ip_handler(_res_1))
if _res_1[-1].ip == _end_int:
# 转换结果恰好为一行
return _res
_middle = IP(_start_int+2**(_len-1)).make_net(_mask_min)
_res_2 = IP(_end).make_net(_mask_min)
if _middle.ip != _res_2.ip:
_res.append(ip_handler(_middle))
if _res_2[-1].ip > _end_int:
_res_2 -= IP(_end_int+1)
for x in _res_2:
if x.ip <= _end_int:
_res.append(ip_handler(x))
else:
_res.append(ip_handler(_res_2))
return _res
def main(self) -> None:
'''入口'''
help()
# session = PromptSession(history=FileHistory('.myhistory'))
pft(str_blue('输入包含完整 IP 地址范围的文件名：(默认 all.txt)'))
_all_ip_file = prompt('> ')
if(_all_ip_file == ''):
_all_ip_file = 'all.txt'
all_ip = self.get_input_ip(_all_ip_file)
pft(str_blue('输入要验证的文件名：(默认 input.txt)'))
_input_ip_file = prompt('> ')
if(_input_ip_file == ''):
_input_ip_file = 'input.txt'
input_ip = self.get_input_ip(_input_ip_file)
self.compare(all_ip, input_ip)
print('--------')
if(self.result):
self.incomplete()
else:
self.complete()
def get_input_ip(self, input_ip_file) -> IPSet:
'''
获取输入的 IP
'''
input_ip_str = _read_file(input_ip_file)
# 不使用下面的单行函数返回结果，是为了在循环中使用try，若格式异常可跳过，不影响整体运行。
# input_ip = IPSet([IP(x.strip(), make_net=True) for x in input_ip_str.split('\n')])
input_ip = IPSet()
for x in input_ip_str:
try:
# 尝试按照 ip / mask 解析
_i = IP(x.strip(), make_net=True)
input_ip.add(_i)
except Exception as err:
# 尝试按照 ip_start - ip_end 解析
try:
_i = ip_from_last(x.strip(), True)
for xx in _i:
input_ip.add(xx)
except Exception as err:
pft(str_red('忽略无效 ip：{}，\n\t错误原因：{}'.format(x, err)))
return input_ip
def main():
tokens = FormattedText([('fg:' + name, name + ' ')
for name in NAMED_COLORS])
pft(HTML('\n<u>Named colors, hex codes.</u>'))
print(", ".join([f"{k}: {v}" for k, v in NAMED_COLORS.items()]))
pft(HTML('\n<u>Named colors, using 16 color output.</u>'))
pft('(Note that it doesn\'t really make sense to use named colors ')
pft('with only 16 color output.)')
pft(tokens, color_depth=ColorDepth.DEPTH_4_BIT)
pft(HTML('\n<u>Named colors, use 256 colors.</u>'))
pft(tokens)
pft(HTML('\n<u>Named colors, using True color output.</u>'))
pft(tokens, color_depth=ColorDepth.TRUE_COLOR)
def main():
pft(HTML('<ansired>helloworl</ansired><ansigreen>d</ansigreen>'))
def complete(self) -> None:
pft(str_green('本次完整性检查无问题，好棒！'))
def help(more=False):
self_name = argv[0]
print(self_name)
title = '*** 欢迎使用 IP 地址完整性核查工具 ***'
pft(str_green(title))
print('→→ 版本：V{}\n→→ 作者：{}'.format(__version__, __author__))
pft(str_red('需要命令帮助，请运行：{} help'.format(self_name)))
pft(str_red('更多使用说明请查看 README.pdf'))
if more:
print('\n')
print('功能1：对比，找出遗漏的 IP 地址')
print('功能2：IP 地址格式转换（自动识别 ip / mask 与 ip_start - ip_end）')
print('功能3：IP 地址段合并')
print('\n')
pft(str_blue('使用示例：'))
pft(str_green('{} cp all.txt input.txt'.format(self_name)))
pft(str_green('{} cv 192.168.0.128-192.168.2.3'.format(self_name)))
pft(str_green('{} cv 2021:abcd::f00/120'.format(self_name)))
pft(str_green('{} cv -f input.txt > result.txt'.format(self_name)))
pft(str_green('{} m 192.168.0.0/25 192.168.0.128/25 192.168.1.0/24'.format(self_name)))
pft(str_green('{} m -f input.txt'.format(self_name)))
print('\n\n')
print('~~~~ !!!不会使用请仔细阅读 README.pdf !!!')
pft(str_green('{} m 192.168.0.0/25 192.168.0.128/25 192.168.1.0/24'.format(self_name)))
pft(str_green('{} m -f input.txt'.format(self_name)))
print('\n\n')
print('~~~~ !!!不会使用请仔细阅读 README.pdf !!!')
if __name__ == '__main__':
if datetime.now() > datetime(2022, 12, 31):
exit(0)
args = argv
try:
if len(args) == 1:
result = ip_integrity()
result.main()
print('\n>>>>>>>>>>\n按回车退出...\n')
input()
elif args[1] == 'compare' or args[1] == 'cp':
ip_compare(args[2], args[3])
elif args[1] == 'convert' or args[1] == 'cv':
print(ip_convert(args[2]))
elif args[1] == 'merge' or args[1] == 'm':
print(ip_merge(args[2]))
elif args[1] == 'help' or args[1] == 'h':
help(True)
else:
print('输入有误，请查看使用说明！')
except Exception as err:
print(err)
traceback.print_exc(file=open(DEBUG_FILE, 'w'))
pft(str_blue_green('出错啦。请反馈目录中的文件：', DEBUG_FILE))