【python】用python脚本实现ansible的推送、下载、上传文件功能

该脚本可以实现：上传文件、下载文件、执行命令


If you want perform command:
    Usage: (for one host) ./paramiko-upload.py -H [hostip] -c [command]
    Example: ./paramiko-upload.py -H 10.100.139.245 -c 'ls -al'


    Usage: (for a host group) ./paramiko-upload.py -F [ip.txt] -c [command]
    Example: ./paramiko-upload.py -F 'ip.txt' -c 'ls -al'


If you want send file:


    Usage: (for one host) ./paramiko-upload.py -H [hostip] -s [local_dir] [client_dir] [file]
    Example: ./paramiko-upload.py -H 10.100.139.245 -s '/opt/Felix' '/tmp' ip.txt


    Usage: (for a host group) ./paramiko-upload.py -F [ip.txt] -s [local_dir] [client_dir] [file]
    Example: ./paramiko-upload.py -F ip.txt -s '/opt/Felix' '/tmp' ip.txt


If you want get file:
    Usage: (for one host) ./paramiko-upload.py -H [hostip] -g [local_dir] [client_dir] [file]
    Example: ./paramiko-upload.py -H 10.100.139.245 -g '/opt/Felix' '/tmp' ip.txt


ip.txt file content like this:
    1.1.1.1
    2.2.2.2\033[0m
    """
    sys.exit()


脚本如下：

1. #!/usr/bin/env python
   
2. #定义函数
   
3. import paramiko
   
4. import os
   
5. import datetime
   
6. import sys
   
7. import re
   
8. #定义基础命令格式
   
9. def perform_command(hostname, port, username, password, comm):
   
10.     try:
    
11.         t=paramiko.Transport((hostname,port))
    
12.         t.connect(username=username,password=password)
    
13.         try:
    
14.             print "\033[33mIP:\033[0m", hostname
    
15.             print "\033[33mPerform command:\033[0m", comm
    
16.             print '\033[32m###########################################################\033[0m'
    
17.             s=paramiko.SSHClient()
    
18.             s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    
19.             s.connect(hostname = hostname, port = port, username = username, password = password)
    
20.             stdin,stdout,stderr=s.exec_command(comm)
    
21.             print stdout.read()
    
22.             print '\033[32m###########################################################\033[0m'
    
23.             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
    
24.             s.close()
    
25.         except Exception:
    
26.             print "\033[31mERROR - Perform error\033[0m"
    
27.     except Exception:
    
28.         print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname
    
29. 
    
30. #定义发送文件格式
    
31. def send_file(hostname,port,username,password,local_dir,client_dir,file):
    
32.     try:
    
33.         t=paramiko.Transport((hostname,port))
    
34.         t.connect(username=username,password=password)
    
35.         try:
    
36.             print "\033[33mIP:\033[0m", hostname
    
37.             print "\033[33mSend file:\033[0m", file
    
38.             print "\033[33mClient_dir:\033[0m", client_dir
    
39.             sftp=paramiko.SFTPClient.from_transport(t)
    
40.             sftp.put(os.path.join(local_dir,file),os.path.join(client_dir,file))
    
41.             t.close()
    
42.             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
    
43.         except Exception:
    
44.             print "\033[31mERROR - Perform error\033[0m"
    
45.     except Exception:
    
46.             print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname
    
47. #定义下载文件格式
    
48. def get_file(hostname,port,username,password,local_dir,client_dir,file):
    
49.     try:
    
50.         t=paramiko.Transport((hostname,port))
    
51.         t.connect(username=username,password=password)
    
52.         try:
    
53.             print "\033[33mIP:\033[0m", hostname
    
54.             print "\033[33mGet file:\033[0m", file
    
55.             print "\033[33mClient_dir:\033[0m", client_dir
    
56.             sftp=paramiko.SFTPClient.from_transport(t)
    
57.             sftp.get(os.path.join(client_dir,file),os.path.join(local_dir,file))
    
58.             t.close()
    
59.             print '\033[32mScript perform success %s \033[0m' % datetime.datetime.now()
    
60.         except Exception:
    
61.             print "\033[31mERROR - Perform error\033[0m"
    
62.     except Exception:
    
63.             print "\033[31mERROR - Connect error! IP: %s!\033[0m" % hostname
    
64. #定义帮助文件
    
65. def help():
    
66.     print """
    
67. \033[33mHelp:
    
68. 
    
69.   Welcome to use  python script, this script can help you run commands, send files and get files.
    
70. 
    
71. Use this script, you can running for one host or a host group, and you know get file can't for a host group.
    
72. 
    
73. Give you examples:
    
74. 
    
75. If you want perform command:
    
76.     Usage: (for one host) ./paramiko-upload.py -H [hostip] -c [command]
    
77.     Example: ./paramiko-upload.py -H 10.100.139.245 -c 'ls -al'
    
78. 
    
79.     Usage: (for a host group) ./paramiko-upload.py -F [ip.txt] -c [command]
    
80.     Example: ./paramiko-upload.py -F 'ip.txt' -c 'ls -al'
    
81. 
    
82. If you want send file:
    
83. 
    
84.     Usage: (for one host) ./paramiko-upload.py -H [hostip] -s [local_dir] [client_dir] [file]
    
85.     Example: ./paramiko-upload.py -H 10.100.139.245 -s '/opt/Felix' '/tmp' ip.txt
    
86. 
    
87.     Usage: (for a host group) ./paramiko-upload.py -F [ip.txt] -s [local_dir] [client_dir] [file]
    
88.     Example: ./paramiko-upload.py -F ip.txt -s '/opt/Felix' '/tmp' ip.txt
    
89. 
    
90. If you want get file:
    
91.     Usage: (for one host) ./paramiko-upload.py -H [hostip] -g [local_dir] [client_dir] [file]
    
92.     Example: ./paramiko-upload.py -H 10.100.139.245 -g '/opt/Felix' '/tmp' ip.txt
    
93. 
    
94. ip.txt file content like this:
    
95.     1.1.1.1
    
96.     2.2.2.2\033[0m
    
97.     """
    
98.     sys.exit()
    
99. 
    
100. #定义命令详情
     
101. 
     
102. if __name__ == '__main__':
     
103. #定义账号、端口、密码等信息
     
104.     username='root'
     
105.     port=2222
     
106.     password='abcdefg'
     
107. 
     
108. 
     
109.     if len(sys.argv) > 2:
     
110.         if sys.argv[1] == '-H':
     
111.             if len(sys.argv) == 5 and sys.argv[3] == '-c':
     
112.                 hostname=sys.argv[2]
     
113.                 comm = sys.argv[4]
     
114.                 perform_command(hostname, port, username, password, comm)
     
115.             elif len(sys.argv) == 7 and sys.argv[3] == '-s':
     
116.                 hostname=sys.argv[2]
     
117.                 local_dir=sys.argv[4]
     
118.                 client_dir=sys.argv[5]
     
119.                 file=sys.argv[6]
     
120.                 send_file(hostname,port,username,password,local_dir,client_dir,file)
     
121.             elif len(sys.argv) == 7 and sys.argv[3] == '-g':
     
122.                 hostname=sys.argv[2]
     
123.     username='root'
     
124.     port=2222
     
125.     password='abcdefg'
     
126. 
     
127. 
     
128.     if len(sys.argv) > 2:
     
129.         if sys.argv[1] == '-H':
     
130.             if len(sys.argv) == 5 and sys.argv[3] == '-c':
     
131.                 hostname=sys.argv[2]
     
132.                 comm = sys.argv[4]
     
133.                 perform_command(hostname, port, username, password, comm)
     
134.             elif len(sys.argv) == 7 and sys.argv[3] == '-s':
     
135.                 hostname=sys.argv[2]
     
136.                 local_dir=sys.argv[4]
     
137.                 client_dir=sys.argv[5]
     
138.                 file=sys.argv[6]
     
139.                 send_file(hostname,port,username,password,local_dir,client_dir,file)
     
140.             elif len(sys.argv) == 7 and sys.argv[3] == '-g':
     
141.                 hostname=sys.argv[2]
     
142.                 local_dir=sys.argv[4]
     
143.                 client_dir=sys.argv[5]
     
144.                 file=sys.argv[6]
     
145.                 get_file(hostname,port,username,password,local_dir,client_dir,file)
     
146.             else:
     
147.                 help()
     
148.         elif sys.argv[1] == '-F':
     
149.             ip_file=sys.argv[2]
     
150.             file_stat=os.path.exists(ip_file)
     
151.             if file_stat:
     
152.                 f=open(ip_file)
     
153.                 ips=f.readlines()
     
154.                 f.close()
     
155.                 for ip in ips:
     
156.                     ip = ip.strip('\n')
     
157.                     if re.match(r'^#', ip):
     
158.                         continue
     
159.                     if len(sys.argv) == 5 and sys.argv[3] == '-c':
     
160.                         hostname = ip
     
161.                         comm = sys.argv[4]
     
162.                         perform_command(hostname, port, username, password, comm)
     
163.                     elif len(sys.argv) == 7 and sys.argv[3] == '-s':
     
164.                         hostname=ip
     
165.                         local_dir=sys.argv[4]
     
166.                         client_dir=sys.argv[5]
     
167.                         file=sys.argv[6]
     
168.                         send_file(hostname,port,username,password,local_dir,client_dir,file)
     
169.                     else:
     
170.                         help()
     
171.             else:
     
172.                 print "\033[31mERROR - Not found ip_file!\033[0m"
     
173.                 help()
     
174.         else:
     
175.             help()
     
176.     else:
     
177.         help()

来自 “ ITPUB博客 ” ，链接：http://blog.itpub.net/31442725/viewspace-2148711/，如需转载，请注明出处，否则将追究法律责任。

转载于:http://blog.itpub.net/31442725/viewspace-2148711/