How can I make one piece of the script run with the on argument?

Question

Python

The code:

def 
mode:
    return arg.split()[1]


@bot.message_handler(commands=['antim'])
def mat(message):
    mod=mode(message.text)
    if mod == 'on':
        names = ['mats']
        for name in names:
            if name in message.text.lower():
                bot.reply_to(message, ' Не матерись!')
    elif mod == 'off':
        None

It is necessary that when antim on is launched, the anti-mat script is launched, and when antim off is introduced, it is off