34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
from .base import Base
|
|
from src.config import trigram_repository
|
|
|
|
|
|
class Moderate(Base):
|
|
aliases = ['mod_f', 'mod_d']
|
|
|
|
@staticmethod
|
|
def execute(bot, command):
|
|
try:
|
|
if not command.is_private() and not Moderate.is_admin(bot, command):
|
|
return Moderate.reply(bot, command, 'You don\'t have admin privileges!')
|
|
|
|
if len(command.args) == 0:
|
|
raise IndexError
|
|
|
|
if command.name == 'mod_f':
|
|
words = trigram_repository.find_word(command.chat_id, command.args[0].strip())
|
|
|
|
Moderate.reply(bot, command, '\n'.join(words))
|
|
elif command.name == 'mod_d':
|
|
trigram_repository.remove_word(command.chat_id, command.args[0].strip())
|
|
except (IndexError, ValueError):
|
|
Moderate.reply(bot, command, """Usage:
|
|
/mod_f <similar_word> for search
|
|
/mod_d <exact_word> for deletion""")
|
|
|
|
@staticmethod
|
|
def is_admin(bot, entity):
|
|
user_id = entity.message.from_user.id
|
|
admin_ids = list(map(lambda m: m.user.id, bot.get_chat_administrators(entity.chat_id)))
|
|
|
|
return user_id in admin_ids
|