salt扩展与python脚本

xywuyiba8

　　
来源 SALTSTACK技术入门与实战
　　

　　
1.扩展grains
　　
通过Python脚本定义grains
　　
http://www.xiaomastack.com/2014/10/31/saltstack-grains/
　　
file_roots:
　　
  base:
　　
    - /srv/salt/
　　
/srv/salt/_grains/example.py
　　
#!/usr/bin/python
　　
def grains():
　　local={}
　　test={'key': 'vaule','key1': 'vaule1','key2': 'vaule2'}
　　local['list']= [1,2,3,4]
　　local['string'] = 'str'
　　local['dict'] = test
　　return local
　　
salt 'Minion' saltutil.sync_grains
　　

　　

　　
2.扩展Module
　　
当前Python版本的site-packages/salt/modules/下
　　
脚本里面的一个函数就是Module的一个方法
　　
def A(host, nameserver=None)
　　
dig = ['dig', '+short', str(host), 'A']
　　
dig.append('@{0}'.format(nameserver))
　　
__salt__['cmd.run_all']
　　

　　
__virtualname__是定义module名称
　　
__virtual__函数的作用主要是在Module的时候需要判断dig命令是否存在
　　

　　

　　
puppet.py
　　
def setmaster
　　
def service(signal=None):
　　
def master(config_file='/etc/puppet/puppet.conf'):
　　
setmaster函数主要指定puppetserver地址
　　
version函数是查看minion上puppet的版本
　　
service函数是去管理puppet的服务状态
　　
master函数是查看目前puppet配置文件里面定义的server地址
　　
使用salt 'Minion' puppet.version
　　

　　
意思就是说在site-packages/salt/modules里面写一个py，然后写很多函数，最后salt就可以像这样puppet.version调用了。
　　

　　
下面这个表明只是在客户端执行，这就是模块的作用。
　　

　　
代码
　　
saltstack之远程触发文件备份、回滚
　　
mkdir /srv/salt/_modules   默认没有此文件，自己生成一个
　　

　　
#!/usr/bin/python
　　
# -*- coding: utf-8 -*-
　　
import sys,string,shutil
　　
import os,tarfile
　　
import datetime,time
　　

　　
tn=datetime.datetime.today()
　　
time_now=tn.strftime("%Y-%m-%d")
　　
data_bak='/data/databak'
　　
data_tmp='/data/databak/tmp/%s' % time_now
　　
com_f="%s/%s" % (data_bak,time_now)
　　
if not os.path.exists(com_f):
　　
     os.makedirs(com_f)
　　

　　
def CpFile():
　　
     id = sys.argv[1]
　　
     dir = sys.argv[2]            #传入两个变量，任务自动生成的id与要替换的文件
　　
     filename = '%s/%s.tar' % (com_f, id)
　　
     mode = 'w:tar'
　　
     os.chdir(data_bak)
　　
     w_file=open("/tmp/tmp.list",'w')
　　
     w_file.write(id+" "+dir)              #记录每次备份的id与相对应的备份目录或文件的路径
　　
     w_file.close()
　　
     file = tarfile.open( filename, mode )
　　
     file.add( '/tmp/tmp.list' )
　　
     file.add( dir )
　　
     file.close()
　　
     return 'ok'       #测试过程，就先让返回ok吧，之后再做判断
　　

　　

　　
def RollBack():
　　
     id = sys.argv[1]        #想要回滚到的版本id
　　
     if not os.path.exists(data_tmp):
　　
          os.makedirs(data_tmp)
　　
     filename = '%s/%s.tar' % (com_f, id)
　　
     tar = tarfile.open("%s" % filename)
　　
     for b in tar:
　　
          tar.extract(b,path="%s" % data_tmp)
　　
     tar.close()
　　
     for line in open('%s/tmp/tmp.list' % data_tmp):
　　
          id = line.split(" ")[:1][0]
　　
          dir = line.split(" ")[1:][0]       #读取备份时的路径
　　
          backup_dir='%s/%s' % (data_tmp,dir)
　　
          os.system('\cp -rp %s %s' % (backup_dir,dir))
　　
          return 'ok'
　　

　　

　　

　　
3.扩展state
　　

　　
state 说白就是调用已经存在的各种modules,具体salt官网很多。
　　

　　
默认SaltStack的state脚本都是在当前Python版本的site-packages/salt/states/下
　　
SaltStack没有一个比较合适的state的话，我们还可以通过Python语言去定义一个state
　　

　　
3.1 __salt__['rabbitmq.user_exists']
　　
def absent(name,runas=None):
　　
result = __salt__['rabbitmq.delete_user']()
　　

　　
最终return ret，返回一个字典
　　

　　
3.2 cat /srv/salt/_states/ansible.py
　　
def files(name='/etc/ansible/ansible.cfg',inventory=None,forks=None,module_lang=None,
　　
host_key_checking=None,timeout=None)
　　
file=__salt__['file.file_exists'](name)
　　
__opts__['test']:
　　
接下来我们编写一个简单的state来验证一下：
　　
SaltStack@Master: cat /srv/salt/ansible.sls
　　
'/etc/ansible/ansible.cfg':
　　
  ansible.files:  #这里的files就是py脚本定义的files函数
　　
    - inventory: /etc/host
　　
    - timeout: 88
　　
    - forks: 8
　　

　　
使用saltutil.sync_states命令同步即可。脚本会同步到
　　
Minion定义的cachedir目录下的extmods/states下。
　　
运行一次 salt 'Minion' state.sls ansible
　　

　　
我们编写state其实大量的工作就是把想要完成的需求，通过state YAML形式定义
　　
出来。然后我们编写的state脚本去真正去完成我们想要的结果
　　

　　
4.ext_pillar与ext_nodes
　　

　　
4.1.1 ext_pillar 写过  245页
　　
官网一个Hiera例子
　　
data = yaml.safe_load(__salt__['cmd.run']()
　　

　　
4.1.2 自己编写一个ext_pillar接口
　　
extension_modules: /srv/salt/modules/
　　
ext_pillar:
　　
- salt: []
　　
这里ext_pillar接口的脚本就叫做salt.py，脚本的路径是在/srv/salt/modules/pillar/目录下
　　

　　
脚本返回字典就行。
　　

　　

　　
4.2 理解ext_nodes流程和案例  238页
　　
平常使用SaltStack的state的时候 top.sls入口文件
　　
top.sls作用是指定Minion与state.sls文件的对应关系
　　
目的top.sls能做到动态
　　

　　
四种形式：
　　
1）通过从MongoDB里面获取Minion与state.sls文件的对应关系：
　　
2）通过ext_nodes的形式：
　　
3）通过reclass_adapter的形式：
　　
4）通过从cobbler直接获取的形式：
　　

　　
ext_nodes支持通过脚本的形式
　　
脚本返回的字典
　　
base:
　　
'Minion':
　　
- cpis
　　
stage:
　　
'Minion01':
　　
- sshd
　　

　　
master_tops:
　　
ext_nodes: /srv/salt/saltbook.py
　　

　　
最后我们来运行state.highstate查看运行结果
　　

　　
4.3　SaltStack git文件服务器  这块可以看做state的扩展
　　

　　
SaltStack除了支持默认roots fileserver以外，还支持git fileserver
　　
salt '*' pillar.get master:environment(这点是至关重要的)
　　

　　
修改master配置文件
　　
fileserver_backend:
　　
- git
　　
gitfs_remotes:
　　
- https://github.com/shencan/saltstates.git
　　
gitfs_root: states
　　

　　
使用salt-run fileserver.update命令更新state文件