From 8ea1c1261edee4b63238d9ad3591609e007e80f4 Mon Sep 17 00:00:00 2001 From: Yannis300307 Date: Tue, 2 Jan 2024 15:04:46 +0100 Subject: [PATCH] Ajout de typage dans le entity_manager --- src/engine/entity_manager.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/engine/entity_manager.py b/src/engine/entity_manager.py index 9632ddb..cad71e8 100644 --- a/src/engine/entity_manager.py +++ b/src/engine/entity_manager.py @@ -9,7 +9,7 @@ class EntityManager: self.player_entity_name = "" self.map_manager = map_manager - def register_entity(self, name: str): + def register_entity(self, name: str) -> Entity: """Crée une entité et l'enregistre dans un dictionnaire.""" entity = Entity(name) self.entities[name] = entity @@ -39,10 +39,10 @@ class EntityManager: else: player.link_animation("player_none") - def get_all_entities(self): + def get_all_entities(self) -> list[Entity]: """Donne la liste de toutes les entités enregistrées.""" return list(self.entities.values()) - def get_by_name(self, name: str): + def get_by_name(self, name: str) -> Entity: """Donne l'entité avec le nom donné.""" return self.entities[name]