refactor: rename src/data/data_router.py to catalog.py
This commit is contained in:
@@ -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",
|
||||
]
|
||||
|
||||
@@ -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"],
|
||||
|
||||
Reference in New Issue
Block a user