interloper-api 0.61.0__tar.gz → 0.63.0__tar.gz

{interloper_api-0.61.0 → interloper_api-0.63.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: interloper-api
-Version: 0.61.0
+Version: 0.63.0
 Summary: Interloper FastAPI routes
 Author: Guillaume Onfroy
 Author-email: Guillaume Onfroy <guillaume@digitlcloud.com>

{interloper_api-0.61.0 → interloper_api-0.63.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "interloper-api"
-version = "0.61.0"
+version = "0.63.0"
 description = "Interloper FastAPI routes"
 readme = "README.md"
 requires-python = ">=3.10"

{interloper_api-0.61.0 → interloper_api-0.63.0}/pyproject.toml.orig

@@ -3,7 +3,7 @@
 # ###############
 [project]
 name = "interloper-api"
-version = "0.61.0"
+version = "0.63.0"
 description = "Interloper FastAPI routes"
 readme = "README.md"
 authors = [{ name = "Guillaume Onfroy", email = "guillaume@digitlcloud.com" }]

{interloper_api-0.61.0 → interloper_api-0.63.0}/src/interloper_api/routes/backfills.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import datetime as dt
 from uuid import UUID
 
 from fastapi import APIRouter, Depends, HTTPException
@@ -27,8 +26,8 @@ class BackfillCreateRequest(BaseModel):
     """Request body for queuing a backfill."""
 
     component_id: UUID
-    start_date: dt.date
-    end_date: dt.date
+    start_key: str
+    end_key: str
     concurrency: int = 1
     fail_fast: bool = False
 
@@ -40,8 +39,8 @@ class BackfillResponse(BaseModel):
     org_id: UUID
     component_id: UUID | None
     status: str
-    start_date: dt.date
-    end_date: dt.date
+    start_key: str
+    end_key: str
     concurrency: int
     fail_fast: bool
     partitions: int
@@ -64,8 +63,8 @@ def _backfill_to_response(backfill: Backfill) -> BackfillResponse:
         org_id=backfill.org_id,
         component_id=backfill.component_id,
         status=backfill.status,
-        start_date=backfill.start_date,
-        end_date=backfill.end_date,
+        start_key=backfill.start_key,
+        end_key=backfill.end_key,
         concurrency=backfill.concurrency,
         fail_fast=backfill.fail_fast,
         partitions=backfill.partitions,
@@ -96,18 +95,21 @@ def create_backfill(
     user: Profile = Depends(get_current_user),
     store: Store = Depends(get_store),
 ) -> BackfillResponse:
-    """Queue a backfill for a job over a date range."""
+    """Queue a backfill for a job over a partition-key range."""
     job = load_authorized(
         lambda i: store.get_component(i, kind="job"), body.component_id, user, store, label="Job", minimum="editor"
     )
-    backfill = store.create_backfill(
-        job.org_id,
-        component_id=body.component_id,
-        start_date=body.start_date,
-        end_date=body.end_date,
-        concurrency=body.concurrency,
-        fail_fast=body.fail_fast,
-    )
+    try:
+        backfill = store.create_backfill(
+            job.org_id,
+            component_id=body.component_id,
+            start_key=body.start_key,
+            end_key=body.end_key,
+            concurrency=body.concurrency,
+            fail_fast=body.fail_fast,
+        )
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
     return _backfill_to_response(backfill)
 
 

{interloper_api-0.61.0 → interloper_api-0.63.0}/src/interloper_api/routes/runs.py

@@ -37,7 +37,7 @@ class RunResponse(BaseModel):
     org_id: UUID
     component_id: UUID | None
     backfill_id: UUID | None
-    partition_date: dt.date | None
+    partition_key: str | None
     status: str
     retry_of: UUID | None = None
     attempt: int = 1
@@ -51,7 +51,7 @@ class RunCreateRequest(BaseModel):
     """Request body for queuing a run targeting a runnable component."""
 
     component_id: UUID
-    partition_date: dt.date | None = None
+    partition_key: str | None = None
 
 
 class RetryRequest(BaseModel):
@@ -105,7 +105,7 @@ def _run_to_response(run: Run) -> RunResponse:
         org_id=run.org_id,
         component_id=run.component_id,
         backfill_id=run.backfill_id,
-        partition_date=run.partition_date,
+        partition_key=run.partition_key,
         status=run.status,
         retry_of=run.retry_of,
         attempt=run.attempt,
@@ -217,7 +217,10 @@ def create_run(
     target = load_authorized(store.get_component, body.component_id, user, store, label="Component", minimum="editor")
     if not KINDS[target.kind].runnable:
         raise HTTPException(status_code=400, detail=f"Components of kind '{target.kind}' cannot be run")
-    run = store.create_run(target.org_id, component_id=body.component_id, partition_date=body.partition_date)
+    try:
+        run = store.create_run(target.org_id, component_id=body.component_id, partition_key=body.partition_key)
+    except ValueError as e:
+        raise HTTPException(status_code=400, detail=str(e))
     return _run_to_response(run)