1 #!/usr/bin/env python
2 # -*- coding: UTF-8 -*-
3
4 import paramiko
5 import os
6 import re
7
8
9 class SSHConnection(object):
10 def __init__(self, host, port, username, password):
11 self.host = host
12 self.port = port
13 self.username = username
14 self.password = password
15 self.__k = None
16
17 def run(self):
18 self.connect()
19 self.sftp_put_dir("/opt/app/appops/sbin/collectd/var", "/opt/app/")
20 # self.cmd('ls /opt/app')
21 self.close()
22
23 def connect(self):
24 transport = paramiko.Transport((self.host, self.port))
25 transport.connect(username=self.username, password=self.password)
26 self.__transport = transport
27
28 def close(self):
29 self.__transport.close()
30
31 def upload(self, loacal_path, target_path):
32 sftp = paramiko.SFTPClient.from_transport(self.__transport)
33 sftp.put(loacal_path, target_path)
34
35 def cmd(self, command):
36 print "commadn in cmd:", command
37 ssh = paramiko.SSHClient()
38
39 ssh._transport = self.__transport
40 stdin, stdout, stderr = ssh.exec_command(command)
41
42 result = stdout.read()
43 print "result:\n", result
44 return result
45
46 def getList(self, path):
47 files_list = []
48 files = sorted(filter(lambda x: re.match(r'^\w', x), os.listdir(path)))
49 for tmp_file in files:
50 path_file = os.path.join(path, tmp_file)
51 if os.path.isdir(path_file) and not os.path.islink(path_file):
52 files_list.append(path_file)
53 files_list += self.getList(path_file)
54 else:
55 files_list.append(path_file)
56 return files_list
57
58 def sftp_put_dir(self, local_dir, remote_dir):
59 sftp = paramiko.SFTPClient.from_transport(self.__transport)
60 if remote_dir[-1] == '/':
61 remote_dir = remote_dir[0:-1]
62 all_files = self.getList(local_dir)
63 for x in all_files:
64 f_path = x.replace(local_dir, "").lstrip('/')
65 if os.path.isdir(x):
66 r_path = os.path.join(remote_dir, f_path)
67 try:
68 sftp.mkdir(r_path)
69 except Exception,e:
70 pass
71 else:
72 remote_filename = os.path.join(remote_dir, f_path)
73 print u'Put文件%s传输中...' % f_path
74 sftp.put(x, remote_filename)
75
76
77 def main(host, port, username, password):
78 print "in main:"
79 obj = SSHConnection(host, port, username, password)
80 obj.run()
81
82
83 if __name__ == '__main__':
84 host = ''
85 port = ''
86 username = ''
87 password = ''
88 main(host, port, username, password)