脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Python - Python远程连接服务器用它就够了

Python远程连接服务器用它就够了

2021-12-06 23:44Python大数据分析费弗里 Python

本文我就将带大家学习在Python中使用非常强大的fabric库来对常用的远程服务器连接管理操作进行自动化。

简介

日常工作中经常需要通过SSH连接到多台远程服务器来完成各种任务,当需要操作的服务器众多,且要执行的任务涉及命令繁多时,如果可以以自动化的方式模拟SSH连接及执行命令的繁琐过程,对工作效率的提升是非常可观的。

本文我就将带大家学习在Python中使用非常强大的fabric库来对常用的远程服务器连接管理操作进行自动化:

Python远程连接服务器用它就够了

一、使用fabric玩转远程服务器管理

fabric基于强大的paramiko、invoke等库,构建出一整套简单易用的API,使得我们使用简洁的语句就可以应付常见的各种远程服务器操作,使用pip install fabric完成安装,本文演示对应版本为2.6。

1. 连接到远程服务器并执行终端命令

要建立并保持对远程服务器的连接,我们需要对fabric.Connection()进行实例化,其基础参数有:

  • 「host:」 str型,格式如'用户名@host:端口',其中':端口'部分可以省略,默认端口为22
  • 「connect_kwargs:」 dict型,用于传入其他连接所需参数,常用的有:

其他可用参数你可以前往

https://docs.paramiko.org/en/latest/api/client.html#paramiko.client.SSHClient.connect的Parameters了解更多。

  • 「password:」 str型,当以密码方式连接时,用于传入与host参数对应的密码
  • 「pkey:」 str型,当以私钥方式连接时,用于设置对应密钥
  • 「timeout:」 float型,用于设置TCP连接的超时时长(单位:秒)
  • 「config:」 fabric.Config型,用于设置更多复杂功能参数,详见https://docs.fabfile.org/en/2.6/api/config.html

先来看一个最基础的例子:

  1. from fabric import Connection
  2. # 建议将ssh连接所需参数变量化
  3. user = '用户名'
  4. host = 'host地址'
  5. password = '密码'
  6. # 利用fabric.Connection快捷创建连接
  7. c = Connection(host=f'{user}@{host}',
  8. connect_kwargs=dict(
  9. passwordpassword=password
  10. ))
  11. # 利用run方法直接执行传入的命令
  12. c.run('pwd');

Python远程连接服务器用它就够了

可以看到,非常简单就完成了连接服务器及执行指定命令的过程,且run()方法所执行的命令打印出的结果,可以通过stdout属性进行保存:

  1. # hide=True抑制run()过程对执行结果的自动打印
  2. output = c.run('df -h', hide=True).stdout
  3. print(output)

Python远程连接服务器用它就够了

获悉了fabric的基础使用后,接下来我们来学习一些进阶内容:

2. 配置sudo命令密码自动预填入

我们都知道,对于非root用户,在执行某些权限较高的命令时需要添加sudo前缀,并在会话的初次执行时需要手动输入当前用户的密码,而在fabric中,有两种方式可以实现这个步骤的自动化:

(1) 方式1:配合invoke.Responder

我们可以配合invoke.Responder来实现当命令行返回密码输入提示时,自动输入并执行指定的命令:

  1. from invoke import Responder
  2. # 配置命令行内容监听规则
  3. sudopass = Responder(
  4. pattern=f'\[sudo\] password for {user}:',
  5. response=password+'\n'
  6. )
  7. # 注意需要设置pty=True
  8. c.run('sudo pwd', pty=True, watchers=[sudopass]);

Python远程连接服务器用它就够了

(2) 方式2:利用fabric.Config设置sudo密码

除了上一种方式外,我们还可以使用fabric.Config在创建连接时就一次性提前配置好sudo密码,之后需要执行sudo命令时用sudo()方法代替run()方法即可:

  1. from fabric import Config
  2. # 预先配置sudo密码
  3. config = Config(overrides={
  4. 'sudo': {
  5. 'password': password
  6. }
  7. })
  8. c = Connection(host=f'{user}@{host}',
  9. connect_kwargs={'password': password},
  10. configconfig=config)
  11. c.sudo('pwd');

Python远程连接服务器用它就够了

3. 远程文件传输

很多朋友都知道可以使用pscp、xshell之类的工具手动进行服务器与本地之间的文件相互传输,这些任务我们同样可以在fabric中自动化进行:

(1) 从本地上传文件到服务器

使用put()方法可以将指定的本地文件上传至服务器的指定位置,remote参数对应服务器目标保存位置:

  1. c = Connection(host=f'{user}@{host}',
  2. connect_kwargs={'password': password})
  3. # 创建示例文件
  4. with open('file_transfer.txt', 'w') as d:
  5. d.write('1')
  6. # 利用put方法上传至服务器
  7. c.put('file_transfer.txt', remote='/home/feffery/')
  8. # 打印已上传文件内容
  9. c.run('cat /home/feffery/file_transfer.txt');

Python远程连接服务器用它就够了

(2) 从服务器下载指定文件到本地

相反的,当我们需要从服务器取回指定文件到本地时,就可以使用get()方法:

  1. c = Connection(host=f'{user}@{host}',
  2. connect_kwargs={'password': password})
  3. # 向文件末尾追加行
  4. c.run('echo "\n2" >> file_transfer.txt')
  5. c.get('/home/feffery/file_transfer.txt')
  6. print(open('file_transfer.txt').read())

Python远程连接服务器用它就够了

原文链接:https://mp.weixin.qq.com/s?__biz=MzA3ODYwNDkzOQ==&mid=2659068475&idx=1&sn=06f2bfe73a0708a5e7cc930dc7185f30&chksm=84cab5dfb3bd3cc94579ae27894fd782084c383ccd4e802c3d369973a58f54e7d17f8560ee79&mpshare=1&s

延伸 · 阅读

精彩推荐