migration to fast-api

mta_manager/feed.py

@@ -1,5 +1,6 @@
 from enum import Enum
 
+
 class Feed(Enum):
     ACE = "https://api-endpoint.mta.info/Dataservice/mtagtfsfeeds/nyct%2Fgtfs-ace"
     BDFM = "https://api-endpoint.mta.info/Dataservice/mtagtfsfeeds/nyct%2Fgtfs-bdfm"
@@ -10,6 +11,7 @@ class Feed(Enum):
     N1234567 = "https://api-endpoint.mta.info/Dataservice/mtagtfsfeeds/nyct%2Fgtfs"
     SIR = "https://api-endpoint.mta.info/Dataservice/mtagtfsfeeds/nyct%2Fgtfs-si"
 
+
 ALL_FEEDS = [
     Feed.ACE,
     Feed.BDFM,
@@ -18,4 +20,4 @@ ALL_FEEDS = [
     Feed.L,
     Feed.N1234567,
     Feed.SIR
-]
+]

mta_manager/route.py

@@ -1,5 +1,6 @@
 from enum import Enum
 
+
 class Route(Enum):
     A = "A"
     C = "C"
@@ -31,6 +32,9 @@ class Route(Enum):
     L = "L"
     SIR = "SIR"
 
+
 _routes = set(item.value for item in Route)
+
+
 def is_valid_route(route: str) -> bool:
-    return route in _routes
+    return route in _routes

mta_manager/stop.py

@@ -5,24 +5,3 @@ from math import trunc
 
 def trip_arrival_in_minutes(stop_time_update: gtfs_realtime_pb2.TripUpdate):
     return trunc(((datetime.fromtimestamp(stop_time_update.arrival.time) - datetime.now()).total_seconds()) / 60)
-
-# class Stop(object):
-#     def __init__(self, id, arrival_time, departure_time, ):
-#         self.id = id
-#         self.arrival_time = arrival_time
-#         self.departure_time = departure_time
-#
-#     def arrival_minutes(self):
-#         return trunc(((datetime.fromtimestamp(self.arrival_time) - datetime.now()).total_seconds()) / 60)
-#
-#     def __str__(self):
-#         now = datetime.now()
-#         time = datetime.fromtimestamp(self.arrival_time)
-#         time_minutes = trunc(((time - now).total_seconds()) / 60)
-#         return f"stop_id:{self.id}| arr:{time_minutes}| dep:{self.departure_time}"
-#
-#     @staticmethod
-#     def get_stop_from_dict(obj):
-#         if "arrival" in obj and "departure" in obj and "stop_id" in obj:
-#             return Stop(obj["stop_id"], obj["arrival"]["time"], obj["departure"]["time"])
-#         return None

mta_manager/train.py

@@ -19,11 +19,11 @@ class Train(object):
                 return trip_arrival_in_minutes(stop_time_update)
         return None
 
-
     def _get_route(self) -> str:
         return self.train_proto.trip_update.trip.route_id
+
     def get_route(self) -> Route:
-       return Route(self.train_proto.trip_update.trip.route_id)
+        return Route(self.train_proto.trip_update.trip.route_id)
 
     def has_trips(self) -> bool:
         return self.train_proto.trip_update is not None \
@@ -31,4 +31,3 @@ class Train(object):
 
     def __str__(self):
         return f"{self.train_proto}"
-