（數據科學學習手札132）Python+Fabric實現遠程伺服器連接
本文示例程式碼及文件已上傳至我的
Github倉庫//github.com/CNFeffery/DataScienceStudyNotes
1 簡介
日常工作中經常需要通過SSH連接到多台遠程伺服器來完成各種任務，當需要操作的伺服器眾多，且要執行的任務涉及命令繁多時，如果可以以自動化的方式模擬SSH連接及執行命令的繁瑣過程，對工作效率的提升是非常可觀的。
本文我就將帶大家學習在Python中使用非常強大的fabric庫來對常用的遠程伺服器連接管理操作進行自動化😋：
2 使用fabric玩轉遠程伺服器管理
fabric基於強大的paramiko、invoke等庫，構建出一整套簡單易用的API，使得我們使用簡潔的語句就可以應付常見的各種遠程伺服器操作，使用pip install fabric完成安裝，本文演示對應版本為2.6。
2.1 連接到遠程伺服器並執行終端命令
要建立並保持對遠程伺服器的連接，我們需要對fabric.Connection()進行實例化，其基礎參數有：
-
host： str型，格式如
'用戶名@host:埠'，其中':埠'部分可以省略，默認埠為22 -
connect_kwargs： dict型，用於傳入其他連接所需參數，常用的有：
- password： str型，當以密碼方式連接時，用於傳入與
host參數對應的密碼 - pkey： str型，當以私鑰方式連接時，用於設置對應密鑰
- timeout： float型，用於設置
TCP連接的超時時長（單位：秒）
其他可用參數你可以前往
//docs.paramiko.org/en/latest/api/client.html#paramiko.client.SSHClient.connect的Parameters了解更多 - password： str型，當以密碼方式連接時，用於傳入與
-
config： fabric.Config型，用於設置更多複雜功能參數，詳見
//docs.fabfile.org/en/2.6/api/config.html
先來看一個最基礎的例子：
from fabric import Connection
# 建議將ssh連接所需參數變數化
user = '用戶名'
host = 'host地址'
password = '密碼'
# 利用fabric.Connection快捷創建連接
c = Connection(host=f'{user}@{host}',
connect_kwargs=dict(
password=password
))
# 利用run方法直接執行傳入的命令
c.run('pwd');
可以看到，非常簡單就完成了連接伺服器及執行指定命令的過程，且run()方法所執行的命令列印出的結果，可以通過stdout屬性進行保存：
# hide=True抑制run()過程對執行結果的自動列印
output = c.run('df -h', hide=True).stdout
print(output)
獲悉了fabric的基礎使用後，接下來我們來學習一些進階內容：
2.2 配置sudo命令密碼自動預填入
我們都知道，對於非root用戶，在執行某些許可權較高的命令時需要添加sudo前綴，並在會話的初次執行時需要手動輸入當前用戶的密碼，而在fabric中，有兩種方式可以實現這個步驟的自動化：
2.2.1 方式1：配合invoke.Responder
我們可以配合invoke.Responder來實現當命令行返回密碼輸入提示時，自動輸入並執行指定的命令：
from invoke import Responder
# 配置命令行內容監聽規則
sudopass = Responder(
pattern=f'\[sudo\] password for {user}:',
response=password+'\n'
)
# 注意需要設置pty=True
c.run('sudo pwd', pty=True, watchers=[sudopass]);
2.2.2 方式2：利用fabric.Config設置sudo密碼
除了上一種方式外，我們還可以使用fabric.Config在創建連接時就一次性提前配置好sudo密碼，之後需要執行sudo命令時用sudo()方法代替run()方法即可：
from fabric import Config
# 預先配置sudo密碼
config = Config(overrides={
'sudo': {
'password': password
}
})
c = Connection(host=f'{user}@{host}',
connect_kwargs={'password': password},
config=config)
c.sudo('pwd');
2.3 遠程文件傳輸
很多朋友都知道可以使用pscp、xshell之類的工具手動進行伺服器與本地之間的文件相互傳輸，這些任務我們同樣可以在fabric中自動化進行：
2.3.1 從本地上傳文件到伺服器
使用put()方法可以將指定的本地文件上傳至伺服器的指定位置，remote參數對應伺服器目標保存位置：
c = Connection(host=f'{user}@{host}',
connect_kwargs={'password': password})
# 創建示例文件
with open('file_transfer.txt', 'w') as d:
d.write('1')
# 利用put方法上傳至伺服器
c.put('file_transfer.txt', remote='/home/feffery/')
# 列印已上傳文件內容
c.run('cat /home/feffery/file_transfer.txt');
2.3.2 從伺服器下載指定文件到本地
相反的，當我們需要從伺服器取回指定文件到本地時，就可以使用get()方法：
c = Connection(host=f'{user}@{host}',
connect_kwargs={'password': password})
# 向文件末尾追加行
c.run('echo "\n2" >> file_transfer.txt')
c.get('/home/feffery/file_transfer.txt')
print(open('file_transfer.txt').read())
以上就是本文的全部內容，歡迎在評論區與我進行討論~
