From cc010b210a1db61653b7530ef9ac4fb54aeb514f Mon Sep 17 00:00:00 2001 From: Mike Lang Date: Thu, 12 Oct 2023 22:44:52 +1100 Subject: [PATCH] Fix streamlog formatting --- sheetsync/sheetsync/streamlog.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/sheetsync/sheetsync/streamlog.py b/sheetsync/sheetsync/streamlog.py index ee5e338..45e6152 100644 --- a/sheetsync/sheetsync/streamlog.py +++ b/sheetsync/sheetsync/streamlog.py @@ -61,7 +61,7 @@ class StreamLogMiddleware: # Omitted columns act as the identity function. self.column_decode = { 'event_start': parse_utc_only, - 'event_end': parse_utc_only, + 'event_end': lambda v: None if v["type"] == "NoTime" else parse_utc_only(v["time"]), 'category': lambda v: v["name"], 'image_links': lambda v: [link.strip() for link in v.split()] if v.strip() else [], 'state': lambda v: v.upper(), @@ -73,7 +73,7 @@ class StreamLogMiddleware: } def get_rows(self): - for row in self.client.get_rows(): + for row in self.client.get_rows()["event_log"]: yield self.parse_row(row) def parse_row(self, row):