python中getopt模块
在日常的工作中免不了需要编写很多python脚本,当脚本的功能比较多的时候,又或者需要外部传入参数的时候,如果以参数
名和参数值的方式执行可能脚本显得更直观,也给自己提供方便。
python下有一个getopt的模块,该模块就是处理命令行参数的。
函数getopt(args,shortopts,longopts = [])
args一般是sys.argv[1:]
shortopys 短格式(-)
longopts 长格式(–) //两个杠
执行方法:
python run_cmd.py -f iplist.txt -c ifconfig
python run_cmd.py –file iplist.txt –command ifconfig
代码如下:
1 #!/usr/bin/env python 2 # -*- coding: utf-8 -*- 3 4 import ConfigParser 5 import paramiko 6 import sys 7 import os 8 import datetime 9 from multiprocessing import Process 10 11 class remote_connection: 12 13 def __init__(self,files,cmd): 14 self.ini = 'iplist.ini' 15 self.allfile = [] 16 17 # 读取INI配置文件 18 def getini(self): 19 if os.path.isfile(self.ini): 20 pass 21 else: 22 sys.exit('Check your %s exists' % self.ini) 23 cfg = ConfigParser.ConfigParser() 24 cfg.read(self.ini) 25 sec = ''.join(cfg.sections()) 26 options = cfg.options(sec) 27 self.user = cfg.get(sec,options[0]) 28 self.passwd = cfg.get(sec,options[1]) 29 self.host_all = cfg.get(sec,options[2]) 30 self.host = ''.join(self.host_all).split(',') 31 32 # 执行远程ssh命令 33 def sshcmd(self,host='',port='',username='',passwd=''): 34 s=paramiko.SSHClient() 35 s.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 36 try: 37 s.connect(host,port,username,passwd,timeout=5) 38 except Exception: 39 sys.exit(host+'\x1b[1;31m\t\t[连接失败]\x1b[0m') 40 chan = s.get_transport().open_session() 41 chan.exec_command(cmd) 42 status = chan.recv_exit_status() 43 if status == 0: 44 print host+'\x1b[1;32m\t\t[执行成功]\x1b[0m' 45 else: 46 print host+'\x1b[1;31m\t\t[执行失败]\x1b[0m' 47 48 if __name__ == '__main__': 49 try: 50 opts,args = opts,args = getopt.getopt(sys.argv[1:],"(hH)f:c:",["help","file=","command="]) 51 if len(sys.argv) == 1: 52 print '用法: python xxx.py -f xxx.file -c ifconfig' 53 sys.exit() 54 if sys.argv[1] in ("-h","-H","--help"): 55 print '用法: python xxx.py -f xxx.file -c ifconfig' 56 elif sys.argv[1] in ("-f","--file"): 57 for opt,arg in opts: 58 if opt in ("-f","--file"): 59 files = arg 60 if opt in ("-c","--command"): 61 cmd = arg 62 63 box = [] 64 p = remote_connection(sys.argv[1],sys.argv[2]) 65 p.getini() 66 for i in p.host: 67 box.append(Process(target=p.sshcmd,args=(i,3389,p.user,p.passwd))) 68 for i in box: 69 i.start() 70 i.join() 71 except Exception,e: 72 print e
