Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

Merge pull request #279 from examknow/develop

Browse source 
add ban-enforce-max config option
This commit is contained in:
jess 2020-10-05 01:28:39 +01:00 committed by GitHub
commit 26b2037c9b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
modules/ban_enforce.py
View file

@ -4,6 +4,8 @@ REASON = "User is banned from this channel"
@utils.export("channelset", utils.BoolSetting("ban-enforce", @utils.export("channelset", utils.BoolSetting("ban-enforce",
"Whether or not to parse new bans and kick who they affect")) "Whether or not to parse new bans and kick who they affect"))
@utils.export("channelset", utils.IntSetting("ban-enforce-max",
"Do not enforce ban if the ban effects more than this many users. Default is half of total channel users."))
class Module(ModuleManager.BaseModule): class Module(ModuleManager.BaseModule):
@utils.hook("received.mode.channel") @utils.hook("received.mode.channel")
def on_mode(self, event): def on_mode(self, event):
@ -14,6 +16,10 @@ class Module(ModuleManager.BaseModule):
if mode[0] == "+" and mode[1] == "b": if mode[0] == "+" and mode[1] == "b":
bans.append(arg) bans.append(arg)
affected = 0
defaultmax = len(event["channel"].users) // 2
realmax = event["channel"].get_setting("ban-enforce-max", defaultmax)
if bans: if bans:
umasks = {u.hostmask(): u for u in event["channel"].users} umasks = {u.hostmask(): u for u in event["channel"].users}
for ban in bans: for ban in bans:
@ -21,7 +27,10 @@ class Module(ModuleManager.BaseModule):
matches = list(utils.irc.hostmask_match_many( matches = list(utils.irc.hostmask_match_many(
umasks.keys(), mask)) umasks.keys(), mask))
for match in matches: for match in matches:
affected = affected + 1
kicks.add(umasks[match]) kicks.add(umasks[match])
if kicks: if kicks:
if affected > realmax:
return
nicks = [u.nickname for u in kicks] nicks = [u.nickname for u in kicks]
event["channel"].send_kicks(sorted(nicks), REASON) event["channel"].send_kicks(sorted(nicks), REASON)