ltasg 0.0.1__py3-none-any.whl

ltasg/__init__.py

@@ -0,0 +1,5 @@
+from transport.bus import *
+from transport.train import *
+from transport.taxi import *
+from traffic import *
+

ltasg/api.py

@@ -0,0 +1,29 @@
+from aiohttp import ClientSession, ClientConnectionError
+import traceback
+class LTADataMall:
+    def __init__(self, api_key) -> None:
+        self.BASE_URL = "http://datamall2.mytransport.sg/ltaodataservice/"
+        self.API_KEY = api_key
+        self.HEADERS = {
+            "Accept": "application/json",
+            "AccountKey": self.API_KEY
+        }
+
+    async def fetch(self, service_endpoint: str, query_params: dict=None, data:dict=None):
+        url = f"{self.BASE_URL}{service_endpoint}"
+        if query_params:
+            url += "?"
+            for param, value in query_params.items():
+                url += f"{param}={value}"
+                if param != list(query_params.keys())[-1]:
+                    url += "&"
+        async with ClientSession() as session:
+            try:
+                async with session.get(url, headers=self.HEADERS) as response:
+                    if response.status == 200:
+                        return await response.json()
+                    else:
+                        raise Exception("Error calling api for {} with status code: {}".format(
+                            service_endpoint, response.status)) from None
+            except ClientConnectionError as e:
+                raise e

ltasg/constants.py

@@ -0,0 +1,36 @@
+from enum import Enum
+
+
+class TransportType(Enum):
+    BUS = "bus"
+    TRAIN = "mrt"
+    TAXI = "taxi"
+
+
+LTA_SERVICES = {
+    "PUBLIC_TRANSPORT_SERVICES": {
+        TransportType.BUS: {
+            "ARRIVAL": "BusArrivalv2",
+            "SERVICES": "BusServices",
+            "ROUTES": "BusRoutes",
+            "STOPS": "BusStops",
+            # "PASSENGER_VOLUME": {
+            #     "BUS_STOPS": "PV/Bus",
+            #     "ORIGIN_TO_DEST": "PV/ODBus"
+            # }
+        },
+        TransportType.TRAIN: {
+            # "PASSENGER_VOLUME": {
+            #     "ORIGIN_TO_DEST": "PV/ODTrain",
+            # },
+            "PLATFORM_CROWD_DENSITY": {
+                "REALTIME": "PCDRealTime",
+                "FORECAST": "PCDForecast"
+            },
+            "SERVICE_ALERTS": "TrainServiceAlerts"
+        },
+        TransportType.TAXI: {
+            "AVAILABILITY": "Taxi-Availability",
+            "STANDS": "TaxiStands",
+        }
+}}

ltasg/traffic.py

@@ -0,0 +1,2 @@
+from api import LTADataMall
+

ltasg/transport/bus.py

@@ -0,0 +1,30 @@
+from api import LTADataMall
+from transport.transport import Transport
+from constants import TransportType
+
+class Bus(Transport):
+    def __init__(self, api_key):
+        super().__init__(api_key)
+        self.bus_services = self.transport_services[TransportType.BUS]
+
+    async def arrival(self, bus_stop_code: int, bus_no: int):
+        data = await self.lta_api.fetch(self.bus_services['ARRIVAL'], query_params={
+            "BusStopCode": bus_stop_code,
+            "ServiceNo": bus_no
+        })
+        return data["Services"]
+    
+    async def services(self):
+        data = await self.lta_api.fetch(self.bus_services['SERVICES'])
+        return data["value"]
+
+    async def routes(self):
+        data = await self.lta_api.fetch(self.bus_services['ROUTES'])
+        return data["value"]
+
+    async def stops(self):
+        data = await self.lta_api.fetch(self.bus_services['STOPS'])
+        return data["value"]
+
+    
+# BusArrivalv2?BusStopCode=71111&ServiceNo=137

ltasg/transport/taxi.py

@@ -0,0 +1,16 @@
+from transport.transport import Transport
+from constants import TransportType
+
+
+class Taxi(Transport):
+    def __init__(self, api_key) -> None:
+        super().__init__(api_key)
+        self.taxi_services = self.transport_services[TransportType.TAXI]
+
+    async def availability(self):
+        data = await self.lta_api.fetch(self.taxi_services["AVAILABILITY"])
+        return data['value']
+    
+    async def stands(self):
+        data = await self.lta_api.fetch(self.taxi_services["STANDS"])
+        return data['value']

