diff --git a/src/data/__init__.py b/src/data/__init__.py index ab47151..d094261 100644 --- a/src/data/__init__.py +++ b/src/data/__init__.py @@ -15,6 +15,14 @@ from src.data.db_manager import ( get_table_info, sync_table, ) +from src.data.catalog import ( + DatabaseCatalog, + SQLQueryBuilder, + build_context_lazyframe, + TableMetadata, + FieldMapping, + TableFrequency, +) __all__ = [ # Configuration @@ -36,4 +44,11 @@ __all__ = [ "ensure_table", "get_table_info", "sync_table", + # Data catalog + "DatabaseCatalog", + "SQLQueryBuilder", + "build_context_lazyframe", + "TableMetadata", + "FieldMapping", + "TableFrequency", ] diff --git a/src/data/data_router.py b/src/data/catalog.py similarity index 100% rename from src/data/data_router.py rename to src/data/catalog.py diff --git a/tests/test_factor_integration.py b/tests/test_factor_integration.py index 29ee34b..eb1dcb0 100644 --- a/tests/test_factor_integration.py +++ b/tests/test_factor_integration.py @@ -15,7 +15,7 @@ from datetime import datetime import polars as pl -from src.data.data_router import DatabaseCatalog +from src.data.catalog import DatabaseCatalog from src.factors.engine import FactorEngine from src.factors.api import close, open, ts_mean, cs_rank @@ -215,7 +215,7 @@ def run_factor_integration_test(): print("=" * 80) print("\n[4.1] 重新构建 Context LazyFrame 并打印前 5 行...") - from src.data.data_router import build_context_lazyframe + from src.data.catalog import build_context_lazyframe context_lf = build_context_lazyframe( required_fields=["close", "open"],