diff --git a/houdini/handlers/play/message.py b/houdini/handlers/play/message.py index 3e36271..fec41af 100644 --- a/houdini/handlers/play/message.py +++ b/houdini/handlers/play/message.py @@ -5,7 +5,7 @@ from houdini.commands import invoke_command_string, has_command_prefix @handlers.handler(XTPacket('m', 'sm')) @handlers.cooldown(.5) -async def handle_send_message(p, penguin_id: int, message): +async def handle_send_message(p, penguin_id: int, message: str): if penguin_id != p.data.id: return await p.close() diff --git a/houdini/handlers/play/player.py b/houdini/handlers/play/player.py index f8b16dd..69bab84 100644 --- a/houdini/handlers/play/player.py +++ b/houdini/handlers/play/player.py @@ -16,7 +16,7 @@ def get_mascot_string_key(_, p, mascot_id): @cached(alias='default', key_builder=get_player_string_key) -async def get_player_string(p, penguin_id): +async def get_player_string(p, penguin_id: int): if penguin_id in p.server.penguins_by_id: return await p.server.penguins_by_id[penguin_id].string else: @@ -26,7 +26,7 @@ async def get_player_string(p, penguin_id): @cached(alias='default', key_builder=get_mascot_string_key) -async def get_mascot_string(p, mascot_id): +async def get_mascot_string(p, mascot_id: int): if mascot_id in p.server.penguins_by_character_id: return await p.server.penguins_by_character_id[mascot_id].string else: