Add update

Author: iewnfod

README.md

@@ -35,6 +35,7 @@ cpc [file_path] [options]
 | `-t` | `--time` | To show the time for the script to run |
 | `-v` | `--version` | To show the version of this interpreter |
 | `-ne` | `--no-error` | To remove all error messages |
+| `-u` | `--update` | To update the version (only useful when using a version equal or greater than `0.1.2` and installed by git) |
 
 ### 常见问题
 #### 出现 `Import Error`

VERSION

@@ -0,0 +1 @@
+0.1.2

main.py

@@ -7,6 +7,7 @@
 from src.parse import *
 import src.options as options
 import src.global_var as global_var
+from src.history import HOME_PATH
 from src.quit import quit
 
 from sys import argv, exit
@@ -20,7 +21,7 @@
 
 preline = '>'
 multi_preline = '.'
-home_path = os.path.dirname(__file__)
+home_path = HOME_PATH
 
 # 清除注释以及多余字符
 def remove_comment(text: str):

src/history.py

@@ -9,8 +9,10 @@
 def parent_path(p):
     return os.path.dirname(p)
 
+HOME_PATH = parent_path(parent_path(__file__))
+
 class Cmd:
-    def __init__(self, home_path=parent_path(parent_path(__file__)), save_path='.history', history_size=1000):
+    def __init__(self, home_path=HOME_PATH, save_path='.history', history_size=1000):
         self.home_path = home_path
         self.save_path = save_path
         self.history_size = history_size

src/options.py

@@ -1,7 +1,7 @@
 from sys import exit
 import platform
+from .update import VERSION, update
 
-VERSION = '0.1.1'
 PLATFORM = f'[ {platform.python_implementation()} {platform.python_version()} ] on {platform.system()}'
 
 options_dict = {
@@ -75,12 +75,17 @@ def show_keywords():
 def remove_error():
     options_dict['show_error'] = False
 
+def update_version():
+    update()
+    quit(0)
+
 arguments = [  # 输入参数: (参数简写, 参数全称, 运行函数, 描述)
     ('-gt', '--get-tree', get_tree, 'To show the tree of the program after being parsed'),
     ('-v', '--version', version, 'To show the version of this interpreter'),
     ('-h', '--help', help, 'To show this help page'),
     ('-p', '--parse', open_parse_info, 'To show parse information during running'),
     ('-t', '--time', get_time, 'To show the time for the script to run'),
     ('-k', '--keywords', show_keywords, 'To show all the keywords'),
-    ('-ne', '--no-error', remove_error, 'To remove all error messages')
+    ('-ne', '--no-error', remove_error, 'To remove all error messages'),
+    ('-u', '--update', update_version, 'To check or update the version (only work if you are installing with git)')
 ]

src/quit.py

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

src/requirements.py

@@ -1,14 +1,15 @@
 requirements = [
-    "ply",
-    "chardet",
+    ('ply', 'ply'),
+    ('chardet', 'chardet'),
+    ('GitPython', 'git')
 ]
 
 def config():
     import importlib
     import pip
-    for require in requirements:
+    for package_name, import_name in requirements:
         try:
-            importlib.import_module(require)
+            importlib.import_module(import_name)
         except:
-            print(f'\033[1mMissing Important Dependence `{require}`\nTrying to Install for You...\033[0m')
-            pip.main(['install', require])
+            print(f'\033[1mMissing Important Dependence `{package_name}`\nTrying to Install for You...\033[0m')
+            pip.main(['install', package_name])

src/update.py

@@ -0,0 +1,36 @@
+from .history import HOME_PATH
+import os
+import git
+
+VERSION = ''
+
+with open(os.path.join(HOME_PATH, 'VERSION'), 'r') as f:
+    VERSION = f.read().strip()
+
+flag = False
+
+def check_update(repo: git.Repo, remote: git.Remote):
+    global flag
+    remote.fetch()
+    status = repo.is_dirty()
+    flag = True
+    return status
+
+def update():
+    repo = git.Repo(HOME_PATH)
+    remote = repo.remote()
+
+    print('Checking Update...', end='\r')
+    if check_update(repo, remote):
+        # 询问是否更新
+        u = input('There is a new version of the program. Do you want to update it? [y/N] ').strip().lower()
+        if u == 'y':
+            try:
+                remote.pull()
+                print('\033[1mUpdate Successful\033[0m')
+            except Exception as e:
+                print(f'\033[1;31mFailed to Update\033[0m\n\t{e}')
+        else:
+            print('Stop Updating')
+    else:
+        print('You are using the latest version!')