42Cummer commited on
Commit
b8eb441
·
verified ·
1 Parent(s): 22c1b41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -149,6 +149,12 @@ def seek():
149
  vehicle_link = p_tag.find('a', href="#MapMain")
150
  if vehicle_link and vehicle_link.text.strip().isdigit():
151
  vehicle_data['vehicle_number'] = vehicle_link.text.strip()
 
 
 
 
 
 
152
 
153
  # Get stop location description
154
  text_nodes = [t for t in p_tag.stripped_strings]
 
149
  vehicle_link = p_tag.find('a', href="#MapMain")
150
  if vehicle_link and vehicle_link.text.strip().isdigit():
151
  vehicle_data['vehicle_number'] = vehicle_link.text.strip()
152
+ else:
153
+ # Case 2: Vehicle number is in plain text like "Vehicle 3417 Load"
154
+ text = p_tag.get_text()
155
+ match = re.search(r'Vehicle (\d+) Load', text)
156
+ if match:
157
+ vehicle_data['vehicle_number'] = match.group(1)
158
 
159
  # Get stop location description
160
  text_nodes = [t for t in p_tag.stripped_strings]