最近写运维自动化平台,需要用python写很多的小功能模块。
这里就分享一个用Python的paramiko来实现功能的一段代码:
复制远程服务器上的文件及文件夹到本地目录。
解释一下什么叫paramiko:
paramiko是用python写的一个模块,遵循SSH2协议,支持以加密和认证的方式,进行远程服务器的连接。利用该模块,可以方便的进行ssh连接和sftp协议进行sftp文件传输以及远程命令执行。
安装paramiko也很简单,我用的是python3,装好了pip,可以直接用pip来安装。
不过由于 paramiko 模块内部依赖pycrypto,所以先下载安装pycrypto:
pip3 install pycrypto
pip3 install paramiko
好了。我定义一个方法,用于远程复制文件。代码如下:
import os
import paramiko
def RemoteScp(host_ip, host_port, host_username, host_password, remote_file, local_file):
scp = paramiko.Transport((host_ip, host_port))
scp.connect(username=host_username, password=host_password)
sftp = paramiko.SFTPClient.from_transport(scp)
sftp.get(remote_file, local_file)
scp.close()
return ("success")
注意这里的语句:
sftp.get(remote_file, local_file)中自带的get方法,只能拷贝文件目录,不能拷贝文件夹。
然后调用这个方法:
if __name__ == '__main__':
host_ip = '192.168.1.123'
host_port = 22
host_username = 'root'
host_password = 'password'
remote_path = '/tmp/test.sql'
local_path = '/tmp/test.sql'
RemoteScp(host_ip, host_port, host_username, host_password, remote_path, local_path)
这就是基本远程文件复制的功能实现。这里只能实现文件对文件的远程复制。
如果我们要远程复制整个文件夹。需要对上面的方法改进一下,加一段循环,让其能遍历远程目录里的所有文件,然后按照上面的方法依次复制到本地。
我在代码中加了一些注释。修改以后的完整代码如下:
import os
import paramiko
def RemoteScp(host_ip, host_port, host_username, host_password, remote_path, local_path):
scp = paramiko.Transport((host_ip, host_port))
scp.connect(username=host_username, password=host_password)
sftp = paramiko.SFTPClient.from_transport(scp)
try:
remote_files = sftp.listdir(remote_path)
for file in remote_files: #遍历读取远程目录里的所有文件
local_file = local_path + file
remote_file = remote_path + file
sftp.get(remote_file, local_file)
except IOError: # 如果目录不存在则抛出异常
return ("remote_path or local_path is not exist")
scp.close()
if __name__ == '__main__':
host_ip = '192.168.1.123' # 远程服务器IP
host_port = 22 # 远程服务器端口
host_username = 'root' #远程服务器用户名
host_password = 'password' #远程服务器密码
remote_path = '/tmp/' #这个是远程目录
local_path = '/tmp/' #这个是本地目录
RemoteScp(host_ip, host_port, host_username, host_password, remote_path, local_path) #调用方法