python使用paramiko自动化部署linux程序

使用paramiko模块，比os模块和command模块更加的兼容各种环境。后面两个只能在服务器本机 执行，此模块写得python文件无论是在本地还是服务器本身上运行，都能兼容。

 

paramiko模块的用法如下。

"""自动部署，可以在本地运行执行linux命令，也可以在远程服务器链接自身运行"""

import paramiko
from app.utils.utils_ydf import mixins, unittest_util


class HotelFaresDeploy(mixins.LoggerMixin):
    def __init__(self, project_env='ydf', ip='115.95.89.xx', port=22, user_name="webuser", user_password='xxxxxxx'):
        """
        :param project_env: 项目环境。不同的环境有不同的文件夹
        :param ip: linux服务器ip
        :param port: 端口
        :param user_name: 用户名
        :param user_password: 密码
        """
        hotel_fares_dir = None
        if project_env == 'ydf':
            hotel_fares_dir = '/home/xxx/hotelf/'
        elif project_env == 'test':
            hotel_fares_dir = '/hotel/hotel_fares'
        if project_env not in ('ydf', 'test'):
            raise Exception('设置的项目环境错误')
        self.hotel_fares_dir = hotel_fares_dir
        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(ip, port, user_name, user_password)
        self.ssh = ssh
        self.ssh_session = self.ssh.get_transport().open_session()
        self.cmd0 = 'cd {} ; export PYTHONPATH=$PYTHONPATH:{};'.format(self.hotel_fares_dir, self.hotel_fares_dir)  # 先进入项目所在文件夹和添加环境变量，这样才能保证后面的程序能顺利执行，linux,运行多条命令用;
        self.logger.debug('cmd0 --> ' + self.cmd0)

    def git_pull(self):
        """
        更新git内容
        :return:
        """
        cmd1 = 'git fetch --all;'
        cmd2 = 'git reset --hard origin/haoqiao1.3;'
        cmd3 = 'git -C {} pull origin haoqiao1.3'.format(self.hotel_fares_dir)
        self.exec_cmd(cmd1 + cmd2 + cmd3)

    def exec_cmd(self, cmd):
        """
        执行cmd命令
        :param cmd: 需要执行的命令
        :return:
        :type cmd : str
        """
        self.logger.debug('要执行的命令是： ' + cmd)
        stdin, stdout, stderr = self.ssh.exec_command(self.cmd0 + cmd)
        stdout_str = stdout.read().decode('utf8')
        stderr_str = stderr.read().decode('utf8')
        if stdout_str != '':
            self.logger.info('执行 {} 命令的stdout是 -- > \n{}'.format(cmd, stdout_str))
        if stderr_str != '':
            self.logger.error('执行 {} 命令的stderr是 -- > \n{}'.format(cmd, stderr_str))
        return stdout_str, stderr_str

    def run_pyhton_file(self, python_file_path):
        """
        :param python_file_path:python文件的相对路径
        :return:
        :type python_file_path:str
        """
        python_file_path_short = python_file_path.split('/')[-1]  # 防止是直接去进内层文件夹手动启动的短命令
        stdout_str, stderr_str = self.exec_cmd("ps aux | grep %s|grep -v grep|awk '{print $2}'" % python_file_path_short)
        pid_before = stdout_str
        self.logger.info('进程名 [{}] 部署前的进程号是 --> {}'.format(python_file_path_short, pid_before))

        self.exec_cmd("ps aux|grep %s|grep -v grep|awk '{print $2}' |xargs kill -9" % python_file_path_short)
        self.exec_cmd('nohup python3 %s >/dev/nohup.out 2>&1 &' % python_file_path)  # 这里要用长路径，不然找不到文件
        # cmd = 'ps aux | grep %s|grep -v grep' % python_file_path
        stdout_str, stderr_str = self.exec_cmd("ps aux | grep %s|grep -v grep|awk '{print $2}'" % python_file_path_short)
        pid_after = stdout_str
        self.logger.info('进程名 [{}] 部署后的进程号是 --> {}'.format(python_file_path_short, pid_after))
        if pid_after == '':
            self.logger.critical('重新启动 [{}] 失败'.format(python_file_path_short))
        elif pid_before != pid_after:
            self.logger.warning('重新启动 [{}] 成功'.format(python_file_path_short))
        else:
            self.logger.critical('启动前后进程号一样'.format(python_file_path_short))

    def recover_screen(self):
        """进入screen"""
        raise NotImplementedError('还没写，不需要用screen')

    def run_redis_hotel_id_task_manager(self):
        self.run_pyhton_file('app/apis/elongin/elong_proj/redis_hotel_id_task_manager.py')

    def run_gunicorn(self, application_name, port):
        """
        部署gunicorn
        :param application_name: 应用名称
        :param port: 端口
        :return:
        """
        application_name_short = application_name.split('.')[-1]
        stdout_str, stderr_str = self.exec_cmd("ps aux | grep %s|grep -v grep|awk '{print $2}'" % application_name_short)
        pid_before = stdout_str
        self.logger.info('进程名 [{}] 部署前的进程号是 --> {}'.format(application_name_short, pid_before))

        self.exec_cmd("ps aux|grep %s|grep -v grep|awk '{print $2}' |xargs kill -9" % application_name_short)
        self.exec_cmd('nohup gunicorn -w 9 -k gevent --bind 0.0.0.0:%s %s  >/dev/webnohup.out 2>&1 &' % (port, application_name))

        stdout_str, stderr_str = self.exec_cmd("ps aux | grep %s|grep -v grep|awk '{print $2}'" % application_name_short)
        pid_after = stdout_str
        self.logger.info('进程名 [{}] 部署后的进程号是 --> {}'.format(application_name_short, pid_after))
        if pid_after == '':
            self.logger.critical('重新启动 [{}] 失败'.format(application_name_short))
        elif pid_before != pid_after:
            self.logger.warning('重新启动 [{}] 成功'.format(application_name_short))
        else:
            self.logger.critical('启动前后进程号一样'.format(application_name_short))

    def query_process(self, process_name):
        """查询进程"""
        cmd = "ps aux|grep %s|grep -v grep" % process_name
        self.exec_cmd(cmd)