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