[Add] Add trader ui widget
This commit is contained in:
vn.py
parent
ca14973b9a
commit
e25787536a
21
vnpy/trader/ui/__init__.py
Normal file
21
vnpy/trader/ui/__init__.py
Normal file
@ -0,0 +1,21 @@
|
||||
import platform
|
||||
import ctypes
|
||||
|
||||
import qdarkstyle
|
||||
from PyQt5 import QtWidgets
|
||||
|
||||
from .mainwindow import MainWindow
|
||||
|
||||
|
||||
def create_qapp():
|
||||
"""
|
||||
Create Qt Application.
|
||||
"""
|
||||
qapp = QtWidgets.QApplication([])
|
||||
qapp.setStyleSheet(qdarkstyle.load_stylesheet_pyqt5())
|
||||
|
||||
if 'Windows' in platform.uname():
|
||||
ctypes.windll.shell32.SetCurrentProcessExplicitAppUserModelID(
|
||||
'VN Trader')
|
||||
|
||||
return qapp
|
||||
585
vnpy/trader/ui/widget.py
Normal file
585
vnpy/trader/ui/widget.py
Normal file
@ -0,0 +1,585 @@
|
||||
"""
|
||||
Basic widgets for VN Trader.
|
||||
"""
|
||||
|
||||
import csv
|
||||
from typing import Any
|
||||
|
||||
from PyQt5 import QtWidgets, QtGui, QtCore
|
||||
|
||||
from vnpy.event import EventEngine, Event
|
||||
from ..constant import DIRECTION_LONG, DIRECTION_SHORT, DIRECTION_NET
|
||||
from ..engine import MainEngine
|
||||
from ..event import (EVENT_TICK, EVENT_ORDER, EVENT_TRADE, EVENT_ACCOUNT,
|
||||
EVENT_POSITION, EVENT_CONTRACT, EVENT_LOG)
|
||||
|
||||
COLOR_LONG = QtGui.QColor("red")
|
||||
COLOR_SHORT = QtGui.QColor("green")
|
||||
COLOR_BID = QtGui.QColor(255, 174, 201)
|
||||
COLOR_ASK = QtGui.QColor(160, 255, 160)
|
||||
COLOR_BLACK = QtGui.QColor("black")
|
||||
|
||||
|
||||
class BaseCell(QtWidgets.QTableWidgetItem):
|
||||
"""
|
||||
General cell used in tablewidgets.
|
||||
"""
|
||||
|
||||
def __init__(self, content: Any, data: Any):
|
||||
""""""
|
||||
super(BaseCel, self).__init__()
|
||||
self.setTextAlignment(QtCore.Qt.AlignCenter)
|
||||
self.set_content(content, data)
|
||||
|
||||
def set_content(self, content: Any, data: Any):
|
||||
"""
|
||||
Set text content.
|
||||
"""
|
||||
self.setText(str(content))
|
||||
self._data = data
|
||||
|
||||
def get_data(self):
|
||||
"""
|
||||
Get data object.
|
||||
"""
|
||||
return self._data
|
||||
|
||||
|
||||
class DirectionCell(BaseCell):
|
||||
"""
|
||||
Cell used for showing direction data.
|
||||
"""
|
||||
|
||||
def __init__(self, content: str, data: Any):
|
||||
""""""
|
||||
super(DirectionCell, self).__init__(content, data)
|
||||
|
||||
def set_content(self, content: Any, data: Any):
|
||||
"""
|
||||
Cell color is set according to direction.
|
||||
"""
|
||||
super(DirectionCell, self).set_content(content, data)
|
||||
|
||||
if content == DIRECTION_SHORT:
|
||||
self.setForeground(COLOR_SHORT)
|
||||
else:
|
||||
self.setForeground(COLOR_LONG)
|
||||
|
||||
|
||||
class BidCell(BaseCell):
|
||||
"""
|
||||
Cell used for showing bid price and volume.
|
||||
"""
|
||||
|
||||
def __init__(self, content: Any, data: Any):
|
||||
""""""
|
||||
super(BidCell, self).__init__(content, data)
|
||||
|
||||
self.setForeground(COLOR_BLACK)
|
||||
self.setForeground(COLOR_BID)
|
||||
|
||||
|
||||
class AskCell(BaseCell):
|
||||
"""
|
||||
Cell used for showing ask price and volume.
|
||||
"""
|
||||
|
||||
def __init__(self, content: Any, data: Any):
|
||||
""""""
|
||||
super(AskCell, self).__init__(content, data)
|
||||
|
||||
self.setForeground(COLOR_BLACK)
|
||||
self.setForeground(COLOR_ASK)
|
||||
|
||||
|
||||
class PnlCell(BaseCell):
|
||||
"""
|
||||
Cell used for showing pnl data.
|
||||
"""
|
||||
|
||||
def __init__(self, content: Any, data: Any):
|
||||
""""""
|
||||
super(PnlCell, self).__init__(content, data)
|
||||
|
||||
def set_content(self, content: Any, data: Any):
|
||||
"""
|
||||
Cell color is set based on whether pnl is
|
||||
positive or negative.
|
||||
"""
|
||||
super(PnlCell, self).set_content(content, data)
|
||||
|
||||
if content.startswith("-"):
|
||||
self.setForeground(COLOR_SHORT)
|
||||
else:
|
||||
self.setForeground(COLOR_LONG)
|
||||
|
||||
|
||||
class TimeCell(BaseCell):
|
||||
"""
|
||||
Cell used for showing time string from datetime object.
|
||||
"""
|
||||
|
||||
def __init__(self, content: Any, data: Any):
|
||||
""""""
|
||||
super(TimeCell, self).__init__(content, data)
|
||||
|
||||
def set_content(self, content: Any, data: Any):
|
||||
"""
|
||||
Time format is 12:12:12.5
|
||||
"""
|
||||
timestamp = content.strftime("%H:%M:%S")
|
||||
|
||||
millisecond = int(content.microsecond / 1000)
|
||||
if millisecond:
|
||||
timestamp = f"{timestamp}.{millisecond}"
|
||||
|
||||
self.setText(timestamp)
|
||||
self._data = data
|
||||
|
||||
|
||||
class BaseMonitor(QtWidgets.QTableWidget):
|
||||
"""
|
||||
Monitor data update in VN Trader.
|
||||
"""
|
||||
|
||||
event_type = ""
|
||||
data_key = ""
|
||||
sorting = False
|
||||
headers = {}
|
||||
|
||||
signal = QtCore.pyqtSignal(Event)
|
||||
|
||||
def __init__(self, main_engine: MainEngine, event_engine: EventEngine):
|
||||
""""""
|
||||
super(BaseMonitor, self).__init__()
|
||||
|
||||
self.main_engine = main_engine
|
||||
self.event_engine = event_engine
|
||||
self.cells = {}
|
||||
|
||||
self.init_ui()
|
||||
self.register_event()
|
||||
|
||||
def init_ui(self):
|
||||
""""""
|
||||
self.init_table()
|
||||
self.init_menu()
|
||||
|
||||
def init_table(self):
|
||||
"""
|
||||
Initialize table.
|
||||
"""
|
||||
self.setColumnCount(len(self.headers))
|
||||
|
||||
labels = [d['display'] for d in self.headers.values()]
|
||||
self.setHorizontalHeaderLabels(labels)
|
||||
|
||||
self.verticalHeader().setVisible(False)
|
||||
self.setEditTriggers(self.NoEditTriggers)
|
||||
self.setAlternatingRowColors(True)
|
||||
self.setSortingEnabled(self.sorting)
|
||||
|
||||
def init_menu(self):
|
||||
"""
|
||||
Create right click menu.
|
||||
"""
|
||||
self.menu = QtWidgets.QMenu(self)
|
||||
|
||||
resize_action = QtWidgets.QAction("调整列宽", self)
|
||||
resize_action.triggered.connect(self.resize_columns)
|
||||
self.menu.addAction(resize_action)
|
||||
|
||||
save_action = QtWidgets.QAction("保存数据", self)
|
||||
save_action.triggered.connect(self.save_csv)
|
||||
self.menu.addAction(save_action)
|
||||
|
||||
def register_event(self):
|
||||
"""
|
||||
Register event handler into event engine.
|
||||
"""
|
||||
self.signal.connect(self.process_event)
|
||||
self.event_engine.register(self.event_type, self.process_event)
|
||||
|
||||
def process_event(self, event):
|
||||
"""
|
||||
Process new data from event and update into table.
|
||||
"""
|
||||
# Disable sorting to prevent unwanted error.
|
||||
if self.sorting:
|
||||
self.setSortingEnabled(False)
|
||||
|
||||
# Update data into table.
|
||||
data = event.data
|
||||
|
||||
if not self.data_key:
|
||||
self.insert_new_row(data)
|
||||
else:
|
||||
key = data.__getattribute__(self.data_key)
|
||||
|
||||
if key in self.cells:
|
||||
self.update_old_row(data)
|
||||
else:
|
||||
self.insert_new_row(data)
|
||||
|
||||
# Enable sorting
|
||||
if self.sorting:
|
||||
self.setSortingEnabled(True)
|
||||
|
||||
def insert_new_row(self, data):
|
||||
"""
|
||||
Insert a new row at the top of table.
|
||||
"""
|
||||
self.insertRow(0)
|
||||
|
||||
row_cells = {}
|
||||
for column, header in enumerate(self.headers.keys()):
|
||||
setting = self.headers[header]
|
||||
|
||||
content = data.__getattribute__(header)
|
||||
cell = setting['cell'](content, data)
|
||||
self.setItem(0, column, cell)
|
||||
|
||||
if setting['update']:
|
||||
row_cells[header] = cell
|
||||
|
||||
if self.data_key:
|
||||
key = data.__getattribute__(self.data_key)
|
||||
self.cells[key] = row_cells
|
||||
|
||||
def update_old_row(self, data):
|
||||
"""
|
||||
Update an old row in table.
|
||||
"""
|
||||
key = data.__getattribute__(self.data_key)
|
||||
row_cells = self.cells[key]
|
||||
|
||||
for header, cell in row_cells:
|
||||
content = data.__getattribute__(header)
|
||||
cell.set_content(content, data)
|
||||
|
||||
def resize_columns(self):
|
||||
"""
|
||||
Resize all columns according to contents.
|
||||
"""
|
||||
self.horizontalHeader().resizeSections(
|
||||
QtWidgets.QHeaderView.ResizeToContents)
|
||||
|
||||
def save_csv(self):
|
||||
"""
|
||||
Save table data into a csv file
|
||||
"""
|
||||
path, _ = QtWidgets.QFileDialog.getSaveFileName(self, "保存数据", "",
|
||||
"CSV(*.csv)")
|
||||
|
||||
if not path:
|
||||
return
|
||||
|
||||
with open(path, "wb") as f:
|
||||
writer = csv.writer(f)
|
||||
writer.writerow(self.headers.keys())
|
||||
|
||||
for row in range(self.rowCount()):
|
||||
row_data = []
|
||||
for column in range(self.columnCount()):
|
||||
item = self.item(row, column)
|
||||
if item:
|
||||
row_data.append(text_type(item.text()))
|
||||
else:
|
||||
row_data.append("")
|
||||
writer.writerow(row_data)
|
||||
|
||||
|
||||
class TickMonitor(BaseMonitor):
|
||||
"""
|
||||
Monitor for tick data.
|
||||
"""
|
||||
event_type = EVENT_TICK
|
||||
data_key = "vt_symbol"
|
||||
sorting = True
|
||||
|
||||
headers = {
|
||||
'symbol': {
|
||||
'display': '代码',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'last_price': {
|
||||
'display': '最新价',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'volume': {
|
||||
'display': '成交量',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'open_price': {
|
||||
'display': '开盘价',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'high_price': {
|
||||
'display': '最高价',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'low_price': {
|
||||
'display': '最低价',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'bid_price_1': {
|
||||
'display': '买1价',
|
||||
'cell': BidCell,
|
||||
'update': True
|
||||
},
|
||||
'bid_volume_1': {
|
||||
'display': '买1量',
|
||||
'cell': BidCell,
|
||||
'update': True
|
||||
},
|
||||
'ask_price_1': {
|
||||
'display': '卖1价',
|
||||
'cell': AskCell,
|
||||
'update': True
|
||||
},
|
||||
'ask_volume_1': {
|
||||
'display': '卖1量',
|
||||
'cell': AskCell,
|
||||
'update': True
|
||||
},
|
||||
'time': {
|
||||
'display': '时间',
|
||||
'cell': TimeCell,
|
||||
'update': True
|
||||
},
|
||||
'gateway_name': {
|
||||
'display': '接口',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
class LogMonitor(BaseMonitor):
|
||||
"""
|
||||
Monitor for log data.
|
||||
"""
|
||||
event_type = EVENT_LOG
|
||||
data_key = ""
|
||||
sorting = False
|
||||
|
||||
headers = {
|
||||
'time': {
|
||||
'display': '时间',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'msg': {
|
||||
'display': '信息',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'gateway_name': {
|
||||
'display': '接口',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
}
|
||||
}
|
||||
|
||||
def process_event(self, event):
|
||||
"""Resize row heights for diplaying long log message."""
|
||||
super(LogMonitor, self).process_event(event)
|
||||
self.resizeRowToContents(0)
|
||||
|
||||
|
||||
class TradeMonitor(BaseMonitor):
|
||||
"""
|
||||
Monitor for trade data.
|
||||
"""
|
||||
event_type = EVENT_TRADE
|
||||
data_key = ""
|
||||
sorting = True
|
||||
|
||||
headers = {
|
||||
'tradeid': {
|
||||
'display': "成交号 ",
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'orderid': {
|
||||
'display': '委托号',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'symbol': {
|
||||
'display': '代码',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'direction': {
|
||||
'display': '方向',
|
||||
'cell': DirectionCell,
|
||||
'update': False
|
||||
},
|
||||
'offset': {
|
||||
'display': '开平',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'price': {
|
||||
'display': '价格',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'volume': {
|
||||
'display': '数量',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'gateway_name': {
|
||||
'display': '接口',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
class OrderMonitor(BaseMonitor):
|
||||
"""
|
||||
Monitor for order data.
|
||||
"""
|
||||
event_type = EVENT_ORDER
|
||||
data_key = "vt_orderid"
|
||||
sorting = True
|
||||
|
||||
headers = {
|
||||
'orderid': {
|
||||
'display': '委托号',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'symbol': {
|
||||
'display': '代码',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'direction': {
|
||||
'display': '方向',
|
||||
'cell': DirectionCell,
|
||||
'update': False
|
||||
},
|
||||
'offset': {
|
||||
'display': '开平',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'price': {
|
||||
'display': '价格',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'volume': {
|
||||
'display': '总数量',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'traded': {
|
||||
'display': '已成交',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'status': {
|
||||
'display': '状态',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'time': {
|
||||
'display': '时间',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'gateway_name': {
|
||||
'display': '接口',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
}
|
||||
}
|
||||
|
||||
class PositionMonitor(BaseMonitor):
|
||||
"""
|
||||
Monitor for position data.
|
||||
"""
|
||||
event_type = EVENT_POSITION
|
||||
data_key = "vt_positionid"
|
||||
sorting = True
|
||||
|
||||
headers = {
|
||||
'symbol': {
|
||||
'display': '代码',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'direction': {
|
||||
'display': '方向',
|
||||
'cell': DirectionCell,
|
||||
'update': False
|
||||
},
|
||||
'volume': {
|
||||
'display': '数量',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'frozen': {
|
||||
'display': '冻结',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'price': {
|
||||
'display': '均价',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'pnl': {
|
||||
'display': '盈亏',
|
||||
'cell': PnlCell,
|
||||
'update': True
|
||||
},
|
||||
'gateway_name': {
|
||||
'display': '接口',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
class AccountMonitor(BaseMonitor):
|
||||
"""
|
||||
Monitor for account data.
|
||||
"""
|
||||
event_type = EVENT_ACCOUNT
|
||||
data_key = "vt_accountid"
|
||||
sorting = True
|
||||
|
||||
headers = {
|
||||
'accountid': {
|
||||
'display': '账号',
|
||||
'cell': BaseCell,
|
||||
'update': False
|
||||
},
|
||||
'balance': {
|
||||
'display': '余额',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'frozen': {
|
||||
'display': '冻结',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
},
|
||||
'available': {
|
||||
'display': '可用',
|
||||
'cell': BaseCell,
|
||||
'update': True
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user