senset/vnpy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Mod] add open/save file function

Browse Source
This commit is contained in:
vn.py 2019-10-31 21:31:09 +08:00
parent 5febee2aa3
commit 990db038ee
1 changed files with 36 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

54
vnpy/trader/ui/editor.py
View File

@ -6,12 +6,14 @@ from PyQt5 import QtWidgets, Qsci, QtGui
class CodeEditor(QtWidgets.QMainWindow): class CodeEditor(QtWidgets.QMainWindow):
"""""" """"""
NEW_FILE_NAME = "Untitled"
def __init__(self): def __init__(self):
"""""" """"""
super().__init__() super().__init__()
self.editors: [str, Qsci.QsciScintilla] = {} self.new_file_count = 1
self.editor_path_map: Dict[Qsci.QsciScintilla, str] = {}
self.init_ui() self.init_ui()
@ -122,20 +124,27 @@ class CodeEditor(QtWidgets.QMainWindow):
return editor return editor
def open_editor(self, file_path: str): def open_editor(self, file_path: str = ""):
"""""" """"""
if file_path in self.editors: # Show editor tab if file already opened
editor = self.editors[file_path] if file_path:
for editor, path in self.editor_path_map.items():
if file_path == path:
editor.show() editor.show()
return return
# Otherwise create new editor
editor = self.new_editor() editor = self.new_editor()
self.editors[file_path] = editor self.editor_path_map[editor] = file_path
if file_path:
buf = open(file_path, encoding="UTF8").read() buf = open(file_path, encoding="UTF8").read()
editor.setText(buf) editor.setText(buf)
file_name = Path(file_path).name file_name = Path(file_path).name
else:
file_name = f"{self.NEW_FILE_NAME}-{self.new_file_count}"
self.new_file_count += 1
i = self.tab.addTab(editor, file_name) i = self.tab.addTab(editor, file_name)
self.tab.setCurrentIndex(i) self.tab.setCurrentIndex(i)
@ -155,31 +164,34 @@ class CodeEditor(QtWidgets.QMainWindow):
def new_file(self): def new_file(self):
"""""" """"""
file_path, _ = QtWidgets.QFileDialog.getSaveFileName( self.open_editor("")
self, "新建", "", "Python(*.py)")
if file_path:
open(file_path, "a").close()
self.open_editor(file_path)
def save_file(self): def save_file(self):
"""""" """"""
pass 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)")
if file_path:
self.editor_path_map[editor] = file_path
with open(file_path, "w") as f:
f.write(editor.text())
def copy(self): def copy(self):
"""""" """"""
editor = self.get_active_editor() self.get_active_editor().copy()
editor.copy()
def paste(self): def paste(self):
"""""" """"""
editor = self.get_active_editor() self.get_active_editor.paste()
editor.paste()
def cut(self): def cut(self):
"""""" """"""
editor = self.get_active_editor() self.get_active_editor.cut()
editor.cut()
def show_shortcut(self): def show_shortcut(self):
"""""" """"""
@ -195,9 +207,7 @@ class CodeEditor(QtWidgets.QMainWindow):
def get_active_editor(self): def get_active_editor(self):
"""""" """"""
for editor in self.editors.values(): return self.tab.currentWidget()
if editor.isVisible():
return editor
if __name__ == "__main__": if __name__ == "__main__":