from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar
from matplotlib.figure import Figure
import matplotlib.animation as animation
import numpy as np
# 动作
class MyQAction(QAction):
def __init__(self, parent, text='', triggered=None, shortcut=None, icon=None, tip=None, checkable=False):
super().__init__(parent)
self.setText(text)
if icon is not None:
self.setIcon(QIcon("ico/%s.ico" % icon))
if shortcut is not None:
self.setShortcut(shortcut)
if tip is not None:
#self.setToolTip(tip)
self.setStatusTip(tip)
if triggered is not None:
self.triggered.connect(triggered)
if checkable:
self.setCheckable(True)
# Canvas
class mainGraphCanvas(FigureCanvas):
def __init__(self, parent=None, width=5, height=4, dpi=100):
self.fig = Figure((width, height), dpi=dpi)
FigureCanvas.__init__(self, self.fig)
self.setParent(parent)
FigureCanvas.setSizePolicy(self, QSizePolicy.Expanding, QSizePolicy.Expanding)
FigureCanvas.updateGeometry(self)
self.plot(self.fig)
def plot(self, fig):
pass
# 界面
class Ui_Form(object):
def setupUi(self, Form):
Form.resize(1234, 669)
Form.setWindowTitle("彩票助手")
frame = QFrame()
#frame.setPalette(QPalette(Qt.blue)) ################################ 蓝色
layout = QHBoxLayout(frame)
#layout.setContentsMargins(2,2,2,2)
#=========================================================
# 左侧TabWidget
#=========================================================
toolbox = QToolBox()
toolbox.addItem(QFrame(), "图")
toolbox.addItem(QFrame(), "表")
toolbox.addItem(QFrame(), "数")
#toolbox.setFixedWidth(150)
groupbox = QGroupBox()
groupbox.setFixedWidth(150)
vlayout = QVBoxLayout(groupbox)
vlayout.setContentsMargins(2,2,2,2)
#vlayout.setAlignment(Qt.AlignCenter)
vlayout.addWidget(toolbox)
tab = QTabWidget()
tab.addTab(groupbox, '正常')
tab.addTab(QFrame(), '其他')
#=========================================================
# 右侧TabWidget
#=========================================================
stack = QStackedWidget()
# 页面1
page1 = QTableWidget(6,4)
page1.setHorizontalHeaderLabels(['This','is','a','TableWidget!'])
page1.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)# 表宽度自适应
#self.page1.resizeColumnsToContents()# 列宽度自适应
for i in range(page1.rowCount()):
for j in range(page1.columnCount()):
page1.setItem(i, j, QTableWidgetItem('(%d,%d)'% (i,j)))
stack.addWidget(page1)
# 页面2
page2 = mainGraphCanvas(stack)
stack.addWidget(page2)
# 页面3
page3 = QTextEdit()
stack.addWidget(page3)
layout.addWidget(tab)
layout.addWidget(stack)
toolbox.currentChanged.connect(stack.setCurrentIndex)
Form.setCentralWidget(frame) # 设置居中
# =================================================
# # 动作
# =================================================
# 数据
self.update_action = MyQAction(Form, "更新", triggered=qApp.quit, shortcut="Ctrl+U", icon="update", tip='??')
self.view_action = MyQAction(Form, "查看", triggered=qApp.quit, shortcut="Ctrl+V", icon="view", tip='??')
self.exit_action = MyQAction(Form, "退出", triggered=qApp.quit, shortcut="Ctrl+Q",tip='??')
# 指标
self.tema_action = MyQAction(Form, "特码", triggered=qApp.quit, icon="tema", tip='??')
self.pingte_action = MyQAction(Form, "平特", triggered=qApp.quit, icon="pingte", tip='??')
self.formula_action = MyQAction(Form, "公式", triggered=qApp.quit, shortcut="Ctrl+F", icon="formula", tip='??')
self.setup_action = MyQAction(Form, "设置", triggered=qApp.quit, shortcut="Ctrl+S", icon="setup", tip='??')
# 图表
self.kline_action = MyQAction(Form, "K 线", triggered=qApp.quit, shortcut="Ctrl+K", icon="kline", tip='??')
self.tline_action = MyQAction(Form, "T 线", triggered=qApp.quit, shortcut="Ctrl+T", icon="tline", tip='??')
self.pie_action = MyQAction(Form, "饼图", triggered=qApp.quit, shortcut="Ctrl+P", icon="pie", tip='??')
# 预测
self.bpnn_action = MyQAction(Form, "神经网络", triggered=qApp.quit, shortcut="Ctrl+N", icon="bpnn", tip='??')
self.svm_action = MyQAction(Form, "支持向量机", triggered=qApp.quit, icon="svm", tip='??')
self.randomforest_action = MyQAction(Form, "随机森林", triggered=qApp.quit, icon="randomforest", tip='??')
self.embled_action = MyQAction(Form, "集成", triggered=qApp.quit, shortcut="Ctrl+E", icon="embled", tip='??')
# 回测
self.policy_action = MyQAction(Form, "策略", triggered=qApp.quit, icon="policy", tip='??')
self.skill_action = MyQAction(Form, "技术", triggered=qApp.quit, icon="skill", tip='??')
self.scheme_action = MyQAction(Form, "方案", triggered=qApp.quit, icon="scheme", tip='??')
self.fund_action = MyQAction(Form, "资金", triggered=qApp.quit, icon="fund", tip='??')
# 选号
self.indexpool_action = MyQAction(Form, "根据指标池", triggered=qApp.quit, shortcut="Ctrl+O", icon="indexpool", tip='??')
self.indexcustom_action = MyQAction(Form, "根据自定义指标", triggered=qApp.quit, icon="indexcustom", tip='??')
self.interunion_action = MyQAction(Form, "交集并集", triggered=qApp.quit, shortcut="Ctrl+I", icon="interunion", tip='??')
self.other_action = MyQAction(Form, "其他", triggered=qApp.quit, icon="other", tip='??')
# 统计
self.limit_action = MyQAction(Form, "极限", triggered=qApp.quit, shortcut="Ctrl+L", icon="limit", tip='??')
self.custom_action = MyQAction(Form, "自定义", triggered=qApp.quit, icon="custom", tip='??')
# 帮助
self.manual_action = MyQAction(Form, "手册", triggered=qApp.quit, icon="manual", tip='??')
self.author_action = MyQAction(Form, "作者", triggered=qApp.quit, icon="author", tip='??')
self.about_action = MyQAction(Form, "关于", triggered=qApp.quit, icon="about", tip='??')
# =================================================
# 状态栏
# =================================================
#Form.statusBar()
# =================================================
# 菜单栏
# =================================================
#menubar = QMenuBar(self)
menubar = Form.menuBar()
menu = menubar.addMenu("数据(&D)")
menu.addAction(self.update_action)
menu.addAction(self.view_action)
menu.addSeparator()
menu.addAction(self.exit_action)
menu = menubar.addMenu("指标(&I)")
menu.addAction(self.tema_action)
menu.addAction(self.pingte_action)
menu.addAction(self.formula_action)
menu.addAction(self.setup_action)
menu = menubar.addMenu("图表(&C)")
menu.addAction(self.kline_action)
menu.addAction(self.tline_action)
menu.addAction(self.pie_action)
menu = menubar.addMenu("预测(&P)")
menu.addAction(self.bpnn_action)
menu.addAction(self.svm_action)
menu.addAction(self.randomforest_action)
menu.addAction(self.embled_action)
menu = menubar.addMenu("回测(&B)")
menu.addAction(self.policy_action)
menu.addAction(self.skill_action)
menu.addAction(self.scheme_action)
menu.addAction(self.fund_action)
menu = menubar.addMenu("统计(&S)")
menu.addAction(self.limit_action)
menu.addAction(self.custom_action)
menu = menubar.addMenu("选号(&X)")
menu.addAction(self.indexpool_action)
menu.addAction(self.indexcustom_action)
menu.addAction(self.interunion_action)
menu.addAction(self.other_action)
menu = menubar.addMenu("帮助(&H)")
menu.addAction(self.manual_action)
menu.addAction(self.author_action)
menu.addAction(self.about_action)
# =================================================
# 工具栏
# =================================================
toolbar = Form.addToolBar('文件')
toolbar.addAction(self.update_action)
toolbar.addAction(self.view_action)
toolbar.addSeparator()
toolbar.addAction(self.tema_action)
toolbar = Form.addToolBar("编辑")
toolbar.addAction(self.pingte_action)
toolbar.addAction(self.formula_action)
toolbar.addAction(self.kline_action)
toolbar.addAction(self.tline_action)
toolbar.addAction(self.pie_action)
toolbar.addAction(self.bpnn_action)
toolbar.addAction(self.svm_action)
toolbar.addAction(self.randomforest_action)
toolbar.addAction(self.embled_action)
toolbar.addAction(self.policy_action)
toolbar.addAction(self.skill_action)
toolbar.addAction(self.scheme_action)
toolbar.addAction(self.fund_action)
toolbar.addAction(self.indexpool_action)
toolbar.addAction(self.limit_action)
# 打包进线程(耗时的操作)
@staticmethod
def thread_it(func, *args):
t = threading.Thread(target=func, args=args)
t.setDaemon(True) # 守护--就算主界面关闭,线程也会留守后台运行(不对!)
t.start() # 启动
# t.join() # 阻塞--会卡死界面!
# 方式一
class MyWindow(QMainWindow):
def __init__(self):
super().__init__()
ui = Ui_Form()
ui.setupUi(self)
# 方式二
if __name__ == "__main__":
import sys
app = QApplication(sys.argv)
Form = QMainWindow()
ui = Ui_Form()
ui.setupUi(Form)
Form.show()
sys.exit(app.exec_())