在Python中可以使用paramiko模块中的sftp登陆远程主机,实现上传和下载功能。
1.功能实现
根据输入参数判断是文件还是目录,进行上传和下载
本地参数local需要与远程参数remote类型一致,文件以文件名结尾,目录以\结尾
上传和下载的本地和远程目录需要存在
异常捕获
2.代码实现
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
#!/usr/bin/python
# coding=utf-8
import paramiko
import os
def sftp_upload(host,port,username,password,local,remote):
sf = paramiko.Transport((host,port))
sf.connect(username = username,password = password)
sftp = paramiko.SFTPClient.from_transport(sf)
try :
if os.path.isdir(local): #判断本地参数是目录还是文件
for f in os.listdir(local): #遍历本地目录
sftp.put(os.path.join(local + f),os.path.join(remote + f)) #上传目录中的文件
else :
sftp.put(local,remote) #上传文件
except Exception,e:
print ( 'upload exception:' ,e)
sf.close()
def sftp_download(host,port,username,password,local,remote):
sf = paramiko.Transport((host,port))
sf.connect(username = username,password = password)
sftp = paramiko.SFTPClient.from_transport(sf)
try :
if os.path.isdir(local): #判断本地参数是目录还是文件
for f in sftp.listdir(remote): #遍历远程目录
sftp.get(os.path.join(remote + f),os.path.join(local + f)) #下载目录中文件
else :
sftp.get(remote,local) #下载文件
except Exception,e:
print ( 'download exception:' ,e)
sf.close()
if __name__ = = '__main__' :
host = '192.168.1.2' #主机
port = 22 #端口
username = 'root' #用户名
password = '123456' #密码
local = 'F:\\sftptest\\' #本地文件或目录,与远程一致,当前为windows目录格式,window目录中间需要使用双斜线
remote = '/opt/tianpy5/python/test/' #远程文件或目录,与本地一致,当前为linux目录格式
sftp_upload(host,port,username,password,local,remote) #上传
#sftp_download(host,port,username,password,local,remote)#下载
|
3.总结
以上代码实现了文件和目录的上传和下载,可以单独上传和下载文件,也可以批量上传和下载目录中的文件,基本实现了所要的功能,但是针对目录不存在的情况,以及上传和下载到多台主机上的情况,还有待完善。
原文链接:http://blog.csdn.net/tianpy5/article/details/52507324