GitHub/PyLink
3
0
Fork 0
mirror of https://github.com/jlu5/PyLink.git synced 2024-11-23 19:19:31 +01:00
Code Issues Projects Releases Wiki Activity

opercmds.checkban: make maxresults configurable, up to a cap of 200

Browse Source
This commit is contained in:
James Lu 2017-08-02 21:43:58 +08:00
parent 981e6c508f
commit bf495a0aae
1 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
plugins/opercmds.py
View File

@ -1,18 +1,30 @@
"""
opercmds.py: Provides a subset of network management commands.
"""
import argparse
from pylinkirc import utils
from pylinkirc.log import log
from pylinkirc.coremods import permissions
# Having a hard limit here is sensible because otherwise it can flood the client or server off.
CHECKBAN_MAX_RESULTS = 200
def _checkban_positiveint(value):
value = int(value)
if value <= 0 or value > CHECKBAN_MAX_RESULTS:
raise argparse.ArgumentTypeError("%s is not a positive integer between 1 and %s." % (value, CHECKBAN_MAX_RESULTS))
return value
checkban_parser = utils.IRCParser()
checkban_parser.add_argument('banmask')
checkban_parser.add_argument('target', nargs='?', default='')
checkban_parser.add_argument('--channel', default='')
checkban_parser.add_argument('--maxresults', type=_checkban_positiveint, default=50)
@utils.add_cmd
def checkban(irc, source, args):
"""<banmask> [<target nick or hostmask>] [--channel #channel]
"""<banmask> [<target nick or hostmask>] [--channel #channel] [--maxresults <num>]
CHECKBAN provides a ban checker command based on nick!user@host masks, user@host masks, and
PyLink extended targets.
@ -22,7 +34,9 @@ def checkban(irc, source, args):
results.
If the --channel argument is given without a target mask, the returned results will only
include users in the given channel."""
include users in the given channel.
The --maxresults option configures how many responses will be shown."""
permissions.checkPermissions(irc, source, ['opercmds.checkban'])
args = checkban_parser.parse_args(args)
@ -39,7 +53,7 @@ def checkban(irc, source, args):
irc.msg(source, "Checking for hosts that match \x02%s\x02:" % args.banmask, notice=True)
for uid, userobj in irc.users.copy().items():
if irc.match_host(args.banmask, uid):
if results < 50: # XXX rather arbitrary limit
if results < args.maxresults:
s = "\x02%s\x02 (%s@%s) [%s] {\x02%s\x02}" % (userobj.nick, userobj.ident,
userobj.host, userobj.realname, irc.get_friendly_name(irc.get_server(uid)))
@ -49,7 +63,7 @@ def checkban(irc, source, args):
else:
if results:
irc.msg(source, "\x02%s\x02 out of \x02%s\x02 results shown." %
(min([results, 50]), results), notice=True)
(min([results, args.maxresults]), results), notice=True)
else:
irc.msg(source, "No results found.", notice=True)
else: