xan/xmet
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Allow fetching soundings by station ID

Browse source
This commit is contained in:
XANTRONIX 2025-03-29 18:49:08 -04:00
parent 1ddc343487
commit cda1413f2c
1 changed files with 12 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
lib/xmet/raob.py
View file

@ -572,16 +572,13 @@ class RAOBReader():
self.soundings = dict()
def sounding(self, station: str, timestamp: str) -> RAOBSounding:
key = station + '.' + timestamp
if station not in self.soundings:
self.soundings[station] = dict()
if key in self.soundings:
return self.soundings[key]
if timestamp not in self.soundings[station]:
self.soundings[station][timestamp] = RAOBSounding(station, timestamp)
sounding = RAOBSounding(station, timestamp)
self.soundings[key] = sounding
return sounding
return self.soundings[station][timestamp]
def parse_chunk(self, text: str) -> RAOBChunk:
meta = {
@ -667,12 +664,17 @@ class RAOBReader():
data.get('wind_speed'),
data.get('wind_dir'))
for key in self.soundings:
sounding = self.soundings[key].finish()
def each_sounding_by_station(self, station: str):
for timestamp in self.soundings[station]:
sounding = self.soundings[station][timestamp].finish()
if sounding is not None:
yield sounding
def each_sounding(self):
for station in self.soundings:
yield from self.each_sounding_by_station(station)
@staticmethod
def each_sounding_from_fh(fh: io.TextIOBase):
reader = RAOBReader(fh)