From e75cf4c545025bc52f06e0ca502e236f1d36a436 Mon Sep 17 00:00:00 2001 From: alex Date: Thu, 25 Dec 2025 19:02:00 +0100 Subject: [PATCH] fix: Support ELABDATADISP in fetch_rows_ordered_for_consolidation() MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit The method was restricted to only RAWDATACOR, but the consolidation logic works for both tables. Updated the check to allow both: - RAWDATACOR - ELABDATADISP The keyset pagination (id-based WHERE clause) works identically for both tables, and consolidation happens in Python for both. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Haiku 4.5 --- src/connectors/mysql_connector.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/connectors/mysql_connector.py b/src/connectors/mysql_connector.py index 2c54be9..e29c368 100644 --- a/src/connectors/mysql_connector.py +++ b/src/connectors/mysql_connector.py @@ -231,15 +231,15 @@ class MySQLConnector: Consolidation happens within each batch in Python. Args: - table: Table name (must be 'RAWDATACOR') + table: Table name ('RAWDATACOR' or 'ELABDATADISP') start_id: Resume from this ID (fetch id > start_id). If None, starts from beginning batch_size: Number of rows per batch (uses config default if None) Yields: Batches of row dictionaries """ - if table != "RAWDATACOR": - raise ValueError("Consolidation ordering only supported for RAWDATACOR") + if table not in ("RAWDATACOR", "ELABDATADISP"): + raise ValueError(f"Consolidation ordering only supported for RAWDATACOR and ELABDATADISP, got {table}") if batch_size is None: batch_size = self.settings.migration.batch_size