Update

app.py

@@ -4,13 +4,16 @@
 # Date: 2025/03/22
 
 import os
+import re
 import json
 import requests
 import streamlit as st
 from datetime import datetime
+from bs4 import BeautifulSoup
 
 nextbus_token = os.getenv("NEXTBUS_TOKEN")
 datamall_token = os.getenv("DATAMALL_TOKEN")
+mrt_token = os.getenv("MRT_TOKEN")
 
 def wide_space_default():
     st.set_page_config(layout='wide')
@@ -62,6 +65,53 @@ def get_lta_bus_arrival(bus_stop_code):
     response = requests.request("GET", url, headers=headers, data=payload)
     return response.json()
 
+def get_SMRT_train_arrival(station_code):
+    url = "https://trainarrivalweb.smrt.com.sg/"
+
+    payload = f"ScriptManager1=UP1%7CddlStation&stnCode=&stnName=&ddlStation={station_code}&{mrt_token}"
+    headers = {
+        'Accept': '*/*',
+        'Accept-Language': 'en-GB,en-US;q=0.9,en;q=0.8,zh-CN;q=0.7,zh-HK;q=0.6,zh-TW;q=0.5,zh;q=0.4',
+        'Cache-Control': 'no-cache',
+        'Connection': 'keep-alive',
+        'Content-Type': 'application/x-www-form-urlencoded; charset=UTF-8',
+        'Origin': 'https://trainarrivalweb.smrt.com.sg',
+        'Referer': 'https://trainarrivalweb.smrt.com.sg/',
+        'Sec-Fetch-Dest': 'empty',
+        'Sec-Fetch-Mode': 'cors',
+        'Sec-Fetch-Site': 'same-origin',
+        'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/134.0.0.0 Safari/537.36',
+        'X-MicrosoftAjax': 'Delta=true',
+        'X-Requested-With': 'XMLHttpRequest',
+        'sec-ch-ua': '"Chromium";v="134", "Not:A-Brand";v="24", "Google Chrome";v="134"',
+        'sec-ch-ua-mobile': '?0',
+        'sec-ch-ua-platform': '"macOS"',
+        'Cookie': 'ASP.NET_SessionId=bwmeirfwzq30nj0ddk0qzyav; _ga=GA1.1.974289719.1742578510; _ga_78C66TP2RM=GS1.1.1742578509.1.1.1742578584.0.0.0'
+    }
+    response = requests.request("POST", url, headers=headers, data=payload)
+    raw_html = response.text    
+    soup = BeautifulSoup(raw_html, 'html.parser')
+
+    tables = soup.find_all("table", id="gvTime")
+    
+    trains = list()
+    
+    for table in tables:
+        time_row = table.find_all("tr")[1]
+        time_cells = [td.get_text(strip=True) for td in time_row.find_all("td")]
+
+        direction_row = table.find_all("tr")[2]
+        direction_cells = [td.get_text(strip=True) for td in direction_row.find_all("td")]
+        
+        for eta_text, direction in zip(time_cells, direction_cells):
+            eta = re.findall(r'\d+', eta_text)[0]
+            trains.append({
+                "direction": direction,
+                "eta": eta
+            })
+    
+    return trains
+
 
 # stops = get_all_stops()
 nus_stops = [
@@ -118,6 +168,13 @@ public_stops = [
     },
 ]
 
+public_mrt_station = [
+    {
+        'name': "KR MRT",
+        'code': 'CKRG'
+    }
+]
+
 st.write(
     """
     <style>
@@ -189,3 +246,22 @@ for stop_info in public_stops:
         for i, bus in enumerate(buses[:5]):
             cols[i+1].metric(bus["type"], bus['service'], bus["eta"])
 
+# SMRT
+for station in public_mrt_station:
+    smrt_data = get_SMRT_train_arrival(station['code'])
+    trains = list()
+    
+    for train in smrt_data:
+        trains.append({
+            "direction": train["direction"],
+            "eta": train["eta"]
+        })
+        
+    trains.sort(key=lambda x: x["eta"])
+    
+with st.container(border=True):
+    cols = st.columns(6)
+    cols[0].metric("MRT Station", station['name'])
+    
+    for i, train in enumerate(trains[:5]):
+        cols[i+1].metric(train['direction'], 'CC', str(train['eta']))

requirments.txt

@@ -0,0 +1 @@
+BeautifulSoup