def test_sanitize_report_string(self):
"""
Tests a function which fixes common mistakes while the report is a string
"""
line = "KJFK 36010 ? TSFEW004SCT012FEW///CBBKN080 C A V O K A2992"
fixed = "KJFK 36010 TS FEW004 SCT012 FEW///CB BKN080 CAVOK A2992"
self.assertEqual(_core.sanitize_report_string(line), fixed)
def parse(report: str) -> PirepData:
"""
Returns a PirepData object based on the given report
"""
if not report:
return None
clean = _core.sanitize_report_string(report)
wxdata, *_ = _core.sanitize_report_list(clean.split())
sanitized = " ".join(wxdata)
wxresp = {
"raw": report,
"sanitized": sanitized,
"station": None,
"remarks": None
}
wxdata = sanitized.split("/")
wxresp.update(_root(wxdata.pop(0).strip()))
for item in wxdata:
if not item or len(item) < 2:
continue
tag = item[:2]
item = item[2:].strip()
if tag in _handlers:
key, handler = _handlers[tag]
wxresp[key] = handler(item)
elif tag in _dict_handlers:
wxresp.update(_dict_handlers[tag](item))
return PirepData(**wxresp)
def parse_in(report: str) -> (MetarData, Units):
"""
Parser for the International METAR variant
"""
units = Units(**IN_UNITS)
wxresp = {"raw": report}
clean = _core.sanitize_report_string(report)
wxdata, wxresp["remarks"] = _core.get_remarks(clean)
wxdata = _core.dedupe(wxdata)
wxdata = _core.sanitize_report_list(wxdata)
wxresp["sanitized"] = " ".join(wxdata + [wxresp["remarks"]])
wxdata, wxresp["station"], wxresp["time"] = _core.get_station_and_time(wxdata)
wxdata, wxresp["runway_visibility"] = _core.get_runway_visibility(wxdata)
if "CAVOK" not in wxdata:
wxdata, wxresp["clouds"] = _core.get_clouds(wxdata)
wxdata, wxresp["wind_direction"], wxresp["wind_speed"], wxresp["wind_gust"], wxresp[
"wind_variable_direction"
] = _core.get_wind(wxdata, units)
wxdata, wxresp["altimeter"] = _core.get_altimeter(wxdata, units, "IN")
if "CAVOK" in wxdata:
wxresp["visibility"] = _core.make_number("CAVOK")
wxresp["clouds"] = []
wxdata.remove("CAVOK")
else:
wxdata, wxresp["visibility"] = _core.get_visibility(wxdata, units)
wxresp["other"], wxresp["temperature"], wxresp["dewpoint"] = _core.get_temp_and_dew(
wxdata
)
condition = _core.get_flight_rules(
wxresp["visibility"], _core.get_ceiling(wxresp["clouds"])
)
wxresp["flight_rules"] = FLIGHT_RULES[condition]
wxresp["remarks_info"] = remarks.parse(wxresp["remarks"])
wxresp["time"] = _core.make_timestamp(wxresp["time"])
return MetarData(**wxresp), units
def parse(station: str, report: str) -> (TafData, Units):
"""
Returns TafData and Units dataclasses with parsed data and their associated units
"""
if not report:
return None, None
valid_station(station)
while len(report) > 3 and report[:4] in ("TAF ", "AMD ", "COR "):
report = report[4:]
retwx = {
"end_time": None,
"raw": report,
"remarks": None,
"start_time": None
}
report = _core.sanitize_report_string(report)
_, station, time = _core.get_station_and_time(report[:20].split())
retwx["station"] = station
retwx["time"] = _core.make_timestamp(time)
report = report.replace(station, "")
if time:
report = report.replace(time, "").strip()
if uses_na_format(station):
use_na = True
units = Units(**NA_UNITS)
else:
use_na = False
units = Units(**IN_UNITS)
# Find and remove remarks
report, retwx["remarks"] = _core.get_taf_remarks(report)
# Split and parse each line
lines = _core.split_taf(report)
parsed_lines = parse_lines(lines, units, use_na)
# Perform additional info extract and corrections
if parsed_lines:
parsed_lines[-1]["other"], retwx["max_temp"], retwx[
"min_temp"] = _core.get_temp_min_and_max(parsed_lines[-1]["other"])
if not (retwx["max_temp"] or retwx["min_temp"]):
parsed_lines[0]["other"], retwx["max_temp"], retwx[
"min_temp"] = _core.get_temp_min_and_max(
parsed_lines[0]["other"])
# Set start and end times based on the first line
start, end = parsed_lines[0]["start_time"], parsed_lines[0]["end_time"]
parsed_lines[0]["end_time"] = None
retwx["start_time"], retwx["end_time"] = start, end
parsed_lines = _core.find_missing_taf_times(parsed_lines, start, end)
parsed_lines = _core.get_taf_flight_rules(parsed_lines)
# Extract Oceania-specific data
if retwx["station"][0] == "A":
parsed_lines[-1]["other"], retwx["alts"], retwx[
"temps"] = _core.get_oceania_temp_and_alt(
parsed_lines[-1]["other"])
# Convert to dataclass
retwx["forecast"] = [TafLineData(**line) for line in parsed_lines]
return TafData(**retwx), units
def parse(report: str) -> AirepData:
"""
"""
if not report:
return None
clean = _core.sanitize_report_string(report)
wxdata, *_ = _core.sanitize_report_list(clean.split())
wxresp = {"raw": report, "sanitized": " ".join(wxdata)}
print(wxdata)
print(wxresp)
return None
