gozerbot.plugs.grep

grep the output of bot comamnds.

gozerbot.plugs.grep.handle_grep(bot, ievent)

<txt> - grep the result list.

CODE

# plugs/grep.py
#
#

""" grep the output of bot comamnds. """

__status__ = "seen"

gozerbot imports

from gozerbot.commands import cmnds
from gozerbot.utils.generic import waitforqueue
from gozerbot.plughelp import plughelp
from gozerbot.examples import examples
from gozerbot.tests import tests

basic importd

import getopt
import re

plughelp

plughelp.add('grep', ' grep elements of a pipeline')

grep command

def handle_grep(bot, ievent):
    """ <txt> - grep the result list. """
    if not ievent.inqueue: ievent.reply('use grep in a pipeline') ; return
    if not ievent.rest: ievent.missing('<txt>') ; return
    try: (options, rest) = getopt.getopt(ievent.args, 'riv')
    except getopt.GetoptError, ex: ievent.reply(str(ex)) ; return
    result = waitforqueue(ievent.inqueue, 30)
    if not result: ievent.reply('no data to grep on') ; return
    doregex = False
    docasein = False
    doinvert = False
    for i, j in options:
        if i == '-r': doregex = True
        if i == '-i': docasein = True
        if i == '-v': doinvert = True
    res = []
    if doregex:
        try:
            if docasein: reg = re.compile(' '.join(rest), re.I)
            else: reg = re.compile(' '.join(rest))
        except Exception, ex: ievent.reply("can't compile regex: %s" % str(ex)) ; return
        if doinvert:
            for i in result:
                if not re.search(reg, i): res.append(i)
        else:
            for i in result:
                if re.search(reg, i): res.append(i)
    else:
        if docasein: what = ' '.join(rest).lower()
        elif doinvert: what = ' '.join(rest)
        else: what = ievent.rest
        for i in result:
            if docasein:
                if what in str(i.lower()): res.append(i)
            elif doinvert:
                if what not in str(i): res.append(i)
            else:
                if what in str(i): res.append(i)
    if not res: ievent.reply('no result')
    else: ievent.reply(res, dot=True)

cmnds.add('grep', handle_grep, ['USER', 'WEB', 'CLOUD'], threaded=True)
examples.add('grep', 'grep the output of a command', 'list | grep core')
tests.add('list | grep core', 'core')