Modified get_environment to provide multiple fields.

2026-09-09 11:19:23 +08:00 · 2014-12-07 08:26:13 -05:00
parent eaea8e5317
commit 0411627169
3 changed files with 38 additions and 2 deletions
@@ -545,8 +545,19 @@ class TradingAlgorithm(object):
            self.add_history(bars, freq, 'volume')

    @api_method
-    def get_environment(self):
-        return self._environment
+    def get_environment(self, field='platform'):
+        env = {
+            'arena': self.sim_params.arena,
+            'data_frequency': self.sim_params.data_frequency,
+            'start': self.sim_params.first_open,
+            'end': self.sim_params.last_close,
+            'capital_base': self.sim_params.capital_base,
+            'platform': self._environment
+        }
+        if field == '*':
+            return env
+        else:
+            return env[field]

    def add_event(self, rule=None, callback=None):
        """