paramiko远程服务器nohup阻塞问题

一、需求描述：

需要来回切换多台服务器（脚本命令不太熟），就用了python的paramiko模块进行远程连接服务器，控制程序的停止和启动。安装：pip install paramiko

二、问题描述：

import paramiko

# 创建SSH对象
ssh = paramiko.SSHClient()
# 允许连接不在know_hosts文件中的主机
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# 连接服务器
ssh.connect(hostname='192.168.0.3', port=22, username='xxx')
# 执行命令
stdin, stdout, stderr = ssh.exec_command('cd ~/ ; nohup python3.6 run_test.py > nohup_test.log 2>&1 &')
# 获取命令结果
result = stdout.read()
# 关闭连接
ssh.close()

这样连接服务器的时候确实可以执行，但是遇到会阻塞的任务时，就无法生效，找了很多方法，最后发现这个比较有效。

 

三、解决方法

import paramiko


# 创建SSH对象
ssh = paramiko.SSHClient()
# 允许连接不在know_hosts文件中的主机
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
# 连接服务器
ssh.connect(hostname='192.168.0.3', port=22, username='xxx', key=private_key)
# 添加下面代码
transport = ssh.get_transport()
channel = transport.open_session()
# 执行命令 此方法没有返回值
channel.exec_command('cd ~/ ; nohup python3.6 run_test.py > nohup_test.log 2>&1 &')

# 关闭连接
ssh.close()

四、类的调用实现：

简单测试，见下面代码

# -*- coding: utf-8 -*-
"""
20190330
"""

import paramiko
import time
from confs.log import logger  # 自行导入logging模块即可


class EasyConnectHandle(object):
    """操作远程服务器"""

    def __init__(self, connect_host_name:dict):
        """初始化参数"""
        """
            "test":{
                "ip":"192.168.0.189",
                "user_name":"xxxx",
                "pwd":"huhuhu"
            },
        """
        self.connect_host = connect_host_name
        self.ssh = paramiko.SSHClient()
        self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())  # 允许连接陌生服务器
        self.ssh.connect(hostname=self.connect_host["ip"], port=22, username=self.connect_host["user_name"],
                         password=self.connect_host["pwd"], timeout=10)  # 初始化的时候连接到新的服务器
        logger.info(f"登录服务器---{self.connect_host['ip']}成功:")

    def __new__(cls, *args, **kwargs):
        """单例模式"""
        if not hasattr(cls, '_instance'):
            cls._instance = super(EasyConnectHandle, cls).__new__(cls)
        return cls._instance

    def exec(self, cmd=""):
        """执行操作"""
        stdin, stdout, stderr = self.ssh.exec_command(cmd)
        return stdout.read().decode()

    def quit(self):
        """断开服务器"""
        self.ssh.close()
        logger.info(f"退出服务器---{self.connect_host['ip']}成功")

if __name__ == '__main__':
    test_host = {
        "test": {
                "ip": "192.168.0.111",
                "user_name": "xxxx",
                "pwd": "xxxx",
                "jobs": [
                    {
                        "path": "/home/lemon",
                        "type": "touch test_1.sh"
                    },
                    {
                        "path": "/home/lemon",
                        "type": "touch test_2.sh"
                    }
                ]
            }
        }
    for i in ["test"]:
        easy_conn = EasyConnectHandle(test_host[i])
        transport = easy_conn.ssh.get_transport()
        if len(test_host[i].get("jobs", [])) >= 1:
            for job in test_host[i]["jobs"]:
                channel = transport.open_session()
                channel.exec_command(f"cd {job['path']};{job['type']}")
                logger.info(f"服务器---{easy_conn.connect_host['ip']}执行---cd {job['path']};{job['type']}---成功")
                time.sleep(2)
        else:
            logger.info(f"服务器---{easy_conn.connect_host['ip']}暂时没有任务")
        easy_conn.quit()