senset/vnpy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Add]return engine/gateway object when adding related class to MainEngine

Browse Source
This commit is contained in:
vn.py 2019-04-06 13:32:56 +08:00
parent d4a70b8a9b
commit 36295b96c5
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
vnpy/trader/engine.py
View File

@ -52,6 +52,7 @@ class MainEngine:
""" """
engine = engine_class(self, self.event_engine) engine = engine_class(self, self.event_engine)
self.engines[engine.engine_name] = engine self.engines[engine.engine_name] = engine
return engine
def add_gateway(self, gateway_class: BaseGateway): def add_gateway(self, gateway_class: BaseGateway):
""" """
@ -59,6 +60,7 @@ class MainEngine:
""" """
gateway = gateway_class(self.event_engine) gateway = gateway_class(self.event_engine)
self.gateways[gateway.gateway_name] = gateway self.gateways[gateway.gateway_name] = gateway
return gateway
def add_app(self, app_class: BaseApp): def add_app(self, app_class: BaseApp):
""" """
@ -67,7 +69,8 @@ class MainEngine:
app = app_class() app = app_class()
self.apps[app.app_name] = app self.apps[app.app_name] = app
self.add_engine(app.engine_class) engine = self.add_engine(app.engine_class)
return engine
def init_engines(self): def init_engines(self):
""" """