Merge pull request #248 from akrherz/pirep_afos

feat: add fake_afos support for PIREP
akrherz · Jul 31, 2024 · 58892c0 · 58892c0
2 parents f84432a + 61c8d03
commit 58892c0
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 0 deletions.
diff --git a/pqact.d/pqact_afos.conf b/pqact.d/pqact_afos.conf
@@ -44,6 +44,9 @@ IDS|DDPLUS	^UR.... (KNHC|KWBC)
 # G-AIRMET (XML) for NTSB
 IDS|DDPLUS	^LW(G|H|I)E.. KKCI
 	PIPE	pywwa-parse-fake-afos-dump
+# PIREPs
+WMO ^UB
+	PIPE	pywwa-parse-fake-afos-dump
 
 IDS|DDPLUS	^CDUS27
 	PIPE	pywwa-parse-dsm2afos

diff --git a/src/pywwa/workflows/fake_afos_dump.py b/src/pywwa/workflows/fake_afos_dump.py
@@ -45,6 +45,8 @@ def compute_afos(textprod):
     ttaaii = textprod.wmo
     if ttaaii[:4] == "NOXX":
         afos = f"ADM{textprod.source[1:]}"
+    elif ttaaii.startswith("UB"):
+        afos = "PIREP"
     elif ttaaii in GMET:
         afos = GMET[ttaaii]
     elif MIS.match(ttaaii):

diff --git a/tests/workflows/test_fake_afos_dump.py b/tests/workflows/test_fake_afos_dump.py
@@ -5,11 +5,20 @@
 
 # Local
 import pywwa
+from pyiem.nws.products.pirep import Pirep
 from pyiem.util import utc
 from pywwa.testing import get_example_file
 from pywwa.workflows import fake_afos_dump
 
 
+def test_fake_pirep():
+    """Test fake PIREP."""
+    tp = Pirep(get_example_file("PIREP.txt"), utcnow=utc(2024, 7, 9, 0, 0))
+    assert tp.afos is None
+    fake_afos_dump.compute_afos(tp)
+    assert tp.afos == "PIREP"
+
+
 @pytest.mark.parametrize("database", ["afos"])
 def test_processor(cursor):
     """Test basic parsing."""