添加图床配置修改子命令

Author: hy6544

README.md

@@ -56,7 +56,7 @@ pip install --upgrade markdown-img-icexmoon
 
 1. 执行`python -m markdown_img -h`。
 
-### 查看版本信息
+### 查看版本及配置信息
 
 执行`pymdimg -v`或`pymdimg --version`。
 
@@ -105,7 +105,9 @@ pip install --upgrade markdown-img-icexmoon
 
 1. 执行`python -m markdown_img -c smms`
 
-> 具体参数可以查看帮助文档。
+> - 具体参数可以查看帮助文档。
+>
+> - 部分图床配置如果是多项，可以使用子命令仅修改其中单一配置，比如仅修改腾讯云的存储目录：`pymdimg -c qcloud --des_dir image`，如果目标目录中间有空格，需要给将其用英文双引号包起来，比如这样：`pymdimg -c qcloud --des_dir "我 love 你"`。更新完配置后可以使用`pymdimg -v`确认配置是否已经设置正确。
 
 ### 扫描图片并创建索引
 
@@ -201,4 +203,10 @@ python -m markdown_img -m refresh
 
 ### 0.2.4
 
-添加了腾讯云对象存储作为图床。
+添加了腾讯云对象存储作为图床。
+
+### 0.2.5
+
+- 添加了修改腾讯云图床的子命令。
+- 修改查看版本命令，增加程序相关配置信息的显示。
+- 修改帮助文档。

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = markdown-img-icexmoon
-version = 0.2.4
+version = 0.2.5
 author = icexmoon
 author_email = [email protected]
 description = A program for find and upload images in markdown file and will replace them.

src/markdown_img/__main__.py

@@ -4,10 +4,17 @@
 from .main import Main
 
 
+def getOptionVal(options, key):
+    for optKey, optVal in options:
+        if optKey == key:
+            return optVal
+    return None
+
+
 def main():
     try:
         opts, args = getopt.gnu_getopt(sys.argv[1:], 'm:hvi:c:s', [
-            'mode=', 'help', 'version', 'img_service=', 'change_token=', 'scan'])
+            'mode=', 'help', 'version', 'img_service=', 'change_token=', 'scan', 'des_dir='])
     except getopt.GetoptError as e:
         print("获取参数信息出错，错误提示：", e.msg)
         exit()
@@ -21,10 +28,7 @@ def main():
             if argKey == '--help' or argKey == '-h':
                 mainProcess.outputHelpInfo()
             elif argKey == '--version' or argKey == '-v':
-                import pkg_resources
-                version = pkg_resources.get_distribution(
-                    'markdown-img-icexmoon').version
-                print("version:{}".format(version))
+                mainProcess.printSysInfo()
             elif argKey == '--mode' or argKey == '-m':
                 mode = argVal
                 if mode == 'img_recove':
@@ -38,7 +42,12 @@ def main():
             elif argKey == '--img_service' or argKey == '-i':
                 mainProcess.changeImgService(argVal)
             elif argKey == '--change_token' or argKey == '-c':
-                mainProcess.changeToken(argVal)
+                optDesDirVal = getOptionVal(opts, '--des_dir')
+                if argVal == 'qcloud' and optDesDirVal is not None:
+                    mainProcess.changeImgServiceOption(
+                        argVal, {'des_dir': optDesDirVal})
+                else:
+                    mainProcess.changeToken(argVal)
             elif argKey == '--scan' or argKey == '-s':
                 mainProcess.scanAndCreateIndex()
             else:

src/markdown_img/help.info

@@ -11,10 +11,12 @@
     rruu    如优图床。
     vimcn   Vim-CN图床，访问速度偏慢。
     yujian  遇见图床。
-    qcloud  腾讯云对象存储（需要提供必要连接信息，如果不知道如何获取，请阅读https://cloud.tencent.com/document/product/436/7751#.E6.9C.AF.E8.AF.AD.E4.BF.A1.E6.81.AF）
+    qcloud  腾讯云对象存储（需要提供必要连接信息）
 -c  --change_token 替换图床访问令牌
     rruu    如优图床。
     smms    sm.ms图床。
     yujian  遇见图床。
     qcloud  腾讯云对象存储
+        支持子选项
+        --des_dir 仅更新腾讯云oss上的存储目录
 -s  --scan 扫描当前目录的图片并生成网络图床索引

src/markdown_img/main.py

@@ -321,3 +321,49 @@ def scanAndCreateIndex(self):
                 print("!["+imgName+"]("+webImgUrl+")\n", file=fopen)
         print("已成功生成网络图床索引文件：markdown_img_index.md")
         return True
+
+    def changeImgServiceOption(self, imgServiceFlag, options):
+        '''修改图床服务的部分配置'''
+        sysConfig = Config()
+        if imgServiceFlag == Config.IMG_SERVICE_QCLOUD:
+            try:
+                qCloudInfo = sysConfig.getQCloudInfo()
+            except UserException as e:
+                self.dealUserException(e)
+            if Config.QCLOUD_INFO_DES_DIR in options and options[Config.QCLOUD_INFO_DES_DIR]:
+                qCloudInfo[Config.QCLOUD_INFO_DES_DIR] = options[Config.QCLOUD_INFO_DES_DIR]
+            sysConfig.setConfigParam(Config.PARAM_QCLOUD_INFO, qCloudInfo)
+        sysConfig.writeMainConfig()
+        print('图床配置已更新')
+
+    def printSysInfo(self):
+        '''打印当前系统相关信息'''
+        sysConfig = Config()
+        import pkg_resources
+        version = pkg_resources.get_distribution(
+            'markdown-img-icexmoon').version
+        print("软件版本：{}".format(version))
+        try:
+            imgService = sysConfig.getConfigParam(Config.PARAM_IMG_SERVICE)
+            print("当前使用的图床：{}".format(imgService))
+            print("图床的相关配置信息：")
+            if imgService!= Config.IMG_SERVICE_QCLOUD:
+                if imgService == Config.IMG_SERVICE_ALI or imgService == Config.IMG_SERVICE_YUJIAN:
+                    token = sysConfig.getYujianToken()
+                elif imgService == Config.IMG_SERVICE_ALI2 or imgService == Config.IMG_SERVICE_RRUU:
+                    token = sysConfig.getRruuToken()
+                elif imgService == Config.IMG_SERVICE_SMMS:
+                    token = sysConfig.getSmmsToken()
+                else:
+                    token = ''
+                print("\t访问令牌：{}".format(token))
+            else:
+                #显示腾讯云相关配置信息
+                qcloudInfo = sysConfig.getQCloudInfo()
+                print("\t存储桶：{}".format(qcloudInfo[Config.QCLOUD_INFO_BUCKET]))
+                print("\tsecret_id:{}".format(qcloudInfo[Config.QCLOUD_INFO_SECRET_ID]))
+                print("\tsecret_key:{}".format(qcloudInfo[Config.QCLOUD_INFO_SECRET_KEY]))
+                print("\t地域：{}".format(qcloudInfo[Config.QCLOUD_INFO_REGION]))
+                print("\t存储目录：{}".format(qcloudInfo[Config.QCLOUD_INFO_DES_DIR]))
+        except UserException as e:
+            self.dealUserException(e)