diff --git a/server/core/datasets.py b/server/core/datasets.py index 3a62fc9..4690454 100644 --- a/server/core/datasets.py +++ b/server/core/datasets.py @@ -114,7 +114,7 @@ class DatasetManager: self.db.execute_batch(query, values) def set_dataset_status(self, dataset_id: int, status: str, status_message: str | None = None): - if status not in ["processing", "complete", "error"]: + if status not in ["fetching", "processing", "complete", "error"]: raise ValueError("Invalid status") query = """ diff --git a/server/db/schema.sql b/server/db/schema.sql index 051a396..4550633 100644 --- a/server/db/schema.sql +++ b/server/db/schema.sql @@ -23,7 +23,7 @@ CREATE TABLE datasets ( -- Enforce valid states CONSTRAINT datasets_status_check - CHECK (status IN ('processing', 'complete', 'error')) + CHECK (status IN ('fetching', 'processing', 'complete', 'error')) ); CREATE TABLE events (