python 运维脚本分享

以查询mysql为例

# -*- coding:utf8 -*-
# Description: 适用于centos7系统,返回json数据
import uuid
import json
import psutil
import socket
import subprocess


class MysqlProcess:

    @classmethod
    def system_command(cls, command):
        # 调用linux系统命令
        shell = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
        stdout, stderr = shell.communicate()
        try:
            return stdout.decode("utf8"), stderr.decode("utf8"), shell.returncode
        except Exception as e:
            return stdout.decode("gbk"), stderr.decode("gbk"), shell.returncode

    @staticmethod
    def localhost_ip():
        # 查询主机ip
        try:
            s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
            s.connect(('8.82.82.8', 280))  # 可随意填一个不能ping通的
            ip = s.getsockname()[0]
            s.close()
            return ip
        except Exception as e:
            return e

    @staticmethod
    def mac_address():
        # 获取mac地址
        mac = uuid.UUID(int=uuid.getnode()).hex[-12:]
        mac_addr = ":".join([mac[e:e + 2] for e in range(0, 11, 2)])
        return mac_addr

    def pid(self):
        # 查询mysql进程id
        pid_command = "ps -ef|egrep -v 'grep|mysqld_safe'|grep mysql|awk -F ' ' '{print $2}'"
        stdout, error, code = self.system_command(pid_command)
        return int(stdout.strip())

    def work_dir(self):
        # 获取工作目录
        mysql_word_dir = "pwdx %s|awk -F ':' '{print $NF}'" % str(self.pid())
        stdout, error, code = self.system_command(mysql_word_dir)
        return stdout.strip()

    def cpu_use_rate(self):
        # mysql进程 cpu使用率
        process = psutil.Process(self.pid())
        use_rate = '{}%'.format(round(process.cpu_percent(), 3))
        return use_rate

    def memory_use_rate(self):
        # mysql进程 内存使用率
        process = psutil.Process(self.pid())
        use_rate = '{}%'.format(round(process.memory_percent(), 3))
        return use_rate

    def run(self):
        # 返回查询数据
        data = {
            'pid': self.pid(), 'ip_address': self.localhost_ip(),
            'word_dir': self.work_dir(), 'cpu_use_rate': self.cpu_use_rate(),
            'memory_use_rate': self.memory_use_rate(), 'mac_address': self.mac_address()
        }
        return json.dumps(data)


if __name__ == '__main__':
    mysql = MysqlProcess()
    mysql.run()