From 6659719e2da4a428827a88d0f1bed6bd7ecddd74 Mon Sep 17 00:00:00 2001
From: "vn.py" <xiaoyou.chen@foxmail.com>
Date: Mon, 24 Jun 2019 13:29:22 +0800
Subject: [PATCH] [Add]mainwindow toolbar for apps

---
 vnpy/trader/ui/mainwindow.py | 43 ++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/vnpy/trader/ui/mainwindow.py b/vnpy/trader/ui/mainwindow.py
index a728885f..3313c780 100644
--- a/vnpy/trader/ui/mainwindow.py
+++ b/vnpy/trader/ui/mainwindow.py
@@ -50,6 +50,7 @@ class MainWindow(QtWidgets.QMainWindow):
         """"""
         self.setWindowTitle(self.window_title)
         self.init_dock()
+        self.init_toolbar()
         self.init_menu()
         self.load_window_setting("custom")
 
@@ -113,6 +114,9 @@ class MainWindow(QtWidgets.QMainWindow):
             self.add_menu_action(
                 app_menu, app.display_name, icon_path, func
             )
+            self.add_toolbar_action(
+                app.display_name, icon_path, func
+            )
 
         # Global setting editor
         action = QtWidgets.QAction("配置", self)
@@ -128,6 +132,11 @@ class MainWindow(QtWidgets.QMainWindow):
             "contract.ico",
             partial(self.open_widget, ContractManager, "contract"),
         )
+        self.add_toolbar_action(
+            "查询合约",
+            "contract.ico",
+            partial(self.open_widget, ContractManager, "contract")
+        )
 
         self.add_menu_action(
             help_menu, "还原窗口", "restore.ico", self.restore_window_setting
@@ -140,6 +149,9 @@ class MainWindow(QtWidgets.QMainWindow):
         self.add_menu_action(
             help_menu, "社区论坛", "forum.ico", self.open_forum
         )
+        self.add_toolbar_action(
+            "社区论坛", "forum.ico", self.open_forum
+        )
 
         self.add_menu_action(
             help_menu,
@@ -148,6 +160,22 @@ class MainWindow(QtWidgets.QMainWindow):
             partial(self.open_widget, AboutDialog, "about"),
         )
 
+    def init_toolbar(self):
+        """"""
+        self.toolbar = QtWidgets.QToolBar(self)
+        self.toolbar.setFloatable(False)
+        self.toolbar.setMovable(False)
+
+        # Set button size
+        w = 40
+        size = QtCore.QSize(w, w)
+        self.toolbar.setIconSize(size)
+
+        # Set button spacing
+        self.toolbar.layout().setSpacing(10)
+
+        self.addToolBar(QtCore.Qt.LeftToolBarArea, self.toolbar)
+
     def add_menu_action(
         self,
         menu: QtWidgets.QMenu,
@@ -164,6 +192,21 @@ class MainWindow(QtWidgets.QMainWindow):
 
         menu.addAction(action)
 
+    def add_toolbar_action(
+        self,
+        action_name: str,
+        icon_name: str,
+        func: Callable,
+    ):
+        """"""
+        icon = QtGui.QIcon(get_icon_path(__file__, icon_name))
+
+        action = QtWidgets.QAction(action_name, self)
+        action.triggered.connect(func)
+        action.setIcon(icon)
+
+        self.toolbar.addAction(action)
+
     def create_dock(
         self, widget_class: QtWidgets.QWidget, name: str, area: int
     ):