From e805a38362e7f6beaae3074b0367577ce91f7c31 Mon Sep 17 00:00:00 2001 From: Stagnant <82290573+iStagnant@users.noreply.github.com> Date: Fri, 25 Aug 2023 13:36:56 +0000 Subject: [PATCH] Added planets function --- endlessparser/datatypes/Map.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/endlessparser/datatypes/Map.py b/endlessparser/datatypes/Map.py index ac7e677..1fdb28d 100644 --- a/endlessparser/datatypes/Map.py +++ b/endlessparser/datatypes/Map.py @@ -61,6 +61,14 @@ def haze(self) -> Optional[str]: def links(self) -> List[str]: return [child.tokens_as_string() for child in self._find_children("link")] + def planets(self) -> List[str]: + l = [] + for object in self.objects(): + if object.tokens: l.append(object.tokens_as_string()) + for o in object.objects(): #To include stations + if o.tokens: l.append(o.tokens_as_string()) + return l + def minables(self) -> Dict[str, Tuple[float, float]]: d = {} for child in self._find_children("minables"):