1 import paramiko
 2 import sys
 3
 4
 5 user = "root"
 6 pwd = "123456"
 7
 8
 9
10 # 上传文件
11 def sftp_upload_file(server_path, local_path):
12     try:
13         t = paramiko.Transport((ip, 22))
14         t.connect(username=user, password=pwd)
15         sftp = paramiko.SFTPClient.from_transport(t)
16         sftp.put(local_path, server_path)
17         t.close()
18     except Exception as  e:
19         print(e)
20
21 # 下载文件
22 def sftp_down_file(server_path, local_path):
23     try:
24         t = paramiko.Transport((ip, 22))
25         t.connect(username=user, password=pwd)
26         sftp = paramiko.SFTPClient.from_transport(t)
27         sftp.get(server_path, local_path)
28         t.close()
29     except Exception as e:
30         print(e)
31
32 # 连接
33 def ssh_conn(ip, cmd):
34
35     ssh = paramiko.SSHClient()
36     # 允许连接不在known_hosts文件上的主机
37     ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
38     # 连接服务器
39     ssh.connect(ip, 22, user, pwd)
40     # 执行命令
41     stdin, stdout, stderr = ssh.exec_command(cmd)
42     # 获取结果
43     print(10 * "-", 'start', 10 * "-")
44     for line in stdout:
45         res=(line.strip('\n').split())
46         print(res)
47     else:
48         print(stdout)
49     print(10 * "-", 'end', 10 * "-")
50
51 def menu():
52     print('''
53     * - - - - - - - - - - - - - - - - - *
54                    菜单
55                 1>上传文件
56                 2>下载文件
57                 3>执行命令
58                 4>退出
59     * - - - - - - - - - - - - - - - - - *
60     ''')
61
62     choice = int(input('请输入你要执行的操作：\n'))
63     if choice == 1:
64         src = input('请输入原路径：\n')
65         dest = input('请输入目标路径：\n')
66         sftp_upload_file(src, dest)
67     elif choice == 2:
68         src = input('请输入原路径：\n')
69         dest = input('请输入目标路径：\n')
70         sftp_down_file(src, dest)
71     elif choice == 3:
72         while True:
73             cmd = input('请输入要执行的命令：\n')
74             if cmd == 'eixt':
75                 sys.exit()
76             ssh_conn(ip, cmd)
77     else:
78         sys.exit()
79
80
81 if __name__ == '__main__':
82     ip = input('请输入目标ip：\n')
83     while True:
84         menu()