add top-level iterable

andrewgryan · Nov 12, 2024 · c65587c · c65587c
1 parent a2d56af
commit c65587c
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/src/detaf/lib.py b/src/detaf/lib.py
@@ -103,6 +103,16 @@ def taf_encode(self) -> str:
             parts.append(weather.taf_encode())
         return " ".join(parts)
 
+    def __iter__(self):
+        yield "TAF"
+        if self.version != Version.ORIGINAL:
+            yield self.version
+        if self.icao_identifier:
+            yield self.icao_identifier
+        if self.issue_time:
+            yield self.issue_time
+        yield from self.weather_conditions
+
 
 def parse(bulletin: str) -> TAF:
     words = bulletin.strip().split()
@@ -289,6 +299,8 @@ def encode(item) -> str:
         return encode_issue_time(item)
     elif isinstance(item, period):
         return encode_period(item)
+    elif isinstance(item, str):
+        return item
     else:
         return item.taf_encode()
 

diff --git a/tests/test_interface.py b/tests/test_interface.py
@@ -211,3 +211,7 @@ def test_parse_wx():
     assert wx.parse("-DZ") == wx.Wx(
         intensity=wx.Intensity.LIGHT, precipitation=wx.Precipitation.DRIZZLE
     )
+
+
+def test_iterable():
+    assert list(detaf.decode("TAF COR")) == ["TAF", detaf.Version.CORRECTED]