玩转Python Paramiko库:基础操作讲解与二次封装实践(轻松实现SSH连接)
最编程
2024-02-10 10:13:38
...
# coding=utf-8
import sys, logging
from paramiko.client import SSHClient, AutoAddPolicy
from paramiko import AuthenticationException
from paramiko.ssh_exception import NoValidConnectionsError
class SshClient:
def __init__(self, host_ip, username, password):
# 创建ssh对象
self.ssh_client = SSHClient()
self.host_ip = host_ip
self.username = username
self.password = password
self.port = 22
def __enter__(self):
try:
# 设置允许连接known_hosts文件中的主机(默认连接不在known_hosts文件中的主机会拒绝连接抛出SSHException)
self.ssh_client.set_missing_host_key_policy(AutoAddPolicy)
# 连接服务器
self.ssh_client.connect(self.host_ip, self.port, self.username, self.password, timeout=60)
except AuthenticationException as e:
logging.warning('username or password error')
raise e
# return 1001
except NoValidConnectionsError as e:
logging.warning('connect time out')
raise e
# return 1002
except Exception as a:
print('Unexpected error:', sys.exc_info()[0])
raise a
# return 1003
return self
def excute_command(self, commands):
# 执行命令
# stdin:标准输入(就是你输入的命令);stdout:标准输出(就是命令执行结果);stderr:标准错误
# (命令执行过程中如果出错了就把错误打到这里),stdout和stderr仅会输出一个
stdin, stdout, stderr = self.ssh_client.exec_command(commands)
# 返回命令结果
return stdout.read().decode()
def __exit__(self, exc_type, exc_val, exc_tb):
# 关闭服务器连接
self.ssh_client.close()
if __name__ == '__main__':
with SshClient('192.168.2.63', 'root', 'password') as ssh_c:
blk = 'vdb'
command = "df -h | grep /dev/%s" % blk
result = ssh_c.excute_command(command)
if len(result) == 0:
mount_command = "mount /dev/%s /mnt" % blk
mount_result = ssh_c.excute_command(mount_command)
print(mount_result)