一、安裝 paramiko 和 psutil
pip install paramiko
pip install psutil
二、實現遠程監控
- 監控入口 run_perf.py
import os
from utils import *
def run_perf(host, port, user, password):
"""
讀取腳本,執行監控
""""
base = os.getcwd()
cmd_file = base + r'/script.txt'
client = connect_to_server(host, port, user, password)
with open(cmd_file) as fp:
cmd = fp.read()
exec_script(client, cmd)
close_connection(client)
if __name__ == '__main__':
run_perf('10.2.2.100', 22, 'root', 'password')
- 遠程連接 utils.py
import json
import paramiko
import psutil
def connect_to_server(host, port, user, password):
"""
SSH 連接服務器
""""
client = paramiko.SSHClient()
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
client.connect(host, port, user, password)
return client
def exec_script(client, script):
"""
運行腳本
""""
stdin, stdout, stderr = client.exec_command(script)
for line in stdout.readlnes():
data = json.loads(line)
print(data)
def close_connection(client):
"""
斷開服務器
""""
client.clsoe()
- 監控腳本 script.py
python <<EOF
import json
import psutil as pu
data = ()
mem_info = {}
mem = pu.virtual_memory()
mem_info['total'] = mem.total
mem_info['available'] = mem.available
mem_info['percent'] = mem.percent
mem_info['used'] = mem.used
mem_info['free'] = mem.free
cpu_info = pu.cpu_percent(interval=2, percpu=True)
data['mem_info'] = mem_info
data['cpu_info '] = cpu_info
data = json.dumps(data)
print(data)
EOF
三、問題 CryptographyDeprecationWarning
- 原因:
paramiko 依賴 cryptography,而最新的 cryptography 裏有些棄用的 API。
- 解放方案
uninstall 最新版,安裝 2.4.2
pip uninstall cryptography
pip install cryptography=2.4.2
PS: 參考鏈接
https://psutil.readthedocs.io/en/latest/