From bdbc04714adac90a7137d69c98791f3c388b221f Mon Sep 17 00:00:00 2001 From: Ernesto Ruge Date: Fri, 24 Jan 2025 09:46:33 +0100 Subject: [PATCH] better realtime output --- webapp/models/source.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/webapp/models/source.py b/webapp/models/source.py index 4bcbb1a..8a48769 100644 --- a/webapp/models/source.py +++ b/webapp/models/source.py @@ -64,6 +64,14 @@ class Source(BaseModel): static_parking_site_error_count: Mapped[int] = mapped_column(Integer(), nullable=False, default=0) realtime_parking_site_error_count: Mapped[int] = mapped_column(Integer(), nullable=False, default=0) + def to_dict(self, *args, ignore: Optional[list[str]] = None, **kwargs) -> dict: + ignore = ignore or [] + if self.static_status in [SourceStatus.PROVISIONED, SourceStatus.DISABLED]: + ignore += ['static_data_updated_at', 'static_parking_site_error_count'] + if self.realtime_status in [SourceStatus.PROVISIONED, SourceStatus.DISABLED]: + ignore += ['realtime_data_updated_at', 'realtime_parking_site_error_count'] + return super().to_dict(*args, ignore=ignore, **kwargs) + @property def combined_status(self) -> SourceStatus: if self.static_status != SourceStatus.ACTIVE or self.realtime_status in [