Ajout de typage dans le entity_manager

This commit is contained in:
Yannis 2024-01-02 15:04:46 +01:00
parent 6ddf98db4a
commit 8ea1c1261e
Signed by: yannis
SSH key fingerprint: SHA256:Bz8K8QiTYUudf8MlthTM9MCLfgiYf/U1md3V9g9Wo14

View file

@ -9,7 +9,7 @@ class EntityManager:
self.player_entity_name = "" self.player_entity_name = ""
self.map_manager = map_manager 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.""" """Crée une entité et l'enregistre dans un dictionnaire."""
entity = Entity(name) entity = Entity(name)
self.entities[name] = entity self.entities[name] = entity
@ -39,10 +39,10 @@ class EntityManager:
else: else:
player.link_animation("player_none") 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.""" """Donne la liste de toutes les entités enregistrées."""
return list(self.entities.values()) 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é.""" """Donne l'entité avec le nom donné."""
return self.entities[name] return self.entities[name]