new final() method, mirror of next()

pyproject.toml

@@ -2,7 +2,7 @@
 name = "traffic"
 homepage = "https://github.com/xoolive/traffic/"
 documentation = "https://traffic-viz.github.io/"
-version = "2.6.3"
+version = "2.6.4"
 description = "A toolbox for manipulating and analysing air traffic data"
 authors = ["Xavier Olive <git@xoolive.org>"]
 license = "MIT"

tests/test_flight.py

@@ -270,6 +270,10 @@ def test_landing_ils() -> None:
     assert aligned is not None
     assert aligned.max("ILS") == "06"
 
+    aligned = belevingsvlucht.final("aligned_on_EHLE")
+    assert aligned is not None
+    assert aligned.ILS_max == "23"
+
     aligned = airbus_tree.aligned_on_ils("EDHI").next()  # noqa: B305
     assert aligned is not None
     assert aligned.max("ILS") == "23"

traffic/core/flight.py

@@ -411,6 +411,27 @@ def next(
         )
         return next(fun(self), None)
 
+    def final(
+        self,
+        method: Union[str, Callable[["Flight"], Iterator["Flight"]]],
+    ) -> Optional["Flight"]:
+        """
+        Returns the first segment of trajectory yielded by flight.method()
+
+        >>> flight.final("go_around")
+        >>> flight.final("runway_change")
+        >>> flight.final(lambda f: f.aligned_on_ils("LFBO"))
+        """
+        fun = (
+            getattr(self.__class__, method)
+            if isinstance(method, str)
+            else method
+        )
+        segment = None
+        for segment in fun(self):
+            continue
+        return segment
+
     # --- Iterators ---
 
     @property

traffic/core/iterator.py

@@ -132,6 +132,22 @@ def next(self) -> Optional["Flight"]:
         """
         return next((segment for segment in self), None)
 
+    def final(self) -> Optional["Flight"]:
+        """Returns the final (last) element in the FlightIterator.
+
+        Example usage:
+
+        >>> first_attempt = flight.runway_change().final()
+
+        This is equivalent to:
+
+        >>> flight.final("runway_change")
+        """
+        segment = None
+        for segment in self:
+            continue
+        return segment
+
     def sum(self) -> int:
         """Returns the size of the FlightIterator.