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.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]