ltasg/transport/train.py

@@ -0,0 +1,19 @@
+from transport.transport import Transport
+from constants import TransportType
+from typing import Literal
+
+class Train(Transport):
+    def __init__(self, api_key) -> None:
+        super().__init__(api_key)
+        self.train_services = self.transport_services[TransportType.TRAIN]
+
+    async def service_alerts(self):
+        data = await self.lta_api.fetch(self.train_services['SERVICE_ALERTS'])
+        return data['value']
+    
+    # API currently results in Error 404 not found
+    # async def platform_crowd_density(self, type: Literal["realtime", "forecast"] = "realtime"):
+    #     if type == "realtime":
+    #         return await self.lta_api.fetch(self.train_services["PLATFORM_CROWD_DENSITY"]['REALTIME'])
+    #     else:
+    #         return await self.lta_api.fetch(self.train_services["PLATFORM_CROWD_DENSITY"]["FORECAST"])

ltasg/transport/transport.py

@@ -0,0 +1,9 @@
+from api import LTADataMall
+from constants import LTA_SERVICES
+
+
+class Transport:
+    def __init__(self, api_key) -> None:
+        self.lta_api = LTADataMall(api_key=api_key)
+        self.transport_services = LTA_SERVICES['PUBLIC_TRANSPORT_SERVICES']
+    

ltasg-0.0.1.dist-info/METADATA

@@ -0,0 +1,15 @@
+Metadata-Version: 2.3
+Name: ltasg
+Version: 0.0.1
+Summary: A LTA Datamall API Wrapper Library for Python
+Project-URL: Homepage, https://github.com/ashe/ltasg
+Project-URL: Issues, https://github.com/ashe/ltasg/issues
+Author-email: Aw Sheng Xiang <awshengxiang@hotmail.com>
+License-File: LICENSE
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+
+# ltasg

ltasg-0.0.1.dist-info/RECORD

@@ -0,0 +1,13 @@
+ltasg/__init__.py,sha256=Ad1sqqFDheWrAP47UrPJaV8wg8XPVbJqmXxSDfIyGP0,115
+ltasg/api.py,sha256=82nQORX48ac4jNxhcwXB59dghLRiqCdl1QDQUnEBQTg,1272
+ltasg/constants.py,sha256=-VPdc29e1yrdf8WfSv6SxJDAp5Dvo2Hl-B9-3Q817ms,983
+ltasg/traffic.py,sha256=THy6DQbo97N_gr2x1F9mIHXNmHTJbpApIJNPs9mk-2s,31
+ltasg/transport/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ltasg/transport/bus.py,sha256=KN1DQtd6aHniU-HI-T2WF3wvpt08JAkcooYQcfSlYMg,1000
+ltasg/transport/taxi.py,sha256=T_36NAqCDX4bMUYV_u-ublrat1GB7w6Dwpz3IweQfJE,537
+ltasg/transport/train.py,sha256=H1m3uqteT0GQh9U-M4Mth2kemeiq40ilMDpYmC5ZkIc,855
+ltasg/transport/transport.py,sha256=MVp9aZFnYQupbbQZRPgpqkzpJv9pUqEsPc6PKR8H_D0,263
+ltasg-0.0.1.dist-info/METADATA,sha256=aV61APtMcZEpkIJedAQjx6w_tdXGqWcadMtz4Z-7JhU,510
+ltasg-0.0.1.dist-info/WHEEL,sha256=zEMcRr9Kr03x1ozGwg5v9NQBKn3kndp6LSoSlVg-jhU,87
+ltasg-0.0.1.dist-info/licenses/LICENSE,sha256=Ek8jf_lINFv3z5tUpSKltxgG4kO149536yWu352HCZU,1082
+ltasg-0.0.1.dist-info/RECORD,,

ltasg-0.0.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.24.2
+Root-Is-Purelib: true
+Tag: py3-none-any

ltasg-0.0.1.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Ashe
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.