def init_child(self) -> None:
""""""
self.minute_child = QtWidgets.QTreeWidgetItem()
self.minute_child.setText(0, "分钟线")
self.tree.addTopLevelItem(self.minute_child)
self.hour_child = QtWidgets.QTreeWidgetItem(self.tree)
self.hour_child.setText(0, "小时线")
self.tree.addTopLevelItem(self.hour_child)
self.daily_child = QtWidgets.QTreeWidgetItem(self.tree)
self.daily_child.setText(0, "日线")
self.tree.addTopLevelItem(self.daily_child)
def get_portfolio_item(self, reference: str) -> QtWidgets.QTreeWidgetItem:
""""""
portfolio_item = self.portfolio_items.get(reference, None)
if not portfolio_item:
portfolio_item = QtWidgets.QTreeWidgetItem()
portfolio_item.setText(0, reference)
for i in range(2, self.column_count):
portfolio_item.setTextAlignment(i, QtCore.Qt.AlignCenter)
self.portfolio_items[reference] = portfolio_item
self.tree.addTopLevelItem(portfolio_item)
self.reference_combo.addItem(reference)
return portfolio_item
def get_contract_item(self, reference: str,
vt_symbol: str) -> QtWidgets.QTreeWidgetItem:
""""""
key = (reference, vt_symbol)
contract_item = self.contract_items.get(key, None)
if not contract_item:
contract_item = QtWidgets.QTreeWidgetItem()
contract_item.setText(1, vt_symbol)
for i in range(2, self.column_count):
contract_item.setTextAlignment(i, QtCore.Qt.AlignCenter)
self.contract_items[key] = contract_item
portfolio_item = self.get_portfolio_item(reference)
portfolio_item.addChild(contract_item)
return contract_item
def refresh_tree(self) -> None:
""""""
self.clear_tree()
data = self.engine.get_bar_data_available()
for d in data:
key = (d["symbol"], d["exchange"], d["interval"])
item = self.tree_items.get(key, None)
if not item:
item = QtWidgets.QTreeWidgetItem()
self.tree_items[key] = item
item.setText(1, ".".join([d["symbol"], d["exchange"]]))
item.setText(2, d["symbol"])
item.setText(3, d["exchange"])
if d["interval"] == Interval.MINUTE.value:
self.minute_child.addChild(item)
elif d["interval"] == Interval.HOUR.value:
self.hour_child.addChild(item)
else:
self.daily_child.addChild(item)
output_button = QtWidgets.QPushButton("导出")
output_func = partial(
self.output_data,
d["symbol"],
Exchange(d["exchange"]),
Interval(d["interval"]),
d["start"],
d["end"]
)
output_button.clicked.connect(output_func)
show_button = QtWidgets.QPushButton("查看")
show_func = partial(
self.show_data,
d["symbol"],
Exchange(d["exchange"]),
Interval(d["interval"]),
d["start"],
d["end"]
)
show_button.clicked.connect(show_func)
delete_button = QtWidgets.QPushButton("删除")
delete_func = partial(
self.delete_data,
d["symbol"],
Exchange(d["exchange"]),
Interval(d["interval"]),
)
delete_button.clicked.connect(delete_func)
self.tree.setItemWidget(item, 7, show_button)
self.tree.setItemWidget(item, 8, output_button)
self.tree.setItemWidget(item, 9, delete_button)
item.setText(4, str(d["count"]))
item.setText(5, d["start"].strftime("%Y-%m-%d %H:%M:%S"))
item.setText(6, d["end"].strftime("%Y-%m-%d %H:%M:%S"))
self.minute_child.setExpanded(True)
self.hour_child.setExpanded(True)
self.daily_child.setExpanded(True)
