import re from metar_taf_parser.model.enum import CloudQuantity, Phenomenon def get_visibility_cat(input_string): v = 0 if input_string == '> 10km': v = 10000 else: v = int(re.sub(r'\D', '', input_string)) if v >= 5000: return 1 elif v >= 1500: return 2 else: return 3 def round_angle(angle): if angle == None: return -1 if not isinstance(angle, int): angle = int(angle) if angle < 0 or angle > 360: raise ValueError("Angle must be between 0 and 359 degrees") rounded_angle = round(angle / 30) * 30 if rounded_angle == 360: rounded_angle = 0 return int(rounded_angle) def get_ceiling_cat(clouds): cat = 0 if clouds is None: return cat for i in range(3): if i < len(clouds): c = clouds[i] cloud_quantity = c.quantity if cloud_quantity == CloudQuantity.BKN or cloud_quantity == CloudQuantity.OVC: if c.height > 1500: cat = 1 else: cat = 2 return cat def cloud_height_to_int(cloud_height): if cloud_height == None: return -1 else: return int(cloud_height) def phenomenon_to_int(p): match p: case Phenomenon.RAIN: return 0 case Phenomenon.DRIZZLE: return 1 case Phenomenon.SNOW: return 2 case Phenomenon.SNOW_GRAINS: return 3 case Phenomenon.ICE_PELLETS: return 4 case Phenomenon.ICE_CRYSTALS: return 5 case Phenomenon.HAIL: return 6 case Phenomenon.SMALL_HAIL: return 7 case Phenomenon.UNKNOW_PRECIPITATION: return 8 case Phenomenon.FOG: return 9 case Phenomenon.VOLCANIC_ASH: return 10 case Phenomenon.MIST: return 11 case Phenomenon.HAZE: return 12 case Phenomenon.WIDESPREAD_DUST: return 13 case Phenomenon.SMOKE: return 14 case Phenomenon.SAND: return 15 case Phenomenon.SPRAY: return 16 case Phenomenon.SQUALL: return 17 case Phenomenon.SAND_WHIRLS: return 18 case Phenomenon.DUSTSTORM: return 19 case Phenomenon.SANDSTORM: return 20 case Phenomenon.FUNNEL_CLOUD: return 21 return -1 class WeatherReport: def __init__(self, time, visibility_distance, clouds, weather_conditions, trends, interval): # time self.time = time # interval if interval == None: if self.time.hour == int_1[1]: self.interval = 1 elif self.time.hour == int_2[1]: self.interval = 2 elif self.time.hour == int_3[1]: self.interval = 3 else: self.interval = interval # visbility self.visibility_cat = get_visibility_cat(visibility_distance) # clouds self.ceiling_cat = get_ceiling_cat(clouds) # phenomenons if len(weather_conditions) > 0: self.has_weather = 1 else: self.has_weather = 0 # trends if trends is not None and len(trends) > 0: for trend in trends: start_day = trend.validity.start_day end_day = trend.validity.end_day start_hour = trend.validity.start_hour end_hour = trend.validity.end_hour pred = pd.to_datetime(prediction_date, format="%d/%m/%Y").day ok = False #print(f"PRED_DAY = {pred}, START_DAY = {start_day}, END_DAY = {end_day}, START_HOUR = {start_hour}, END_HOUR = {end_hour}, INTERVAL = {interval}") if start_day == pred and end_day == pred: if interval == 1 and (int_1[0] < start_hour <= int_1[2] or int_1[0] < end_hour <= int_1[2]): #print("TREND FOR 1") ok = True if interval == 2 and (int_2[0] < start_hour <= int_2[2] or int_2[0] < end_hour <= int_2[2]): #print("TREND FOR 2") ok = True if interval == 3 and (int_3[0] < start_hour <= int_3[2] or int_3[0] < end_hour <= int_3[2]): #print("TREND FOR 3") ok = True elif start_day == pred and end_day != pred: if interval == 1 and (int_1[0] < start_hour <= int_1[2]): #print("TREND FOR 1") ok = True if interval == 2 and (int_2[0] < start_hour <= int_2[2]): #print("TREND FOR 2") ok = True if interval == 3 and (int_3[0] < start_hour <= int_3[2]): #print("TREND FOR 3") ok = True if ok: if trend.visibility is not None: self.visibility_cat = max(get_visibility_cat(trend.visibility.distance), self.visibility_cat) self.ceiling_cat = max(get_ceiling_cat(trend.clouds), self.ceiling_cat) if len(trend.weather_conditions) > 0: self.has_weather = 1 def get_date_and_interval(self): day = self.time.day if day < 10: day = f"0{day}" month = self.time.month if month < 10: month = f"0{month}" year = self.time.year interval = self.interval r = f"{day}/{month}/{year} - {interval}" return r def print_info(self): print(f"{self.time} [int: {self.interval}]") print(f"\tVISIBILITY CAT. : {self.visibility_cat}") print(f"\tCEILING CAT. : {self.ceiling_cat}") print(f"\tHAS WEATHER: {self.has_weather}")