refactor: rename src/data/data_router.py to catalog.py

This commit is contained in:
2026-03-04 22:09:53 +08:00
parent af5c96cd53
commit 620696c842
3 changed files with 17 additions and 2 deletions

View File

@@ -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",
]

View File

@@ -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"],