一、功能說明

有時候項目需要通過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

本文出自 “學地止境” 博客，請務必保留此出處http://dyc2005.blog.51cto.com/270872/1943298

python腳本簡單檢測ip合法性並添加到白名單文件