investing-algorithm-framework 3.7.0__py3-none-any.whl → 7.19.15__py3-none-any.whl

investing_algorithm_framework/infrastructure/__init__.py

@@ -1,16 +1,22 @@
 from .database import setup_sqlalchemy, Session, \
-    create_all_tables
+    create_all_tables, clear_db
 from .models import SQLPortfolio, SQLOrder, SQLPosition, \
-    SQLPortfolioSnapshot, SQLPositionSnapshot, \
-    CCXTOHLCVBacktestMarketDataSource, CCXTOrderBookMarketDataSource, \
-    CCXTTickerMarketDataSource, CCXTOHLCVMarketDataSource, \
-    CSVOHLCVMarketDataSource, CSVTickerMarketDataSource
+    SQLPortfolioSnapshot, SQLPositionSnapshot, SQLTrade, \
+    SQLTradeTakeProfit, SQLTradeStopLoss
 from .repositories import SQLOrderRepository, SQLPositionRepository, \
-    SQLPortfolioRepository, \
-    SQLPortfolioSnapshotRepository, SQLPositionSnapshotRepository
-from .services import PerformanceService, CCXTMarketService
+    SQLPortfolioRepository, SQLTradeRepository, \
+    SQLPortfolioSnapshotRepository, SQLPositionSnapshotRepository, \
+    SQLTradeTakeProfitRepository, SQLTradeStopLossRepository, \
+    SQLOrderMetadataRepository
+from .services import AzureBlobStorageStateHandler, AWSS3StorageStateHandler
+from .data_providers import CSVOHLCVDataProvider, get_default_data_providers, \
+    get_default_ohlcv_data_providers, CCXTOHLCVDataProvider, \
+    PandasOHLCVDataProvider
+from .order_executors import CCXTOrderExecutor, BacktestOrderExecutor
+from .portfolio_providers import CCXTPortfolioProvider
 
 __all__ = [
+    "clear_db",
     "create_all_tables",
     "SQLPositionRepository",
     "SQLPortfolioRepository",
@@ -20,18 +26,25 @@ __all__ = [
     "setup_sqlalchemy",
     "Session",
     "SQLPortfolio",
+    "SQLTrade",
     "SQLOrder",
     "SQLPosition",
-    "PerformanceService",
     "SQLPortfolioSnapshot",
     "SQLPositionSnapshot",
-    "CCXTOHLCVMarketDataSource",
-    "CCXTOrderBookMarketDataSource",
-    "CCXTTickerMarketDataSource",
-    "CCXTOHLCVMarketDataSource",
-    "CCXTMarketService",
-    "CSVOHLCVMarketDataSource",
-    "CSVTickerMarketDataSource",
-    "CCXTOHLCVBacktestMarketDataSource",
-    "CCXTOrderBookMarketDataSource",
+    "AzureBlobStorageStateHandler",
+    "SQLTradeRepository",
+    "SQLTradeTakeProfit",
+    "SQLTradeStopLoss",
+    "SQLTradeTakeProfitRepository",
+    "SQLTradeStopLossRepository",
+    "SQLOrderMetadataRepository",
+    "CSVOHLCVDataProvider",
+    "CCXTOrderExecutor",
+    "CCXTPortfolioProvider",
+    "get_default_data_providers",
+    "get_default_ohlcv_data_providers",
+    "AWSS3StorageStateHandler",
+    "CCXTOHLCVDataProvider",
+    "BacktestOrderExecutor",
+    "PandasOHLCVDataProvider",
 ]

investing_algorithm_framework/infrastructure/data_providers/__init__.py

@@ -0,0 +1,36 @@
+from .ccxt import CCXTOHLCVDataProvider
+from .csv import CSVOHLCVDataProvider
+from .pandas import PandasOHLCVDataProvider
+
+
+def get_default_data_providers():
+    """
+    Function to get the default data providers.
+
+    Returns:
+        list: List of default data providers.
+    """
+    return [
+        CCXTOHLCVDataProvider(),
+    ]
+
+
+def get_default_ohlcv_data_providers():
+    """
+    Function to get the default OHLCV data providers.
+
+    Returns:
+        list: List of default OHLCV data providers.
+    """
+    return [
+        CCXTOHLCVDataProvider(),
+    ]
+
+
+__all__ = [
+    'CSVOHLCVDataProvider',
+    'CCXTOHLCVDataProvider',
+    'get_default_data_providers',
+    'get_default_ohlcv_data_providers',
+    'PandasOHLCVDataProvider',
+]