-
Notifications
You must be signed in to change notification settings - Fork 21
/
Copy pathuiCustomMenu.py
151 lines (128 loc) · 8.01 KB
/
uiCustomMenu.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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
# encoding: UTF-8
from vnpy.trader.uiQt import QtGui, QtWidgets, QtCore, BASIC_FONT
from qtpy.QtCore import Qt,QRect
from functools import partial
from qtpy.QtWidgets import QApplication, QWidget,QPushButton,QMenu
from qtpy.QtGui import QPainter, QPainterPath, QPen, QColor, QPixmap, QIcon, QBrush, QCursor
from vnpy.trader import vtText
from vnpy.event import Event
import qdarkstyle #Qt黑色主题
import sys
class CustomMenu( QtWidgets.QPushButton):
"""合约管理组件"""
signal = QtCore.Signal(type(Event()))
# ----------------------------------------------------------------------
def __init__(self,parent):
"""Constructor"""
super(CustomMenu, self).__init__()
self.parent=parent
# self.initUi()
self.initMenu()
#-----------------------------------------------------------------------
def initMenu(self):
self.setStyleSheet("QMenu{background:purple;}"
"QMenu{border:1px solid lightgray;}"
"QMenu{border-color:green;}"
"QMenu::item{padding:0px 20px 0px 15px;}"
"QMenu::item{height:30px;}"
"QMenu::item{color:blue;}"
"QMenu::item{background:white;}"
"QMenu::item{margin:1px 0px 0px 0px;}"
"QMenu::item:selected:enabled{background:lightgray;}"
"QMenu::item:selected:enabled{color:blue;}"
"QMenu::item:selected:!enabled{background:transparent;}"
"QMenu::separator{height:50px;}"
"QMenu::separator{width:1px;}"
"QMenu::separator{background:white;}"
"QMenu::separator{margin:1px 1px 1px 1px;}"
"QMenu#menu{background:white;}"
"QMenu#menu{border:1px solid lightgray;}"
"QMenu#menu::item{padding:0px 20px 0px 15px;}"
"QMenu#menu::item{height:15px;}"
"QMenu#menu::item:selected:enabled{background:lightgray;}"
"QMenu#menu::item:selected:enabled{color:white;}"
"QMenu#menu::item:selected:!enabled{background:transparent;}"
"QMenu#menu::separator{height:1px;}"
"QMenu#menu::separator{background:lightgray;}"
"QMenu#menu::separator{margin:2px 0px 2px 0px;}"
"QMenu#menu::indicator {padding:5px;}"
)
self.color = QColor(Qt.gray)
self.opacity = 1.0
''''''' 创建右键菜单 '''
# 必须将ContextMenuPolicy设置为Qt.CustomContextMenu
# 否则无法使用customContextMenuRequested信号
self.setContextMenuPolicy(Qt.CustomContextMenu)
self.customContextMenuRequested.connect(self.showContextMenu)
# 创建QMenu
self.contextMenu = QMenu(self)
self.trendMenu=self.contextMenu.addMenu(u"k线形态")
self.swingMenu = self.contextMenu.addMenu(u"技术指标")
self.amountMenu = self.contextMenu.addMenu(u"策略研究")
self.pzxzMenu = self.contextMenu.addMenu(u"合约切换")
# 添加二级菜单
#趋势分析指标
self.actionSAR= self.trendMenu.addAction(u'k线')
self.actionSAR.triggered.connect(lambda: self.parent.initIndicator(u"KLINE"))
self.actionMA = self.trendMenu.addAction(u'信号隐藏')
self.actionMA.triggered.connect(lambda: self.parent.initIndicator(u"信号隐藏"))
self.actionMA = self.trendMenu.addAction(u'信号显示')
self.actionMA.triggered.connect(lambda: self.parent.initIndicator(u"信号显示"))
#摆动分析
self.actionCCI = self.swingMenu.addAction(u'MA SHORT')
self.actionCCI.triggered.connect(lambda: self.parent.initIndicator(u"MA SHORT"))
self.actionROC = self.swingMenu.addAction(u'MA LONG')
self.actionROC.triggered.connect(lambda: self.parent.initIndicator(u"MA LONG"))
self.actionSHORTTERM = self.swingMenu.addAction(u'SHORT TERM(Limit)')
self.actionSHORTTERM.triggered.connect(lambda: self.parent.initIndicator(u"SHORT TERM(Limit)"))
self.actionSHORTTERMF = self.swingMenu.addAction(u'SHORT TERM(First)')
self.actionSHORTTERMF.triggered.connect(lambda: self.parent.initIndicator(u"SHORT TERM(First)"))
self.actionSHORTTERMALL = self.swingMenu.addAction(u'SHORT TERM(All)')
self.actionSHORTTERMALL.triggered.connect(lambda: self.parent.initIndicator(u"SHORT TERM(All)"))
self.actionWAIBAORI = self.swingMenu.addAction(u'外包日')
self.actionWAIBAORI.triggered.connect(lambda: self.parent.initIndicator(u"外包日"))
self.actionGJR_BUY = self.swingMenu.addAction(u'攻击日（买入）')
self.actionGJR_BUY.triggered.connect(lambda: self.parent.initIndicator(u"攻击日（买入）"))
self.actionGJR_SELL = self.swingMenu.addAction(u'攻击日（卖出）')
self.actionGJR_SELL.triggered.connect(lambda: self.parent.initIndicator(u"攻击日（卖出）"))
self.actionRB = self.pzxzMenu.addAction(u'RB9999')
self.actionRB.triggered.connect(lambda: self.parent.initIndicator(u"RB9999"))
self.actionCU = self.pzxzMenu.addAction(u'CU9999')
self.actionCU.triggered.connect(lambda: self.parent.initIndicator(u"CU9999"))
self.actionCF = self.pzxzMenu.addAction(u'CF9999')
self.actionCF.triggered.connect(lambda: self.parent.initIndicator(u"CF9999"))
self.actionC = self.pzxzMenu.addAction(u'C9999')
self.actionC.triggered.connect(lambda: self.parent.initIndicator(u"C9999"))
self.actionJ = self.pzxzMenu.addAction(u'J9999')
self.actionJ.triggered.connect(lambda: self.parent.initIndicator(u"J9999"))
##设为起始日期
self.actionOPI = self.amountMenu.addAction(u'设为起始日期')
self.actionOPI.triggered.connect(lambda: self.parent.initIndicator(u"设为起始日期"))
self.actionOPI1 = self.amountMenu.addAction(u'设为结束日期')
self.actionOPI1.triggered.connect(lambda: self.parent.initIndicator(u"设为结束日期"))
##量仓分析
self.actionOPI2 = self.amountMenu.addAction(u'MA_螺纹空_PLUS')
self.actionOPI2.triggered.connect(lambda: self.parent.initIndicator(u"MA_螺纹空_PLUS"))
##成交量分析
self.actionVOL = self.amountMenu.addAction(u'SHORTTERM_螺纹_多')
self.actionVOL.triggered.connect(lambda: self.parent.initIndicator(u"SHORTTERM_螺纹_多"))
self.action1 = self.amountMenu.addAction(u'SHORTTERM_螺纹_空')
self.action1.triggered.connect(lambda: self.parent.initIndicator(u"SHORTTERM_螺纹_空"))
self.action5 = self.amountMenu.addAction(u'VOLATILITY_螺纹_多')
self.action5.triggered.connect(lambda: self.parent.initIndicator(u"VOLATILITY_螺纹_多"))
self.action6 = self.amountMenu.addAction(u'VOLATILITY_螺纹_空')
self.action6.triggered.connect(lambda: self.parent.initIndicator(u"VOLATILITY_螺纹_空"))
self.action9 = self.amountMenu.addAction(u'VOLATILITY_螺纹_V1')
self.action9.triggered.connect(lambda: self.parent.initIndicator(u"VOLATILITY_螺纹_V1"))
self.action7 = self.amountMenu.addAction(u'外包日_螺纹_多')
self.action7.triggered.connect(lambda: self.parent.initIndicator(u"外包日_螺纹_多"))
#self.contextMenu.exec_(QCursor.pos()) # 在鼠标位置显示
#添加二级菜单
def showContextMenu(self, pos):
'''''
右键点击时调用的函数
'''
# 菜单显示前，将它移动到鼠标点击的位置
# self.contextMenu.move(self.pos() + pos)
self.contextMenu.show()
self.contextMenu.exec_(QCursor.pos())