扩展saltstack组件

yangcctv

[iyunv@linux-node1 _modules]# pwd　　
/srv/salt/base/_modules
　　
[iyunv@linux-node1 _modules]# cat puppet.py
　　
#!/usr/bin/env python
　　
# _*_ coding:utf-8 _*_
　　
from __future__ import absolute_import
　　
from salt.ext.six.moves.urllib.request import urlopen as _urlopen
　　
import salt.utils
　　
import salt.utils.decorators as decorators
　　

　　
@decorators.memoize
　　
def __detect_os():
　　
     return salt.utils.which('puppet')
　　

　　
def __virtual__():
　　
     if __detect_os():
　　
         return True
　　
     return False
　　

　　
def setmaster(master='linux-node1.oldboyedu.com',config_file='/etc/puppet/puppet.conf'):
　　
     check='grep server {0}'.format(config_file)
　　
     outcheck = __salt__['cmd.run'](check)
　　
     if outcheck:
　　
         cmdline = 'sed -i "s/server = .*/server = {0}/g" {1}'.format(master,config_file)
　　
         output = __salt__['cmd.run'](cmdline)
　　
         return 'has change server to {0}'.format(master)
　　
     else:
　　
         cmdline = 'echo "   server = {0}" >> {1}'.format(master,config_file)
　　
         output = __salt__['cmd.run'](cmdline)
　　
         return 'has change server to {0} need restart the service'.format(master)
　　

　　
def version():
　　
     cmd = 'rpm -qf {0}'.format(__detect_os())
　　
     output = __salt__['cmd.run'](cmd).splitlines()
　　
     ret = output[0].split(': ')
　　
     return ret[-1]
　　

　　
def service(single=None):
　　
     status = ('start','stop','status','restart','reload','force-reload','condrestart','once','genconfig')
　　
     if single not in status:
　　
         return 'puppet can not suppert this single'
　　
     cmdline = 'systemctl {0} puppet.service'.format(single)
　　
     output = __salt__['cmd.run'](cmdline)
　　
     return output
　　

　　
def master(config_file='/etc/puppet/puppet.conf'):
　　
     cmdline= 'grep server' + '{0}'.format(config_file)
　　
     output=__salt__['cmd.run'](cmdline,python_shell=False)
　　
     if output:
　　
         return output
　　
     else:
　　
         return 'puppet server not setting'