使用salt models 自定义

使用salt models 自定义获取supervisor服务状态

参考文档：http://www.supervisord.org/api.html

saltstack提供的模块没有提供获取supervisor的服务状态信息，使用cmd.run命令返回的结果需要字符串整理，太low，那么你需要了解下salt的 models使用。

salt models功能可以让你自己定义salt没有的功能，这里编写python脚本从supervisor XML-RPC API来获取想要的数据。

官网给出的方式是使用http的方式获取supervisor的服务状态信息，supervisor默认不开启http服务，配置如下

[rpcinterface:supervisor]
supervisor.rpcinterface_factory = supervisor.rpcinterface:make_main_rpcinterface
;[inet_http_server]       
;port=127.0.0.1:9001

xmlrpclib要求传递的URL以http或https开头，解决这个问题的方法是定义一个忽略该URL的自定义传输。以下是使用来自主管的传输的一些代码：

import supervisor.xmlrpc
import xmlrpclib
proxy = xmlrpclib.ServerProxy('http://127.0.0.1',
                               transport=supervisor.xmlrpc.SupervisorTransport(
                                    None, None, serverurl='unix://'+socketpath))
proxy.supervisor.getState()

如果这是没有用的，这里发现的代码的更新版本在这里：

import httplib
import socket
import xmlrpclib

class UnixStreamHTTPConnection(httplib.HTTPConnection, object):
    def __init__(self, *args, **kwargs):
        self.socketpath = kwargs.pop('socketpath')
        super(UnixStreamHTTPConnection, self).__init__(*args, **kwargs)

    def connect(self):
        self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
        self.sock.connect(self.socketpath)

class UnixStreamTransport(xmlrpclib.Transport, object):
    def __init__(self, *args, **kwargs):
        self.socketpath = kwargs.pop('socketpath')
        super(UnixStreamTransport, self).__init__(*args, **kwargs)

    def make_connection(self, host):
        return UnixStreamHTTPConnection(host, socketpath=self.socketpath)
def all_process_info():
    server = xmlrpclib.ServerProxy('http://localhost', transport=UnixStreamTransport(socketpath="/var/run/supervisor/supervisor.sock"))
    grains = {"supervisor": server.supervisor.getAllProcessInfo()}
    return grains