mir/microbot
2
0
Fork 0
mirror of https://github.com/Nick80835/microbot synced 2025-08-22 18:19:16 +00:00
Code Issues Releases Wiki Activity

allow customizing nsfw warnings

Browse Source
This commit is contained in:
Nick80835 2021-02-11 10:47:08 -05:00
parent 93819eacfa
commit 0328df1f79
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ubot/command.py
View File

@ -18,6 +18,7 @@ class Command:
self.sudo = args.get("sudo", False) self.sudo = args.get("sudo", False)
self.admin = args.get("admin", False) self.admin = args.get("admin", False)
self.nsfw = args.get("nsfw", False) self.nsfw = args.get("nsfw", False)
self.nsfw_warning = args.get("nsfw_warning", None)
self.pass_nsfw = args.get("pass_nsfw", False) self.pass_nsfw = args.get("pass_nsfw", False)
self.locking = args.get("locking", False) self.locking = args.get("locking", False)
self.lock_reason = None self.lock_reason = None

4
ubot/command_handler.py
View File

@ -223,7 +223,7 @@ class CommandHandler():
command.lock_reason = None command.lock_reason = None
elif command.user_locking: elif command.user_locking:
if event.sender_id in command.locked_users: if event.sender_id in command.locked_users:
await event.reply(f"Please don't spam that command.") await event.reply("Please don't spam that command.")
return return
if command.chance and randint(0, 100) <= command.chance or not command.chance: if command.chance and randint(0, 100) <= command.chance or not command.chance:
@ -269,7 +269,7 @@ class CommandHandler():
return False return False
if event.chat and command.nsfw and str(event.chat.id) in self.settings.get_list("nsfw_blacklist"): if event.chat and command.nsfw and str(event.chat.id) in self.settings.get_list("nsfw_blacklist"):
await event.reply("NSFW commands are disabled in this chat!") await event.reply(command.nsfw_warning or "NSFW commands are disabled in this chat!")
print(f"Attempted NSFW command ({event.raw_text}) in blacklisted chat ({event.chat.id}) from ID {event.sender_id}") print(f"Attempted NSFW command ({event.raw_text}) in blacklisted chat ({event.chat.id}) from ID {event.sender_id}")
return False return False