CLI: Colored live output

locomotive/cli/commands/live.py

@@ -8,6 +8,7 @@
 import dateparser
 import requests
 import tableformatter as tf
+from colored import bg
 
 from ...api.gc import BoardRequest, Client
 from ...models import BoardEntry, Station
@@ -18,7 +19,7 @@ def get_rows(entries: List[BoardEntry]) -> List[tuple]:
         (
             x.tofrom.name,
             f"{x.transport.label} {x.transport.number}",
-            x.time,
+            x.time.strftime("%H:%M"),
             f"{x.delay} min",
             x.platform,
         )
@@ -34,6 +35,7 @@ def get_rows(entries: List[BoardEntry]) -> List[tuple]:
 #    default="pretty",
 #    help="Output format.",
 # )
+@click.option("--color/--no-color", default=False)
 @click.pass_context
 def live(ctx: click.Context, **args: str) -> None:
     """
@@ -48,10 +50,16 @@ def live(ctx: click.Context, **args: str) -> None:
 
     client = Client(stations)
 
+    dep_gs = None
+    arr_gs = None
+    if args["color"]:
+        dep_gs = tf.AlternatingRowGrid(bg(25), bg(18))
+        arr_gs = tf.AlternatingRowGrid(bg(34), bg(28))
+
     departures = client.board_request(BoardRequest(station, "departure"))
     cols = ["Destination", "Number", "Time", "Delay", "Platform"]
-    click.echo(tf.generate_table(get_rows(departures), cols))
+    click.echo(tf.generate_table(get_rows(departures), cols, grid_style=dep_gs))
 
     arrivals = client.board_request(BoardRequest(station, "arrival"))
     cols = ["Origin", "Number", "Time", "Delay", "Platform"]
-    click.echo(tf.generate_table(get_rows(arrivals), cols))
+    click.echo(tf.generate_table(get_rows(arrivals), cols, grid_style=arr_gs))

pyproject.toml

@@ -32,6 +32,7 @@ requests = "^2.20"
 tableformatter = "^0.1.4"
 text-unidecode = "^1.3"
 py-money = "^0.4.0"
+colored = "^1.4.2"
 
 [tool.poetry.dev-dependencies]
 # https://github.com/python-poetry/poetry/issues/649