python批量生成本地ip地址

这段代码用于在本地计算机上生成本地ip地址绑定到网卡，生成的是一个bat的批处理文件，运行此批处理文件，可以通过ipconfig查看

python">#!/usr/bin/python2.7
# -*- coding: utf-8 -*- 
# Filename: AddIPAliases.py
# Author: Ken, http://IJMeter.com/
# Notes：亲！尊重原创，转载请注明出处，谢谢您的合作！

import re,sys,socket,struct   

# 1. 判断IP地址是否合法； 2. 判断用户输入的IP是否在Class A，Class B 或 Class C中
def CheckIP(IP,IPClassesInt):
    regexIP=re.compile('^([01]?\d\d?|2[0-4]\d|25[0-5])\.([01]?\d\d?|2[0-4]\d|25[0-5])\.([01]?\d\d?|2[0-4]\d|25[0-5])\.([01]?\d\d?|2[0-4]\d|25[0-5])$')
    Checking=regexIP.match(IP)
    if Checking==None:
        IP=raw_input("亲！请输入合法的IP地址哈: ")
        return CheckIP(IP,IPClassesInt)
    else:
        IPInt=struct.unpack('!L',socket.inet_aton(IP))[0]
        if IPInt not in range(IPClassesInt[0],IPClassesInt[1]+1) and IPInt not in range(IPClassesInt[2],IPClassesInt[3]+1) and IPInt not in range(IPClassesInt[4],IPClassesInt[5]+1):
            IP=raw_input("亲！您输入的IP地址不在Class A，Class B或Class C中，请阅读提示信息并重新输入IP地址：")
            return CheckIP(IP,IPClassesInt)
        else: return IP

# 判断输入的数字是否合法
def CheckIPCount(IPCount):
    regexIPCount=re.compile('\d+')
    Checking=regexIPCount.match(IPCount)
    if Checking==None:
        IPCount=raw_input("亲！请输入合法的数量：")
        return CheckIPCount(IPCount)
    else: return IPCount

# 为用户生成指定数量的IP
def MakeIps(IPInt,IPCount,IPIntBool):
    targetBat=open('AddIPAliases.bat','w+')
    targetCSV=open('AddIPAliases.csv','w+')
    # 判断IP地址的数量是否在Class A，Class B或Class C的范畴中
    if int(IPCount)>(IPClassesInt[IPIntBoolTrue[1][2]]-IPInt+1):
        IPCount=raw_input("亲！您输入的IP数量超过了"+IPIntBoolTrue[0]+"的范畴，请重新输入：")
        return MakeIps(IPInt,IPCount,IPIntBool)
    else:
        for i in range(int(IPCount)):
            IPIntTrans=socket.inet_ntoa(struct.pack("!L", IPInt))
            IPInt+=1
            targetBat.write('netsh interface ip add address "本地连接" '+IPIntTrans+' '+IPIntBoolTrue[1][1]+'\n')
            targetCSV.write(IPIntTrans+'\n')
    targetBat.write('pause')

# Main 函数
print '''
**************************************************
以下信息能够帮助你更好的运行此脚本：

1. 在运行脚本前，请移步：控制面板->查看网络->本地连接->属性->IPv4，将自动获取IP改成手动，如上图所示http://IJMeter.com

2. IP分为三类：
   Class A：数量16777216，范围10.0.0.0 - 10.255.255.255
   Class B：数量1048576， 范围172.16.0.0 - 172.31.255.255
   Class C：数量65536，   范围192.168.0.0 – 192.168.255.255
!!!所以大家在输入IP的时候，请保证您输入的IP属于这三个分类之中!!!

3. 如果你使用的是英文系统，请将自定义函数MakeIPs()中的“本地连接”改成“Local Area Connection”。

4. 脚本由于要对长整型的数据进行range()，所以计算时间有点长，请等待成功提示。

5. 在脚本存放的目录将会生成一个bat文件和csv文件，bat文件用于向系统中添加IP，生成成功后请手动运行它，csv文件用于在Jmeter中调用这些IP。

6. 如果想清除在系统中插入的IP，请移步：控制面板->查看网络->本地连接->属性->IPv4，将手动获取IP改成自动获取IP
**************************************************
'''

# 将各个Class的起始和结束的IP地址转换成整数
IPClasses=['10.0.0.0','10.255.255.255','172.16.0.0','172.31.255.255','192.168.0.0','192.168.255.255']
IPClassesInt=[]
for i in range(len(IPClasses)):
    IPClassesInt.append(struct.unpack('!L',socket.inet_aton(IPClasses[i]))[0])

# 用户输入
IP=raw_input("请输入起始IP地址：")
IPCount=raw_input("请输入生成的IP数量：")

# 判断输入是否合法
IPAddress=CheckIP(IP,IPClassesInt)
IPCount=CheckIPCount(IPCount)

IPInt=struct.unpack('!L',socket.inet_aton(IPAddress))[0]
IPIntClassABool=IPInt in range(IPClassesInt[0],IPClassesInt[1]+1)
IPIntClassBBool=IPInt in range(IPClassesInt[2],IPClassesInt[3]+1)
IPIntClassCBool=IPInt in range(IPClassesInt[4],IPClassesInt[5]+1)

IPIntBool={"ClassA":[IPIntClassABool,'255.0.0.0',1],"ClassB":[IPIntClassBBool,'255.240.0.0',3],"ClassC":[IPIntClassCBool,'255.255.0.0',5]}
IPIntBoolTrue=[]

for i in range (len(IPIntBool)):
    if True in IPIntBool.values()[i]:
        IPIntBoolTrue.append(IPIntBool.keys()[i])
        IPIntBoolTrue.append(IPIntBool.values()[i])
        break

# 调用函数为用户生成IP地址
MakeIps(IPInt,IPCount,IPIntBoolTrue)

print "Bat文件生成完毕，请移步至存放的脚本的文件夹找到并运行此文件。"

标签：python