Feature: Persist console history

Author: iewnfod

.gitignore

@@ -6,3 +6,4 @@ build/
 .idea/
 target/
 .vs/
+.history

main.py

@@ -7,6 +7,7 @@
 from src.parse import *
 import src.options as options
 import src.global_var as global_var
+from src.quit import quit
 
 from sys import argv, exit
 import os
@@ -16,16 +17,10 @@
 from ply import lex
 from chardet import detect
 
-# 尝试导入 readline，无法导入也不会导致核心功能受损
-try:
-    import readline
-    readline.clear_history()
-except:
-    pass
-
 
 preline = '>'
 multi_preline = '.'
+home_path = os.path.dirname(__file__)
 
 # 清除注释以及多余字符
 def remove_comment(text: str):
@@ -37,7 +32,7 @@ def remove_comment(text: str):
 
 # 预加载文件
 def preload_scripts():
-    scripts_path = os.path.join(os.path.dirname(__file__), 'scripts')
+    scripts_path = os.path.join(home_path, 'scripts')
     for p, _dir_list, file_list in os.walk(scripts_path):
         for i in file_list:
             path = os.path.join(p, i)
@@ -193,7 +188,7 @@ def main():
         main()
     except EOFError:
         print("EXIT")
-        exit(0)
+        quit(0)
     except KeyboardInterrupt:
         print('Keyboard Interrupt')
-        exit(0)
+        quit(0)

src/global_var.py

@@ -1,15 +1,18 @@
 from .error import Error, StackError
 from . import options
+from .history import Cmd
 
 error_messages = []
 running_mod = 'file'  # file / line
 running_path = ''  # 当前运行文件
+console = Cmd()
 
 # 变量
 def __init__():
     global error_messages, running_mod
     error_messages = []
     running_mod = 'file'
+    console.preloop()
 
 def add_error_message(msg, obj):
     error_messages.append(Error(msg, obj))

src/history.py

@@ -0,0 +1,26 @@
+import os
+
+# 尝试导入 readline，无法导入也不会导致核心功能受损
+try:
+    import readline
+except:
+    readline = None
+
+def parent_path(p):
+    return os.path.dirname(p)
+
+class Cmd:
+    def __init__(self, home_path=parent_path(parent_path(__file__)), save_path='.history', history_size=1000):
+        self.home_path = home_path
+        self.save_path = save_path
+        self.history_size = history_size
+        self.path = os.path.join(self.home_path, self.save_path)
+
+    def preloop(self):
+        if readline and os.path.exists(self.path):
+            readline.read_history_file(self.path)
+
+    def postloop(self):
+        if readline:
+            readline.set_history_length(self.history_size)
+            readline.write_history_file(self.path)

src/quit.py

@@ -1,6 +1,7 @@
 from os import _exit
-from .global_var import output_error
+from .global_var import output_error, console
 
 def quit(code):
     output_error()
+    console.postloop()
     _exit(code)