Saltstack自动化(七)自定义模块

一、内置模块与自定义模块

saltstack内置了很多模块,这些模块使我们简单的执行一条命令就可以返回我们所需的结果。具体可以查看--- SaltStack内置模块列表 。不过也有很多时候,这些内置模块不能满足我们的正常需要,或者返回的结果不理想,这就需要通过自定义模块实现我们想要的结果。自定义SaltStack模块是需要存放在由/etc/salt/master配置文件中file_roots指定的路径下的_modules目录下。(需要创建),默认位置为/srv/salt/_modules。

二、编写自定义模块

1、创建自定义模块存放目录

Master上创建存放模块的目录:

mkdir -pv /srv/salt/_modules
cd /srv/salt/_modules

2、编写自定义模块

这里我想要让返回的格式是txt格式,我在这里模块里定义了两个函数:

# cat ndisk.py
#coding:utf-8
import commands,random
__outputter__ = {
  'test': 'txt',
  'usage': 'txt'
}
def test():
   return random.randint(1,100)%2==0
def usage(per=35):
    ret = []
    (status, output) = commands.getstatusoutput('df -P')
    out = output.splitlines()
    for line in out:
        if not line:
            continue
        if line.startswith('Filesystem'):
            continue
        comps = line.split()
        if len(comps) >= 2:
              perct = (comps[-2]).replace('%', '')
              if int(perct) > per:
                 data = (comps[-1],perct)
                 ret.append(data)
    return  ret

3、同步文件到minion端

同步自定义模块到客户端的方法有三种,这里一般以第二种属于较正统的同步模块方法,第1,3方法同步的时候也会同步其他信息到客户端:

state.highstate
saltutil.sync_modules
saltutil.sync_all

同步完成后,默认会将自定义模块同步到/var/cache/salt/minion/extmods/modules目录下:

4、自定义模块的调用

同步完成后,自定义模块的调用结果如下:

salt-ext-module

这里返回的格式会发现有双方括号,返回的并不美观,一般建议返回字典值,我们还可以修改上面的函数为如下:

def usage(per=35):
    ret = {}
    (status, output) = commands.getstatusoutput('df -P')
    out = output.splitlines()
    for line in out:
        if not line:
            continue
        if line.startswith('Filesystem'):
            continue
        comps = line.split()
        #print comps
        if len(comps) >= 2:
              #print line
              perct = (comps[-2]).replace('%', '')
              #print perct
              if int(perct) > per:
                 key = comps[-1]
                 ret[key] = perct
    return  ret

其执行后,结果如下:

这个结果是不是美观多了?

写模块时,也可以参考salt内置模块,其默认通过rpm安装的位置为:/usr/lib/python2.7/site-packages/salt/modules 。

注:默认salt也带有一个disk模块,不过其执行时的输出结果不满足我们进行巡检的需求,可以通过编写一大批自定义的模块,满足平时巡检的需求

三、自定义模块中的特殊调用

自定义模块中也可以调用salt内部的一些变量或其他模块,同时还可以给模块指定一个虚拟名称。示例如下:

[root@ZJHZ-CMREAD-TEST93 _modules]# cat test.py
#!/usr/bin/python
#coding:utf-8
import time
__virtualname__ = 'hoho'
def __virtual__():
    return __virtualname__
def date():
    return time.time()
def get_osfinger():
    return __grains__['osfinger']
def get_cachedir():
    return __pillar__['productname']
def foo():
    return __salt__['cmd.run']('df')

执行结果如下:

salt-virtual




本站的发展离不开您的资助,金额随意,欢迎来赏!

You can donate through PayPal.
My paypal id: itybku@139.com
Paypal page: https://www.paypal.me/361way

  1. 本文目前尚无任何评论.
  1. 本文目前尚无任何 trackbacks 和 pingbacks.