wassname/catalyst
1
0
Fork 0
mirror of https://github.com/wassname/catalyst.git synced 2026-06-30 17:56:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adds source id when specifying events.

Browse Source 
So that when we add events for testing, we populate source_ids.
This commit is contained in:
Eddie Hebert
2012-08-14 16:50:24 -04:00
parent 473dafcfb6
commit 76ff8e93eb
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
zipline/gens/tradegens.py
+7 -1
View File
@@ -101,10 +101,16 @@ class SpecificEquityTrades(object):
def get_hash(self):
return self.__class__.__name__ + "-" + self.arg_string
def update_source_id(self, gen):
for event in gen:
event.source_id = self.get_hash()
yield event
def create_fresh_generator(self):
if self.event_list:
unfiltered = (event for event in self.event_list)
event_gen = (event for event in self.event_list)
unfiltered = self.update_source_id(event_gen)
# Set up iterators for each expected field.
else: