✨ support for udpating acces tokens through the config

Author: InnoFang

code_counter/conf/config.py

@@ -43,6 +43,8 @@ def invoke(self, args):
         if args.restore:
             self.restore()
         else:
+            if any([args.github_token, args.gitee_token]):
+                self.__update_access_token(args.github_token, args.gitee_token)
             if any([args.suffix_reset, args.comment_reset, args.ignore_reset]):
                 self.__reset_config(args.suffix_reset, args.comment_reset, args.ignore_reset)
             if any([args.suffix_add, args.comment_add, args.ignore_add]):
@@ -59,6 +61,16 @@ def __confirm(self, tips):
         check = input(tips)
         return check.strip().lower() == 'y'
 
+    def __update_access_token(self, github_access_token, gitee_access_token):
+        if github_access_token:
+            if self.__confirm("the old Github access token will be updated to `{}` . (y/n) ".format(github_access_token)):
+                self.access_tokens.github = github_access_token
+        if gitee_access_token:
+            if self.__confirm("the old Gitee access token will be updated to `{}` . (y/n) ".format(gitee_access_token)):
+                self.access_tokens.gitee = gitee_access_token
+
+        self.__update()
+
     def __reset_config(self, suffix_reset, comment_reset, ignore_reset):
         if suffix_reset:
             if self.__confirm("'suffix' will be replaced with {} . (y/n) ".format(suffix_reset)):

code_counter/core/args.py

@@ -177,6 +177,11 @@ def __config(self):
         parser.add_argument('--ignore-del', dest='ignore_del', type=split_args,
                             help="delete some values of the 'ignore' in the config")
 
+        parser.add_argument('--github-token', dest='github_token',
+                            help="update the Github access token")
+        parser.add_argument('--gitee-token', dest='gitee_token',
+                            help="update the Gitee access token")
+
         parser.add_argument('--restore', dest='restore', action='store_true',
                             help="restore default config")