trading/shared/models/__init__.py

"""Shared SQLAlchemy models — import all models here so Alembic can discover them."""

from shared.models.base import Base, TimestampMixin
from shared.models.trading import (
    Signal,
    SignalDirection,
    Strategy,
    StrategyWeightHistory,
    Trade,
    TradeSide,
    TradeStatus,
    Position,
)
from shared.models.news import Article, ArticleSentiment
from shared.models.learning import LearningAdjustment, TradeOutcome
from shared.models.auth import User, UserCredential
from shared.models.timeseries import MarketData, PortfolioSnapshot, StrategyMetric
from shared.models.fundamentals import Fundamentals

__all__ = [
    "Base",
    "TimestampMixin",
    # Trading
    "Strategy",
    "Signal",
    "SignalDirection",
    "Trade",
    "TradeSide",
    "TradeStatus",
    "Position",
    "StrategyWeightHistory",
    # News
    "Article",
    "ArticleSentiment",
    # Learning
    "TradeOutcome",
    "LearningAdjustment",
    # Auth
    "User",
    "UserCredential",
    # Timeseries
    "MarketData",
    "PortfolioSnapshot",
    "StrategyMetric",
    # Fundamentals
    "Fundamentals",
]