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,
|
get_table_info,
|
||||||
sync_table,
|
sync_table,
|
||||||
)
|
)
|
||||||
|
from src.data.catalog import (
|
||||||
|
DatabaseCatalog,
|
||||||
|
SQLQueryBuilder,
|
||||||
|
build_context_lazyframe,
|
||||||
|
TableMetadata,
|
||||||
|
FieldMapping,
|
||||||
|
TableFrequency,
|
||||||
|
)
|
||||||
|
|
||||||
__all__ = [
|
__all__ = [
|
||||||
# Configuration
|
# Configuration
|
||||||
@@ -36,4 +44,11 @@ __all__ = [
|
|||||||
"ensure_table",
|
"ensure_table",
|
||||||
"get_table_info",
|
"get_table_info",
|
||||||
"sync_table",
|
"sync_table",
|
||||||
|
# Data catalog
|
||||||
|
"DatabaseCatalog",
|
||||||
|
"SQLQueryBuilder",
|
||||||
|
"build_context_lazyframe",
|
||||||
|
"TableMetadata",
|
||||||
|
"FieldMapping",
|
||||||
|
"TableFrequency",
|
||||||
]
|
]
|
||||||
|
|||||||
@@ -15,7 +15,7 @@ from datetime import datetime
|
|||||||
|
|
||||||
import polars as pl
|
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.engine import FactorEngine
|
||||||
from src.factors.api import close, open, ts_mean, cs_rank
|
from src.factors.api import close, open, ts_mean, cs_rank
|
||||||
|
|
||||||
@@ -215,7 +215,7 @@ def run_factor_integration_test():
|
|||||||
print("=" * 80)
|
print("=" * 80)
|
||||||
|
|
||||||
print("\n[4.1] 重新构建 Context LazyFrame 并打印前 5 行...")
|
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(
|
context_lf = build_context_lazyframe(
|
||||||
required_fields=["close", "open"],
|
required_fields=["close", "open"],
|
||||||
|
|||||||
Reference in New Issue
Block a user