From 76ff8e93eb3c259a2273e07849207dd0dadba5ca Mon Sep 17 00:00:00 2001 From: Eddie Hebert Date: Tue, 14 Aug 2012 16:50:24 -0400 Subject: [PATCH] Adds source id when specifying events. So that when we add events for testing, we populate source_ids. --- zipline/gens/tradegens.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/zipline/gens/tradegens.py b/zipline/gens/tradegens.py index e099091d..0842b3bd 100644 --- a/zipline/gens/tradegens.py +++ b/zipline/gens/tradegens.py @@ -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: