Refactor the listeners of the commands.
continuous-integration/drone/push Build is passing Details

main
KKlochko 2 years ago
parent fa81921fad
commit 32aa9f3225

@ -23,11 +23,19 @@ def setup(bot: botlib.Bot, prefix: str):
if not match.prefix(): if not match.prefix():
return return
# No registered user required
if match.command('help'): if match.command('help'):
await help_handler(bot=bot, room_id=room.room_id, sender=message.sender, admin_id=bot.config.admin_id) await help_handler(bot=bot, room_id=room.room_id, sender=message.sender, admin_id=bot.config.admin_id)
@bot.listener.on_message_event
async def guests_commands(room: nio.rooms.MatrixRoom, message: nio.events.room_events.Event):
match = botlib.MessageMatch(room, message, bot, prefix=prefix)
if not match.is_not_from_this_bot():
return
if not match.prefix():
return
if match.command('city-list'): if match.command('city-list'):
await city_list_handler(bot=bot, room_id=room.room_id, sender=message.sender, admin_id=bot.config.admin_id) await city_list_handler(bot=bot, room_id=room.room_id, sender=message.sender, admin_id=bot.config.admin_id)
@ -39,7 +47,15 @@ def setup(bot: botlib.Bot, prefix: str):
if match.command('register'): if match.command('register'):
await register_handler(bot=bot, room_id=room.room_id, sender=message.sender, admin_id=bot.config.admin_id, args=match.args()) await register_handler(bot=bot, room_id=room.room_id, sender=message.sender, admin_id=bot.config.admin_id, args=match.args())
# A registered user required @bot.listener.on_message_event
async def authorized_commands(room: nio.rooms.MatrixRoom, message: nio.events.room_events.Event):
match = botlib.MessageMatch(room, message, bot, prefix=prefix)
if not match.is_not_from_this_bot():
return
if not match.prefix():
return
if match.command('company-list'): if match.command('company-list'):
await company_list_handler(bot=bot, room_id=room.room_id, sender=message.sender, admin_id=bot.config.admin_id) await company_list_handler(bot=bot, room_id=room.room_id, sender=message.sender, admin_id=bot.config.admin_id)

Loading…
Cancel
Save