ENH: Yield capital change information

This commit is contained in:
Andrew Liang
2016-08-01 15:19:11 -04:00
parent 18aba63da9
commit 5904ecb40f
3 changed files with 56 additions and 9 deletions
+12 -2
View File
@@ -847,14 +847,16 @@ class TradingAlgorithm(object):
self.perf_tracker.prepare_capital_change(is_interday)
if capital_change['type'] == 'target':
capital_change_amount = capital_change['value'] - \
target = capital_change['value']
capital_change_amount = target - \
self.updated_portfolio().portfolio_value
self.portfolio_needs_update = True
log.info('Processing capital change to target %s at %s. Capital '
'change delta is %s' % (capital_change['value'], dt,
'change delta is %s' % (target, dt,
capital_change_amount))
elif capital_change['type'] == 'delta':
target = None
capital_change_amount = capital_change['value']
log.info('Processing capital change of delta %s at %s'
% (capital_change_amount, dt))
@@ -867,6 +869,14 @@ class TradingAlgorithm(object):
self.perf_tracker.process_capital_change(capital_change_amount,
is_interday)
yield {
'capital_change':
{'date': dt,
'type': 'cash',
'target': target,
'delta': capital_change_amount}
}
@api_method
def get_environment(self, field='platform'):
"""Query the execution environment.