python小实战之分配ip


一、背景介绍

    由于工作需要,管理着公司的全球机房ip,其他部门同事提取工单需要我这边协助分配一些ping不通的ip来使用。故而有时候会出现大工作量的情况。今日值班正好不巧,碰见同事需要分配16*64个ip,所以身为小菜的我花了1个小时写了的小程序以分配ip,特此记录以便后期回顾优化,也欢迎各位指点优化下。


二、涉及知识点

    os模块,使python可以与系统进行交互执行命令。

    os.system()    #执行系统命令并返回状态码

##系统帮助文档
>> help(os.system)
Help on built-in function system in module posix:

system(...)
    system(command) -> exit_status
    
    Execute the command (a string) in a subshell.


2.1 os.system()使用摸索

    os.system()具体使用如下:

##代码展示
ip_str = "ping -c 10 -i 0.2 -w 3  192.168.137.1 &> /dev/null"
In [38]: t = os.system(ip_str)

In [39]: print t 
0

##t = 0表示ip可以ping通,相反不为0则表示ping不通。
In [34]: if t == 0:
    ...:     print "True"
    ...: else :
    ...:     print t 
    ...:     
True

In [31]: if t == 0:
    ...:     print "True"
    ...: else :
    ...:     print t 
    ...:     
256


三、 ping.py小程序


3.1 ping.py的具体代码

[root@test home]# cat ping.py 
#/usr/local/python
#coding:utf-8
##定义ping的函数,参数有前三段ip段,第四段ip的起始,以及需要的ip个数
 

import os ##调用os.system()执行ping命令
def batchPing(head='192.168.137',start=0,end=12,num = 10):
	tmp = [] 
	#count = 0
	for i in range(start,end+1):
		#count += 1	
		#ip_str = 'ping -c 10' +head + '.' + str(i) 
		ip_str = "%s %s.%s %s" %('ping -c 10 -i 0.2 -w 3',head,str(i),'&> /home/ipresult.txt') 
		#print "di %s ci : %s" %(count,ip_str)
		##执行ping命令
		res = os.system(ip_str)
		if res  != 0:
			tmp.append(ip_str)
	return tmp[:num]

##将上述整理的ip写入到文件中去			
def operateFile(tmp=[]):
	with open('/home/useableIp.txt','w') as files:
		for i in tmp:
			#i = i.strip('&> /home/ipresult.txt')
			i = i.split(' ')[7]
			files.write('%s\n' %i)
	return "Done"
	
	
if __name__ == "__main__":
	linshi =  batchPing(head='112.90.148',start=237,end=254,num = 64)
	print operateFile(linshi)


3.2 分析代码:

  1.拼接字符串的过于冗长ip_str = "%s %s.%s %s" %('ping -c 10 -i 0.2 -w 3',head,str(i),'&> /home/ipresult.txt')。可以写成' '.join(列表)。列表名 = ['ping -c 10 -i 0.2 -w 3',head,str(i),'&> /home/ipresult.txt']。这样应该会美观很多。


  2.定义batchPing(head='192.168.137',start=0,end=12,num = 10)函数,是由于ip地址前三段的ip基本不变,仅最后一段更改,所有拆分成head和最后一位ip两段,遍历最后一个ip的开头start和结束end,在提取出自己想要的ip格式num.处理好后作为函数的返回值return回去。


  3.将return返回值写入文件/home/useableIp.txt中,以便记录。


  4.个人感觉参数的赋值(linshi =  batchPing(head='112.90.148',start=237,end=254,num = 64))也略有不雅,可以试试列表或字典的方式即*或者**。当然小程序的任务当然是完成工作,目前工作已经完成了,仅此记录下。便于他日回溯今日思路并优化这段小程序。


  5.其实程序bug多多,比如在ping ip环节处,本人是采取ping完后再获取自己想要的数目;若是能够一遍ping一遍累加等足够自己想要的数目就终止的话,这样程序的执行时间在某种程度上会有很大的缩减的




总结:

    学习要有定位,明确目标地去学习。学是为了更好的使用,提高自己的工作效率。for growing ---leaves