Merge pull request #2172 from vnpy/dev-editor

Dev editor

requirements.txt

@@ -19,3 +19,4 @@ ibapi
 deap
 pyzmq
 wmi
+QScintilla

vnpy/app/cta_backtester/engine.py

@@ -4,6 +4,7 @@ import traceback
 from datetime import datetime
 from threading import Thread
 from pathlib import Path
+from inspect import getfile
 
 from vnpy.event import Event, EventEngine
 from vnpy.trader.engine import BaseEngine, MainEngine
@@ -116,6 +117,12 @@ class BacktesterEngine(BaseEngine):
             msg = f"策略文件{module_name}加载失败，触发异常：\n{traceback.format_exc()}"
             self.write_log(msg)
 
+    def reload_strategy_class(self):
+        """"""
+        self.classes.clear()
+        self.load_strategy_class()
+        self.write_log("策略文件重载刷新完成")
+
     def get_strategy_class_names(self):
         """"""
         return list(self.classes.keys())
@@ -425,3 +432,9 @@ class BacktesterEngine(BaseEngine):
     def get_history_data(self):
         """"""
         return self.backtesting_engine.history_data
+
+    def get_strategy_class_file(self, class_name: str):
+        """"""
+        strategy_class = self.classes[class_name]
+        file_path = getfile(strategy_class)
+        return file_path

vnpy/app/cta_backtester/ui/widget.py

@@ -13,6 +13,7 @@ from vnpy.trader.constant import Interval, Direction
 from vnpy.trader.engine import MainEngine
 from vnpy.trader.ui import QtCore, QtWidgets, QtGui
 from vnpy.trader.ui.widget import BaseMonitor, BaseCell, DirectionCell, EnumCell
+from vnpy.trader.ui.editor import CodeEditor
 from vnpy.event import Event, EventEngine
 from vnpy.chart import ChartWidget, CandleItem, VolumeItem
 
@@ -117,6 +118,12 @@ class BacktesterManager(QtWidgets.QWidget):
         self.candle_button.clicked.connect(self.show_candle_chart)
         self.candle_button.setEnabled(False)
 
+        edit_button = QtWidgets.QPushButton("代码编辑")
+        edit_button.clicked.connect(self.edit_strategy_code)
+
+        reload_button = QtWidgets.QPushButton("策略重载")
+        reload_button.clicked.connect(self.reload_strategy_class)
+
         for button in [
             backtesting_button,
             optimization_button,
@@ -125,7 +132,9 @@ class BacktesterManager(QtWidgets.QWidget):
             self.order_button,
             self.trade_button,
             self.daily_button,
-            self.candle_button
+            self.candle_button,
+            edit_button,
+            reload_button
         ]:
             button.setFixedHeight(button.sizeHint().height() * 2)
 
@@ -142,18 +151,24 @@ class BacktesterManager(QtWidgets.QWidget):
         form.addRow("回测资金", self.capital_line)
         form.addRow("合约模式", self.inverse_combo)
 
+        result_grid = QtWidgets.QGridLayout()
+        result_grid.addWidget(self.trade_button, 0, 0)
+        result_grid.addWidget(self.order_button, 0, 1)
+        result_grid.addWidget(self.daily_button, 1, 0)
+        result_grid.addWidget(self.candle_button, 1, 1)
+
         left_vbox = QtWidgets.QVBoxLayout()
         left_vbox.addLayout(form)
         left_vbox.addWidget(backtesting_button)
         left_vbox.addWidget(downloading_button)
         left_vbox.addStretch()
-        left_vbox.addWidget(self.trade_button)
+        left_vbox.addLayout(result_grid)
-        left_vbox.addWidget(self.order_button)
-        left_vbox.addWidget(self.daily_button)
-        left_vbox.addWidget(self.candle_button)
         left_vbox.addStretch()
         left_vbox.addWidget(optimization_button)
         left_vbox.addWidget(self.result_button)
