fabric实例——批量操作服务器

转自:http://blog.csdn.net/xiangliangyu2008/article/details/8235467

借助fabric模块写了个批量操作服务器的脚本:

准备工作:

安装python2.6.5:

yum -y install readline*

tar xf Python-2.6.5.tar.bz2

cd Python-2.6.5

将目录下Modules/Setup.dist文件中”readline readline.c-lreadline -ltermcap”行前的注释去掉

编译安装:

./configure –enable-shared

make -j8 && make install

安装setuptools

tar xf setuptools-0.6c11.tar.gz

cd setuptools-0.6c11

python setup.py install

安装fabric

执行安装时,软件会自动从网上查找依赖的安装包并进行安装

tar xf fabric-0.9rc2.tar.gz

cd goosemo-fabric-1eacbf2

python setup.py install


[python]  view plain copy
  1. #!/usr/bin/env python  
  2. # -*- coding: utf-8 -*-  
  3. #####################################################   
  4. # Version: 2.6.5   
  5. # Filename: python_pssh.py   
  6. #####################################################   
  7. from fabric.api import env,run,put,get  
  8. from os import path  
  9. from re import findall   
  10. from sys import argv  
  11. from fabric.context_managers import hide  
  12. from time import sleep  
  13. #USER 登录用户,默认root  
  14. #HOST 单个主机  
  15. #IP_LIST 多个主机所在文件  
  16. #PORT ssh端口,默认22  
  17. #PRI_KEY ssh私钥  
  18. #PASSWORD ssh密码  
  19. #CMD 远程操作命令  
  20. #uSRC 上传的文件  
  21. #uDST 上传目的服务器的目录路径  
  22. #dSRC 下载的文件  
  23. #dDST 下载本地的目录路径  
  24. #time 超时时间,默认1s  
  25. USER='root'  
  26. HOST,IP_LIST=[],[]  
  27. PORT='22'  
  28. PRI_KEY,PASSWORD,CMD,uSRC,uDST,dSRC,dDST= "","","","","","",""  
  29. timeout=1  
  30. for i in range(1,len(argv)+1):  
  31.     if argv[i-1] == '-h' or len(argv)==1:  
  32.             print """ 
  33.             USAGE: 
  34.        -u [user]       Use this argument to specify the user,default is 'root' 
  35.        -H [host]       The host that you want to connect 
  36.        -f [file]       The file content multiple ip address you want to connect 
  37.        -P [port]       The ssh port,default is 22 
  38.        -p [pwd|file]   You can specify password or a priviate key file to connect the host 
  39.        -c [command]    The command you want the host(s) to run 
  40.        -U [src,dst]    The local file that you want to upload to the remote host(s) 
  41.        -D [src,dst]    The remote file that you want to download to the local host 
  42.        -t [timeout]    The program running timeout,default is 1(s) 
  43.        -h              Print this help screen 
  44.             """  
  45.             #选项是第i-1个,那么值就是第i个  
  46.     if argv[i-1] == '-u':  
  47.             USER=argv[i]  
  48.             env.user='%s'%(USER)  
  49.     else:  
  50.             env.user='%s'%(USER)  
  51.     if argv[i-1] == '-H':  
  52.         #正则找出ip,存入HOST数组  
  53.         arg=findall('(\d+\.\d+\.\d+\.\d+|\s+\.{3,4})',argv[i])  
  54.         for j in arg:  
  55.             #判断是否是有类型,str  
  56.             if type(j).__name__ !='NoneType':  
  57.                 HOST.append(j)  
  58.             else:  
  59.                 print 'The HostIP input error'  
  60.                 sys.exit(1)    
  61.     if argv[i-1] == '-P':  
  62.         PORT=argv[i]  
  63.     if argv[i-1] == '-f':  
  64.         if path.isfile('%s'%(argv[i])) == True:  
  65.             IP_LIST=open('%s'%(argv[i]),'r').readlines()  
  66.     if argv[i-1] == '-p':  
  67.         if path.isfile(argv[i]) == True:  
  68.             PRI_KEY=argv[i]  
  69.             env.key_filename='%s'%(PRI_KEY)  
  70.         else:  
  71.             PASSWORD=argv[i]  
  72.             env.password='%s'%(PASSWORD)  
  73.     if argv[i-1] == '-c':  
  74.         CMD=argv[i]  
  75.     if argv[i-1]=='-t':                                                                                         
  76.         timeout=argv[i]    
  77.     SLP='sleep %s'%(timeout)  
  78.     if argv[i-1] == '-U':  
  79.         x=src=argv[i].split(',')  
  80.         uSRC=x[0]  
  81.         uDST=x[1]  
  82.     if argv[i-1] == '-D':  
  83.         y=src=argv[i].split(',')  
  84.         dSRC=y[0]  
  85.         dDST=y[1]             
  86. #   else:  
  87.     IP_PORT=[]  
  88.     if len(IP_LIST)!=0:  
  89.         for k in IP_LIST:  
  90.             IP_PORT.append(k.strip()+':'+PORT)  
  91.     if len(HOST)!=0:  
  92.         for k in HOST:  
  93.             IP_PORT.append(k.strip()+':'+PORT)  
  94.               
  95.               
  96. #执行CMD  
  97. if CMD != "":  
  98.     def command():  
  99.             with hide('running'):  
  100.                 run("%s;%s" %(CMD,SLP))  
  101.     for ip in IP_PORT:  
  102.         env.host_string=ip  
  103.         print "Execute command : \"%s\" at Host : %s" %(CMD,ip.split(':')[0])  
  104.         print "-----------------"  
  105.         command()  
  106.         print "-----------------"  
  107.   
  108. #上传  
  109. if uSRC and uDST != "":  
  110.     def upload():  
  111.         with hide('running'):  
  112.             put("%s" %(uSRC),"%s" %(uDST))  
  113.     for ip in IP_PORT:  
  114.             env.host_string=ip  
  115.             print "Upload local file : \"%s\" to Host : %s \"%s\"" %(uSRC,ip.split(':')[0],uDST)  
  116.             print "-----------------"  
  117.             upload()  
  118.             print "-----------------"  
  119.   
  120. #下载              
  121. if dSRC and dDST != "":  
  122.     def download():  
  123.         with hide('running'):  
  124.             get("%s" %(dSRC),"%s" %(dDST))  
  125.     for ip in IP_PORT:  
  126.         env.host_string=ip  
  127.             print "Download remote file : \"%s\" from Host : %s to local \"%s\"" %(dSRC,ip.split(':')[0],dDST)  
  128.             print "-----------------"  
  129.             download()  
  130.             print "-----------------"
相关文章
相关标签/搜索