nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

[grep] use subparse feature

Browse Source
This commit is contained in:
nemunaire 2016-01-14 23:54:11 +01:00
parent 1d13d56dce
commit 645c18c981
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
modules/grep.py
View File

@ -14,13 +14,12 @@ from more import Response
# MODULE CORE ######################################################### # MODULE CORE #########################################################
def grep(fltr, cmd, args, msg, icase=False, only=False): def grep(fltr, cmd, msg, icase=False, only=False):
"""Perform a grep like on known nemubot structures """Perform a grep like on known nemubot structures
Arguments: Arguments:
fltr -- The filter regexp fltr -- The filter regexp
cmd -- The subcommand to execute cmd -- The subcommand to execute
args -- subcommand arguments
msg -- The original message msg -- The original message
icase -- like the --ignore-case parameter of grep icase -- like the --ignore-case parameter of grep
only -- like the --only-matching parameter of grep only -- like the --only-matching parameter of grep
@ -28,11 +27,7 @@ def grep(fltr, cmd, args, msg, icase=False, only=False):
fltr = re.compile(fltr, re.I if icase else 0) fltr = re.compile(fltr, re.I if icase else 0)
for r in context.subtreat(Command(cmd, for r in context.subtreat(context.subparse(msg, cmd)):
args,
to_response=msg.to_response,
frm=msg.frm,
server=msg.server)):
if isinstance(r, Response): if isinstance(r, Response):
for i in range(len(r.messages) - 1, -1, -1): for i in range(len(r.messages) - 1, -1, -1):
if isinstance(r.messages[i], list): if isinstance(r.messages[i], list):
@ -79,8 +74,7 @@ def cmd_grep(msg):
only = "only" in msg.kwargs only = "only" in msg.kwargs
l = [m for m in grep(msg.args[0] if msg.args[0][0] == "^" else ".*?(" + msg.args[0] + ").*?", l = [m for m in grep(msg.args[0] if msg.args[0][0] == "^" else ".*?(" + msg.args[0] + ").*?",
msg.args[1][1:] if msg.args[1][0] == "!" else msg.args[1], " ".join(msg.args[1:]),
msg.args[2:],
msg, msg,
icase="nocase" in msg.kwargs, icase="nocase" in msg.kwargs,
only=only) if m is not None] only=only) if m is not None]