Implements custom seed function to CustomSecurityInitializerAlgorithm

This commit is contained in:
AlexCatarino
2018-09-17 17:19:15 +01:00
parent de3798fa35
commit 720e0400fc
2 changed files with 35 additions and 2 deletions
@@ -40,7 +40,7 @@ class CustomSecurityInitializerAlgorithm(QCAlgorithm):
# set our initializer to our custom type
self.SetBrokerageModel(BrokerageName.InteractiveBrokersBrokerage)
func_security_seeder = FuncSecuritySeeder(Func[Security, BaseData](self.GetLastKnownPrice))
func_security_seeder = FuncSecuritySeeder(Func[Security, BaseData](self.custom_seed_function))
self.SetSecurityInitializer(CustomSecurityInitializer(self.BrokerageModel, func_security_seeder, DataNormalizationMode.Raw))
self.SetStartDate(2013,10,1)
@@ -52,6 +52,23 @@ class CustomSecurityInitializerAlgorithm(QCAlgorithm):
if not self.Portfolio.Invested:
self.SetHoldings("SPY", 1)
def custom_seed_function(self, security):
resolution = Resolution.Hour
df = self.History(security.Symbol, 1, resolution)
if df.empty:
return None
last_bar = df.unstack(level=0).iloc[-1]
date_time = last_bar.name.to_pydatetime()
open = last_bar.open.values[0]
high = last_bar.high.values[0]
low = last_bar.low.values[0]
close = last_bar.close.values[0]
volume = last_bar.volume.values[0]
return TradeBar(date_time, security.Symbol, open, high, low, close, volume, Extensions.ToTimeSpan(resolution))
class CustomSecurityInitializer(BrokerageModelSecurityInitializer):
'''Our custom initializer that will set the data normalization mode.