[server]: Add additional fields to fetch_transient_data

This commit is contained in:
2025-03-24 18:00:01 +00:00
parent e7af6295fe
commit c6282aca14
2 changed files with 11 additions and 5 deletions

View File

@ -46,6 +46,10 @@ def fetch_trains():
train_status = train["TrainStatus"]
public_message = train["PublicMessage"]
split_message = public_message.split("\\n")
trainDetails = split_message[1].split("(")[0]
trainUpdate = split_message[2]
# Regex to extract punctuality: Matches positive/negative number followed by "mins late"
match = re.search(r"(-?\d+)\s+mins\s+late", public_message)
punctuality = int(match.group(1)) if match else 0 # Default to 0 if no match
@ -88,7 +92,9 @@ def fetch_trains():
"trainDirection": train["Direction"],
"trainPunctuality": punctuality,
"trainPunctualityStatus": punctuality_status,
"latenessMessage": lateness_message
"latenessMessage": lateness_message,
"trainDetails": trainDetails,
"trainUpdate": trainUpdate
})
return trains