forked from 51bitquant/howtrader
-
Notifications
You must be signed in to change notification settings - Fork 0
/
main_window.py
82 lines (63 loc) · 2.46 KB
/
main_window.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
from howtrader.event import EventEngine, Event
from howtrader.trader.event import EVENT_TV_SIGNAL
from howtrader.trader.engine import MainEngine
from howtrader.trader.ui import MainWindow, create_qapp
from howtrader.trader.setting import SETTINGS
from howtrader.gateway.binance import BinanceUsdtGateway, BinanceSpotGateway, BinanceInverseGateway
from howtrader.gateway.okx import OkxGateway
from howtrader.app.cta_strategy import CtaStrategyApp
# from howtrader.app.data_manager import DataManagerApp
# from howtrader.app.data_recorder import DataRecorderApp
# from howtrader.app.algo_trading import AlgoTradingApp
# from howtrader.app.risk_manager import RiskManagerApp
# from howtrader.app.spread_trading import SpreadTradingApp
from howtrader.app.tradingview import TradingViewApp
from threading import Thread
import json
from flask import Flask, request
# create global event_engine
event_engine: EventEngine = EventEngine()
passphrase = SETTINGS.get("passphrase", "")
port = SETTINGS.get("port", 9999)
app = Flask(__name__)
@app.route('/', methods=['GET'])
def welcome():
return "Hi, this is tv server!"
@app.route('/webhook', methods=['POST'])
def webhook():
try:
data = json.loads(request.data)
if data.get('passphrase', None) != passphrase:
return {"status": "failure", "msg": "passphrase is incorrect"}
del data['passphrase'] # del it for safety.
event: Event = Event(type=EVENT_TV_SIGNAL, data=data)
event_engine.put(event)
return {"status": "success", "msg": ""}
except Exception as error:
return {"status": "error", "msg": str(error)}
def start_tv_server():
app.run(host="127.0.0.1", port=port)
def main():
""""""
qapp = create_qapp()
main_engine = MainEngine(event_engine)
main_engine.add_gateway(BinanceSpotGateway)
main_engine.add_gateway(BinanceUsdtGateway)
main_engine.add_gateway(BinanceInverseGateway)
main_engine.add_gateway(OkxGateway)
main_engine.add_app(CtaStrategyApp)
main_engine.add_app(TradingViewApp)
# if you don't use
# main_engine.add_app(DataManagerApp)
# main_engine.add_app(DataRecorderApp)
# main_engine.add_app(AlgoTradingApp)
# main_engine.add_app(RiskManagerApp)
# main_engine.add_app(SpreadTradingApp)
main_window = MainWindow(main_engine, event_engine)
main_window.showMaximized()
t1 = Thread(target=start_tv_server)
t1.daemon = True
t1.start()
qapp.exec()
if __name__ == "__main__":
main()