+        left_vbox.addStretch()
+        left_vbox.addWidget(edit_button)
+        left_vbox.addWidget(reload_button)
 
         # Result part
         self.statistics_monitor = StatisticsMonitor()
@@ -197,6 +212,9 @@ class BacktesterManager(QtWidgets.QWidget):
         hbox.addWidget(self.chart)
         self.setLayout(hbox)
 
+        # Code Editor
+        self.editor = CodeEditor(self.main_engine, self.event_engine)
+
     def register_event(self):
         """"""
         self.signal_log.connect(self.process_log_event)
@@ -403,6 +421,21 @@ class BacktesterManager(QtWidgets.QWidget):
 
         self.candle_dialog.exec_()
 
+    def edit_strategy_code(self):
+        """"""
+        class_name = self.class_combo.currentText()
+        file_path = self.backtester_engine.get_strategy_class_file(class_name)
+
+        self.editor.open_editor(file_path)
+        self.editor.show()
+
+    def reload_strategy_class(self):
+        """"""
+        self.backtester_engine.reload_strategy_class()
+
+        self.class_combo.clear()
+        self.init_strategy_settings()
+
     def show(self):
         """"""
         self.showMaximized()

vnpy/trader/ui/editor.py

@@ -0,0 +1,454 @@
+from typing import Callable, Dict
+from pathlib import Path
+
+from PyQt5 import QtWidgets, Qsci, QtGui
+
+
+class CodeEditor(QtWidgets.QMainWindow):
+    """"""
+    NEW_FILE_NAME = "Untitled"
+
+    _instance = None
+
+    def __new__(cls, *args, **kwargs):
+        """"""
+        if not cls._instance:
+            cls._instance = QtWidgets.QMainWindow.__new__(cls, *args, **kwargs)
+        return cls._instance
+
+    def __init__(self, main_engine=None, event_engine=None):
+        """"""
+        super().__init__()
+
+        self.new_file_count = 0
+        self.editor_path_map: Dict[Qsci.QsciScintilla, str] = {}
+
+        self.init_ui()
+
+    def init_ui(self):
+        """"""
+        self.setWindowTitle("策略编辑器")
+
+        self.init_menu()
+        self.init_central()
+
+    def init_central(self):
+        """"""
+        self.tab = QtWidgets.QTabWidget()
+        self.tab.currentChanged.connect(self.update_path_label)
+
+        self.path_label = QtWidgets.QLabel()
+
+        vbox = QtWidgets.QVBoxLayout()
+        vbox.addWidget(self.tab)
+        vbox.addWidget(self.path_label)
+
+        widget = QtWidgets.QWidget()
+        widget.setLayout(vbox)
+
+        self.setCentralWidget(widget)
+
+    def init_menu(self):
+        """"""
+        bar = self.menuBar()
+
+        file_menu = bar.addMenu("文件")
+        self.add_menu_action(file_menu, "新建文件", self.new_file, "Ctrl+N")
+        self.add_menu_action(file_menu, "打开文件", self.open_file, "Ctrl+O")
+        self.add_menu_action(file_menu, "关闭文件", self.close_editor, "Ctrl+W")
+        file_menu.addSeparator()
+        self.add_menu_action(file_menu, "保存", self.save_file, "Ctrl+S")
+        self.add_menu_action(
+            file_menu,
+            "另存为",
+            self.save_file_as,
+            "Ctrl+Shift+S"
+        )
+        file_menu.addSeparator()
+        self.add_menu_action(file_menu, "退出", self.close)
+
+        edit_menu = bar.addMenu("编辑")
+        self.add_menu_action(edit_menu, "撤销", self.undo, "Ctrl+Z")
+        self.add_menu_action(edit_menu, "恢复", self.redo, "Ctrl+Y")
+        edit_menu.addSeparator()
+        self.add_menu_action(edit_menu, "复制", self.copy, "Ctrl+C")
+        self.add_menu_action(edit_menu, "粘贴", self.paste, "Ctrl+P")
+        self.add_menu_action(edit_menu, "剪切", self.cut, "Ctrl+X")
+        edit_menu.addSeparator()
+        self.add_menu_action(edit_menu, "查找", self.find, "Ctrl+F")
+        self.add_menu_action(edit_menu, "替换", self.replace, "Ctrl+H")
+
+    def add_menu_action(
+        self,
+        menu: QtWidgets.QMenu,
+        action_name: str,
+        func: Callable,
+        shortcut: str = "",
+    ):
+        """"""
+        action = QtWidgets.QAction(action_name, self)
+        action.triggered.connect(func)
+        menu.addAction(action)
+
+        if shortcut:
+            sequence = QtGui.QKeySequence(shortcut)
+            action.setShortcut(sequence)
+
+    def new_editor(self):
+        """"""
+        # Create editor object
+        editor = Qsci.QsciScintilla()
+
+        # Set editor font
+        font = QtGui.QFont()
+        font.setFamily('Consolas')
+        font.setFixedPitch(True)
+        font.setPointSize(10)
+
+        editor.setFont(font)
+        editor.setMarginsFont(font)
+
+        # Set margin for line numbers
+        font_metrics = QtGui.QFontMetrics(font)
+        editor.setMarginWidth(0, font_metrics.width("00000") + 6)
+        editor.setMarginLineNumbers(0, True)
+        editor.setMarginsBackgroundColor(QtGui.QColor("#cccccc"))
+
+        # Set brace matching
+        editor.setBraceMatching(Qsci.QsciScintilla.SloppyBraceMatch)
+
+        # Hide horizontal scroll bar
+        editor.SendScintilla(Qsci.QsciScintilla.SCI_SETHSCROLLBAR, 0)
+
+        # Set current line color
+        editor.setCaretLineVisible(True)
+        editor.setCaretLineBackgroundColor(QtGui.QColor("#ffe4e4"))
+
+        # Set Python lexer
+        lexer = Qsci.QsciLexerPython()
+        lexer.setDefaultFont(font)
+        editor.setLexer(lexer)
+
+        # Add minimum editor size
+        editor.setMinimumSize(600, 450)
+
+        # Enable auto complete
+        editor.setAutoCompletionSource(Qsci.QsciScintilla.AcsAll)
+        editor.setAutoCompletionThreshold(2)
+        editor.setAutoCompletionCaseSensitivity(False)
+        editor.setAutoCompletionReplaceWord(False)
+
+        # Use space indentation
+        editor.setIndentationsUseTabs(False)
+        editor.setTabWidth(4)
+        editor.setIndentationGuides(True)
+
+        # Enable folding
+        editor.setFolding(True)
+
+        return editor
+
+    def open_editor(self, file_path: str = ""):
+        """"""
+        # Show editor tab if file already opened
+        if file_path:
+            file_path = str(Path(file_path))
+
+            for editor, path in self.editor_path_map.items():
+                if file_path == path:
+                    editor.show()
+                    return
+
+        # Otherwise create new editor
+        editor = self.new_editor()
+
+        if file_path:
+            buf = open(file_path, encoding="UTF8").read()
+            editor.setText(buf)
+            file_name = Path(file_path).name
+
+            self.editor_path_map[editor] = file_path
+        else:
+            self.new_file_count += 1
+            file_name = f"{self.NEW_FILE_NAME}-{self.new_file_count}"
+
+            self.editor_path_map[editor] = file_name
+
+        i = self.tab.addTab(editor, file_name)
+        self.tab.setCurrentIndex(i)
+
+        self.path_label.setText(file_path)
+
+    def close_editor(self):
+        """"""
+        i = self.tab.currentIndex()
+
+        # Close editor if last file closed
+        if not i:
+            self.close()
+        # Otherwise only close current tab
+        else:
+            self.save_file()
+
+            editor = self.get_active_editor()
+            self.editor_path_map.pop(editor)
+
+            self.tab.removeTab(i)
+
+    def open_file(self):
+        """"""
+        file_path, _ = QtWidgets.QFileDialog.getOpenFileName(
+            self, "打开文件", "", "Python(*.py)")
+
+        if file_path:
+            self.open_editor(file_path)
+
+    def new_file(self):
+        """"""
+        self.open_editor("")
+
+    def save_file(self):
+        """"""
+        editor = self.get_active_editor()
+        file_path = self.editor_path_map[editor]
+
+        if self.NEW_FILE_NAME in file_path:
+            file_path, _ = QtWidgets.QFileDialog.getSaveFileName(
+                self, "保存", "", "Python(*.py)")
+
+        self.save_editor_text(editor, file_path)
+
+    def save_file_as(self):
+        """"""
+        editor = self.get_active_editor()
+
+        file_path, _ = QtWidgets.QFileDialog.getSaveFileName(
+            self, "保存", "", "Python(*.py)")
+
+        self.save_editor_text(editor, file_path)
+
+    def save_editor_text(self, editor: Qsci.QsciScintilla, file_path: str):
+        """"""
+        if file_path:
+            self.editor_path_map[editor] = file_path
+
+            i = self.tab.currentIndex()
+            file_name = Path(file_path).name
+            self.tab.setTabText(i, file_name)
+
+            with open(file_path, "w", encoding="UTF8") as f:
+                f.write(editor.text())
+
+        self.update_path_label()
+
+    def copy(self):
+        """"""
+        self.get_active_editor().copy()
+
+    def paste(self):
+        """"""
+        self.get_active_editor().paste()
+
+    def undo(self):
+        """"""
+        self.get_active_editor().undo()
+
+    def redo(self):
+        """"""
+        self.get_active_editor().redo()
+
+    def cut(self):
+        """"""
+        self.get_active_editor().cut()
+
+    def find(self):
+        """"""
+        dialog = FindDialog(
+            self.get_active_editor(),
+            False
+        )
+        dialog.exec_()
+
+    def replace(self):
+        """"""
+        dialog = FindDialog(
+            self.get_active_editor(),
+            True
+        )
+        dialog.exec_()
+
+    def get_active_editor(self):
+        """"""
+        return self.tab.currentWidget()
+
+    def closeEvent(self, event):
+        """"""
+        for editor, path in self.editor_path_map.items():
+            i = QtWidgets.QMessageBox.question(
+                self,
+                "退出保存",
+                f"是否要保存{path}？",
+                QtWidgets.QMessageBox.Save | QtWidgets.QMessageBox.No | QtWidgets.QMessageBox.Cancel,
+                QtWidgets.QMessageBox.Save
+            )
+
+            if i == QtWidgets.QMessageBox.Save:
+                if self.NEW_FILE_NAME in path:
+                    path, _ = QtWidgets.QFileDialog.getSaveFileName(
+                        self, "保存", "", "Python(*.py)")
+
+                if path:
+                    self.save_editor_text(editor, path)
+            elif i == QtWidgets.QMessageBox.Cancel:
+                break
+
+        event.accept()
+
+    def show(self):
+        """"""
+        if not self.tab.count():
+            self.open_editor()
+
+        self.showMaximized()
+
+    def update_path_label(self):
+        """"""
+        editor = self.get_active_editor()
+        path = self.editor_path_map[editor]
+        self.path_label.setText(path)
+
+
+class FindDialog(QtWidgets.QDialog):
+    """"""
+
+    def __init__(
+        self,
+        editor: Qsci.QsciScintilla,
+        show_replace: bool = False
+    ):
+        """"""
+        super().__init__()
+
+        self.editor: Qsci.QsciScintilla = editor
+        self.show_replace: bool = show_replace
+        self.new_task: bool = True
+
+        self.init_ui()
+
+    def init_ui(self):
+        """"""
+        find_label = QtWidgets.QLabel("查找")
+        replace_label = QtWidgets.QLabel("替换")
+
+        selected_text = self.editor.selectedText()
+        self.find_line = QtWidgets.QLineEdit(selected_text)
+        self.find_line.textChanged.connect(self.reset_task)
+
+        self.replace_line = QtWidgets.QLineEdit()
+
+        self.case_check = QtWidgets.QCheckBox("大小写")
+        self.case_check.setChecked(True)
+        self.case_check.stateChanged.connect(self.reset_task)
+
+        self.whole_check = QtWidgets.QCheckBox("全词匹配")
+        self.whole_check.stateChanged.connect(self.reset_task)
+
+        self.selection_check = QtWidgets.QCheckBox("选中区域")
+        self.selection_check.stateChanged.connect(self.reset_task)
+
+        find_button = QtWidgets.QPushButton("查找")
+        find_button.clicked.connect(self.find_text)
+
+        self.replace_button = QtWidgets.QPushButton("替换")
+        self.replace_button.clicked.connect(self.replace_text)
+        self.replace_button.setEnabled(False)
+
+        check_hbox = QtWidgets.QHBoxLayout()
+        check_hbox.addWidget(self.case_check)
+        check_hbox.addStretch()
+        check_hbox.addWidget(self.whole_check)
+        check_hbox.addStretch()
+        check_hbox.addWidget(self.selection_check)
+        check_hbox.addStretch()
+
+        button_hbox = QtWidgets.QHBoxLayout()
+        button_hbox.addWidget(find_button)
+        button_hbox.addWidget(self.replace_button)
+
+        form = QtWidgets.QFormLayout()
+        form.addRow(find_label, self.find_line)
+        form.addRow(replace_label, self.replace_line)
+        form.addRow(check_hbox)
+        form.addRow(button_hbox)
+
+        self.setLayout(form)
+
+        if self.show_replace:
+            self.setWindowTitle("替换")
+        else:
+            self.setWindowTitle("查找")
+
+            replace_label.setVisible(False)
+            self.replace_line.setVisible(False)
+            self.replace_button.setVisible(False)
+
+    def find_text(self):
+        """"""
+        if not self.new_task:
+            result = self.editor.findNext()
+
+            if result:
+                self.new_task = False
+                self.replace_button.setEnabled(True)
+                return
+            else:
+                self.new_task = True
+
+        self.editor.cancelFind()
+
+        if not self.selection_check.isChecked():
+            result = self.editor.findFirst(
+                self.find_line.text(),
+                False,
+                self.case_check.isChecked(),
+                self.whole_check.isChecked(),
+                False,
+                line=1,
+                index=1
+            )
+
+        else:
+            result = self.editor.findFirstInSelection(
+                self.find_line.text(),
+                False,
+                self.case_check.isChecked(),
+                self.whole_check.isChecked(),
+                False
+            )
+
+        if result:
+            self.new_task = False
+            self.replace_button.setEnabled(True)
+        else:
+            self.new_task = True
+
+    def replace_text(self):
+        """"""
+        new_text = self.replace_line.text()
+
+        self.editor.replace(new_text)
+        self.editor.findNext()
+
+    def reset_task(self):
+        """"""
+        self.new_task = True
+        self.replace_button.setEnabled(False)
+
+
+if __name__ == "__main__":
+    from vnpy.trader.ui import create_qapp
+
+    app = create_qapp()
+    editor = CodeEditor()
+    editor.show()
+    app.exec_()

vnpy/trader/ui/mainwindow.py

@@ -24,6 +24,7 @@ from .widget import (
     AboutDialog,
     GlobalDialog
 )
+from .editor import CodeEditor
 from ..engine import MainEngine
 from ..utility import get_icon_path, TRADER_DIR
 
@@ -138,6 +139,18 @@ class MainWindow(QtWidgets.QMainWindow):
             partial(self.open_widget, ContractManager, "contract")
         )
 
+        self.add_menu_action(
+            help_menu,
+            "代码编辑",
+            "editor.ico",
+            partial(self.open_widget, CodeEditor, "editor")
+        )
+        self.add_toolbar_action(
+            "代码编辑",
+            "editor.ico",
+            partial(self.open_widget, CodeEditor, "editor")
+        )
+
         self.add_menu_action(
             help_menu, "还原窗口", "restore.ico", self.restore_window_setting
         )