bitbot-3.11-fork/modules/hostmask_tracking.py

36 lines
1.4 KiB
Python

from src import ModuleManager, utils
class Module(ModuleManager.BaseModule):
_name = "Hostmasks"
@utils.hook("new.user")
def new_user(self, event):
userhost = event["user"].userhost()
if not userhost == None:
known_hostmasks = event["user"].get_setting("known-hostmasks", [])
if not userhost in known_hostmasks:
known_hostmasks.append(userhost)
event["user"].set_setting("known-hostmasks", known_hostmasks)
@utils.hook("received.command.maskfind")
@utils.kwarg("min_args", 1)
@utils.kwarg("private_only", True)
@utils.kwarg("help", "Find all nicknames that used a given hostmask")
@utils.kwarg("usage", "<hostmask>")
@utils.kwarg("permission", "maskfind")
def maskfind(self, event):
all_userhosts = event["server"].get_all_user_settings("known-hostmasks")
nicknames = set([])
hostmask_str = event["args_split"][0]
hostmask = utils.irc.hostmask_parse(hostmask_str)
for nickname, userhosts in all_userhosts:
for userhost in userhosts:
if hostmask.match(userhost):
nicknames.add(nickname)
if nicknames:
event["stdout"].write("%s: %s" %
(hostmask_str, ", ".join(sorted(nicknames))))
else:
event["stderr"].write("Hostmask not found")