首页 > 代码库 > python脚本简单检测ip合法性并添加到白名单文件
python脚本简单检测ip合法性并添加到白名单文件
一、功能说明
有时候项目需要通过ip地址来判定是否允许访问,通常通过一个白名单地址文件来存放这些允许放行的ip,但每次打开文件编辑比较麻烦,容易出错,也不知道是否添加过,故用python写了一个自动添加白名单的脚本。
二、脚本内容
#!/usr/bin/env python #coding:utf-8 #2017-06-09 ver:1.0 import sys white_list="white_list.txt" add_ip=sys.argv[1].strip() checkIp=add_ip.split(‘.‘) if len(checkIp)!= 4: print("ip[长度]不合法.程序退出!") sys.exit(5) elif not checkIp[0].isdigit() or not checkIp[1].isdigit() or not checkIp[2].isdigit() or not checkIp[3].isdigit(): print("ip[不是数字]不合法.程序退出!") sys.exit(5) elif int(checkIp[0]) >= 254 or int(checkIp[1]) > 255 or int(checkIp[2]) >255 or int(checkIp[3]) >255: print("ip[数字范围]不合法.程序退出!") sys.exit(5) new_ip = str(add_ip + "\n") f=open(white_list,‘r‘) IPS=f.readlines() if new_ip in IPS: print "The add ip %s is in white list." % new_ip.split() else: fw=open(white_list,‘a+‘) fw.write(new_ip) print "The add ip %s add in white list OK~" % new_ip.split() fw.close()
三、执行结果
$ ./add_white.py 192.168.1.256
ip[数字范围]不合法.程序退出!
$ ./add_white.py 256.0.0..
ip[长度]不合法.程序退出!
$ ./add_white.py afdafda
ip[长度]不合法.程序退出!
$ ./add_white.py a.a.a.a
ip[不是数字]不合法.程序退出!
$ ./add_white.py 192.1.1
ip[长度]不合法.程序退出!
$ ./add_white.py 192.168.1.2
The add ip [‘192.168.1.2‘] add in white list OK~
]$ ./add_white.py 192.168.1.2
The add ip [‘192.168.1.2‘] is in white list.
写的比较简单,有不当之处欢迎指正交流~
本文出自 “學地止境” 博客,请务必保留此出处http://dyc2005.blog.51cto.com/270872/1943298
python脚本简单检测ip合法性并添加到白名单文件
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。