feat: check get_all and get_latest methods return next_check_at

datagouv · Nov 22, 2024 · 2fd106a · 2fd106a
1 parent d58d4af
commit 2fd106a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/udata_hydra/db/check.py b/udata_hydra/db/check.py
@@ -61,7 +61,7 @@ async def get_latest(
         async with pool.acquire() as connection:
             q = f"""
             SELECT catalog.id as catalog_id, checks.id as check_id,
-                catalog.status as catalog_status, checks.status as check_status, catalog.deleted as deleted, *
+                catalog.status as catalog_status, checks.status as check_status, checks.next_check_at as next_check_at, catalog.deleted as deleted, *
             FROM checks, catalog
             WHERE catalog.{column} = $1
             AND checks.id = catalog.last_check
@@ -75,7 +75,7 @@ async def get_all(cls, url: str | None = None, resource_id: str | None = None) -
         async with pool.acquire() as connection:
             q = f"""
             SELECT catalog.id as catalog_id, checks.id as check_id,
-                catalog.status as catalog_status, checks.status as check_status, catalog.deleted as deleted, *
+                catalog.status as catalog_status, checks.status as check_status, checks.next_check_at as next_check_at, catalog.deleted as deleted, *
             FROM checks, catalog
             WHERE catalog.{column} = $1
             AND catalog.{column} = checks.{column}

diff --git a/udata_hydra/schemas/check.py b/udata_hydra/schemas/check.py
@@ -16,6 +16,7 @@ class CheckSchema(Schema):
     error = fields.Str()
     dataset_id = fields.Str()
     resource_id = fields.UUID()
+    next_check_at = fields.DateTime()
     deleted = fields.Boolean()
     parsing_started_at = fields.DateTime()
     parsing_finished_at = fields.DateTime()