[Portfolio] enable custom simulated portfolio

Author: Guillaume De Saint Martin

octobot_trading/api/__init__.py

@@ -145,6 +145,7 @@
     get_portfolio_historical_values,
     get_portfolio_currency,
     get_origin_portfolio,
+    set_simulated_portfolio_initial_config,
     refresh_real_trader_portfolio,
     format_portfolio,
     get_draw_down,
@@ -301,6 +302,7 @@
     "get_portfolio_historical_values",
     "get_portfolio_currency",
     "get_origin_portfolio",
+    "set_simulated_portfolio_initial_config",
     "refresh_real_trader_portfolio",
     "get_draw_down",
     "get_coefficient_of_determination",

octobot_trading/api/portfolio.py

@@ -62,6 +62,13 @@ def get_origin_portfolio(exchange_manager, as_decimal=True) -> dict:
     )
 
 
+def set_simulated_portfolio_initial_config(exchange_manager, portfolio_content):
+    if exchange_manager.exchange_personal_data.portfolio_manager:
+        exchange_manager.exchange_personal_data.portfolio_manager.set_simulated_portfolio_initial_config(
+            portfolio_content
+        )
+
+
 def format_portfolio(portfolio, as_decimal) -> dict:
     if as_decimal:
         return portfolio

octobot_trading/personal_data/portfolios/portfolio_manager.pxd

@@ -38,6 +38,7 @@ cdef class PortfolioManager(util.Initializable):
     cdef public portfolio_history.HistoricalPortfolioValueManager historical_portfolio_value_manager
 
     cdef bint _is_initialized_event_set
+    cdef dict _simulated_portfolio_initial_config
 
     cpdef object handle_balance_updated(self)
     cpdef bint handle_balance_update(self, dict balance, bint is_diff_update=*)
@@ -48,6 +49,7 @@ cdef class PortfolioManager(util.Initializable):
     cdef object _apply_locked_assets(self, dict previous_assets)
     cdef void _load_portfolio(self, bint reset_from_config)
     cdef object _load_simulated_portfolio_from_history(self)
+    cdef void set_simulated_portfolio_initial_config(self, dict portfolio_config)
     cdef object _reset_portfolio(self)
-    cdef void _set_starting_simulated_portfolio(self)
+    cdef void _apply_starting_simulated_portfolio(self)
     cdef bint _refresh_simulated_trader_portfolio_from_order(self, object order)

octobot_trading/personal_data/portfolios/portfolio_manager.py

@@ -14,6 +14,7 @@
 #  You should have received a copy of the GNU Lesser General Public
 #  License along with this library.
 import contextlib
+import copy
 
 import octobot_commons.logging as logging
 import octobot_commons.constants as commons_constants
@@ -44,6 +45,7 @@ def __init__(self, config, trader, exchange_manager):
         self.historical_portfolio_value_manager = None
         self.reference_market = None
         self._is_initialized_event_set = False
+        self._simulated_portfolio_initial_config = None
 
     async def initialize_impl(self):
         """
@@ -53,6 +55,9 @@ async def initialize_impl(self):
         if self.exchange_manager.is_storage_enabled() and self.historical_portfolio_value_manager is None:
             self.historical_portfolio_value_manager = personal_data.HistoricalPortfolioValueManager(self)
             await self.historical_portfolio_value_manager.initialize()
+        self.set_simulated_portfolio_initial_config(
+            self.config[commons_constants.CONFIG_SIMULATOR][commons_constants.CONFIG_STARTING_PORTFOLIO]
+        )
         self._reset_portfolio()
 
     def handle_balance_update(self, balance, is_diff_update=False):
@@ -263,7 +268,7 @@ def _load_portfolio(self, reset_from_config):
                 if reset_from_config \
                         or self.historical_portfolio_value_manager is None \
                         or not self.historical_portfolio_value_manager.has_previous_session_portfolio():
-                    self._set_starting_simulated_portfolio()
+                    self._apply_starting_simulated_portfolio()
                 else:
                     self._load_simulated_portfolio_from_history()
             self.logger.info(f"{constants.CURRENT_PORTFOLIO_STRING} {self.portfolio.portfolio}")
@@ -278,13 +283,14 @@ def _load_simulated_portfolio_from_history(self):
         )
         self.handle_balance_update(self.portfolio.get_portfolio_from_amount_dict(portfolio_amount_dict))
 
-    def _set_starting_simulated_portfolio(self):
+    def set_simulated_portfolio_initial_config(self, portfolio_config):
+        self._simulated_portfolio_initial_config = copy.deepcopy(portfolio_config)
+
+    def _apply_starting_simulated_portfolio(self):
         """
         Load new portfolio from config settings
         """
-        portfolio_amount_dict = personal_data.parse_decimal_config_portfolio(
-            self.config[commons_constants.CONFIG_SIMULATOR][commons_constants.CONFIG_STARTING_PORTFOLIO]
-        )
+        portfolio_amount_dict = personal_data.parse_decimal_config_portfolio(self._simulated_portfolio_initial_config)
         self.handle_balance_update(self.portfolio.get_portfolio_from_amount_dict(portfolio_amount_dict))
 
     def _set_initialized_event(self):