Python中的paramiko模块
paramiko支持SSH协议，可以与Linux或Windows（搭建了SSH服务）进行交互，包括远程执行命令或执行上传/下载文件等操作。
1.paramiko远程密码连接
# 1. 基于ssh用于连接远程服务器做操作:远程执行命令， 上传文件， 下载文件
import paramiko
# ssh [email protected]
# 创建一个ssh对象;
client = paramiko.SSHClient()
# 2. 解决问题:如果之前没有；连接过的ip, 会出现
# Are you sure you want to continue connecting (yes/no)? yes
# 自动选择yes
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# 3. 连接服务器
client.connect(hostname='172.25.254.46',
port=22,
username='root',
password='villa')
# 4. 执行操作
stdin, stdout, stderr = client.exec_command('hostname')
# 5. 获取命令的执行结果;
#result=stdout.read() 为2进制的对象，因此对2进制的对象进行转换
result = stdout.read().decode('utf-8')
print(result)
print(stderr.read())
# 6. 关闭连接
client.close()
另：
stdin是标准输入文件，stdout是标准输出文件，stderr标准出错文件，应用在输出的重新定位上。
程序按如下方式使用这些文件：
标准输入是程序可以读取其输入的位置。缺省情况下，进程从键盘读取 stdin。
标准输出是程序写入其输出的位置。缺省情况下，进程将 stdout写到终端屏幕上。
标准错误是程序写入其错误消息的位置。缺省情况下，进程将 stderr写到终端屏幕上。
2.paramiko远程密码连接
# 基于ssh用于连接远程服务器做操作:远程执行命令， 上传文件， 下载文件
import paramiko
import logging
from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException
def connect(cmd, hostname, port=22, username='root', password='villa'):
# ssh [email protected]
# 创建一个ssh对象;
client = paramiko.SSHClient()
# 2. 解决问题:如果之前没有；连接过的ip, 会出现
# Are you sure you want to continue connecting (yes/no)? yes
# 自动选择yes
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
# 3. 连接服务器
client.connect(hostname=hostname,
port=port,
username=username,
password=password)
print("正在连接主机%s......." %(hostname))
except NoValidConnectionsError as e:
print("连接失败")
except AuthenticationException as e:
print("密码错误")
else:
# 4. 执行操作
stdin, stdout, stderr = client.exec_command(cmd)
# 5. 获取命令的执行结果;
result = stdout.read().decode('utf-8')
print(result)
# 6. 关闭连接
client.close()
with open('host.txt') as f:
for line in f:
line = line.strip()
hostname, port, username, password = line.split(':')
print(hostname.center(50, '*'))
connect('hostname', hostname, port, username, password)
3.paramiko基于公钥密钥连
# 基于ssh用于连接远程服务器做操作:远程执行命令， 上传文件， 下载文件
import paramiko
from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException
def connect(cmd, hostname, port=22, user='root'):
# ssh [email protected]
# 创建一个ssh对象;
client = paramiko.SSHClient()
# 返回一个私钥对象
private_key = paramiko.RSAKey.from_private_key_file('id_rsa')
# 2. 解决问题:如果之前没有；连接过的ip, 会出现
# Are you sure you want to continue connecting (yes/no)? yes
# 自动选择yes
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
# 3. 连接服务器
client.connect(hostname=hostname,
port=port,
username=user,
pkey=private_key
)
# 4. 执行操作
stdin, stdout, stderr = client.exec_command(cmd)
except NoValidConnectionsError as e:
print("连接失败")
except AuthenticationException as e:
print("密码错误")
else:
# 5. 获取命令的执行结果;
result = stdout.read().decode('utf-8')
print(result)
finally:
# 6. 关闭连接
client.close()
for count in range(254):
host = '172.25.254.%s' %(count+1)
print(host.center(50, '*'))
connect('uname', host)
4.paramiko再次封装
import os
import paramiko
from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException, SSHException
class SshRemoteHost(object):
def __init__(self, hostname, port, user, passwd, cmd):
# 指的不是shell命令
# cmd shell命令
# put
# get
self.hostname = hostname
self.port = port
self.user = user
self.passwd = passwd
self.cmd = cmd
def run(self):
"""默认调用的内容"""
# cmd hostname
# put
# get
cmd_str = self.cmd.split()[0] # cmd
# 类的反射, 判断类里面是否可以支持该操作?
if hasattr(self, 'do_'+ cmd_str): # do_cmd
getattr(self, 'do_'+cmd_str)()
else:
print("目前不支持该功能")
def do_cmd(self):
# 创建一个ssh对象;
client = paramiko.SSHClient()
# 2. 解决问题:如果之前没有；连接过的ip, 会出现
# Are you sure you want to continue connecting (yes/no)? yes
# 自动选择yes
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
# 3. 连接服务器
client.connect(hostname=self.hostname,
port=self.port,
username=self.user,
password=self.passwd)
print("正在连接主机%s......." % (self.hostname))
except NoValidConnectionsError as e:
print("连接失败")
except AuthenticationException as e:
print("密码错误")
else:
# 4. 执行操作
# cmd uname
# cmd ls /etc/
# *******注意：
cmd = ' '.join(self.cmd.split()[1:])
stdin, stdout, stderr = client.exec_command(cmd)
# 5. 获取命令的执行结果;
result = stdout.read().decode('utf-8')
print(result)
# 6. 关闭连接
client.close()
def do_put(self):
# put /tmp/passwd /tmp/passwd
# put /tmp/passwd /tmp/pwd
# put /tmp/passwd # 将本机的/tmp/passwd文件上传到远程主机的/tmp/passwd;
print("正在上传.....")
try:
transport = paramiko.Transport((self.hostname, int(self.port)))
transport.connect(username=self.user, password=self.passwd)
except SSHException as e:
print("连接失败")
else:
sftp = paramiko.SFTPClient.from_transport(transport)
newCmd = self.cmd.split()[1:]
if len(newCmd) == 2:
# 上传文件, 包含文件名
sftp.put(newCmd[0], newCmd[1])
print("%s文件上传到%s主机的%s文件成功" %(newCmd[0],
self.hostname, newCmd[1]))
else:
print("上传文件信息错误")
transport.close()
def do_get(self):
print("正在下载.....")
# 2. 根据选择的主机组， 显示包含的主机IP/主机名;
# 3. 让用户确认信息， 选择需要批量执行的命令；
# - cmd shell命令
# - put 本地文件 远程文件
# - get 远程文件 本地文件
def main():
# 1. 选择操作的主机组：eg: mysql, web, ftp
groups = [file.rstrip('.conf') for file in os.listdir('conf')]
print("主机组显示:".center(50, '*'))
for group in groups: print('\t', group)
choiceGroup = input("清选择批量操作的主机组(eg:web):")
# 2. 根据选择的主机组， 显示包含的主机IP/主机名;
# 1). 打开文件conf/choiceGroup.conf
# 2). 依次读取文件每一行，
# 3). 只拿出ip
print("主机组包含主机:".center(50, '*'))
with open('conf/%s.conf' %(choiceGroup)) as f:
for line in f:
print(line.split(':')[0])
f.seek(0,0) # 把指针移动到文件最开始
hostinfos = [line.strip() for line in f.readlines()]
# 3. 让用户确认信息， 选择需要批量执行的命令；
print("批量执行脚本".center(50, '*'))
while True:
cmd = input(">>:").strip() # cmd uname
if cmd:
if cmd == 'exit' or cmd =='quit':
print("执行结束， 退出中......")
break
# 依次让该主机组的所有主机执行
for info in hostinfos:
# 'ip:port:user:passwd'
host, port, user, passwd = info.split(":")
print(host.center(50, '-'))
clientObj = SshRemoteHost(host, port, user, passwd, cmd)
clientObj.run()
if __name__ == '__main__':
main()