wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-07-10 20:16:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #449 from quantopian/bug_5089

Browse Source 
expand get_environment
This commit is contained in:
fawce
2014-12-12 17:42:58 -05:00
parent 51e43e3d61 34647ec6c4
commit cd976ee2dd
3 changed files with 42 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/algorithm.py
+14 -3
View File
@@ -152,7 +152,7 @@ class TradingAlgorithm(object):
self._recorded_vars = {}
self.namespace = kwargs.get('namespace', {})
self._environment = kwargs.pop('environment', 'zipline')
self._platform = kwargs.pop('platform', 'zipline')
self.logger = None
@@ -549,8 +549,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._platform
}
if field == '*':
return env
else:
return env[field]
def add_event(self, rule=None, callback=None):
"""