add other image server support

Author: hy6544

README.md

@@ -1,5 +1,10 @@
 # Markdown-img使用指南
 
+## 项目地址
+
+- pypi：<https://pypi.org/project/markdown-img-icexmoon/>
+- github：<https://github.com/icexmoon/markdown-img>
+
 ## 用途
 
 本程序用于扫描工作目录下的markdown文件，查找其中的本地图片并自动上传到`sm.ms`图床，并生成一个使用网络图片替换本地图片的markdown副本，存放于`当前工作目录/markdown_image`目录下。
@@ -9,10 +14,17 @@
 ## 注意事项
 
 - 本程序不会改变原始markdown文件，请放心使用。
+
 - 对于已经生成副本的原始文件，本程序不会再次处理，如果需要重新生成副本，请手动删除相应的已生成副本。
+
 - 本程序依赖于`sm.ms`图床，请自行注册账号并生成token。
+
+  > 已加入其它图床支持，详情见功能，不过默认依然使用sm.ms图床。
+
 - 因为`sm.ms`图床接口有调用限制，如果出现图片上传出错的情况，可能是上传频繁导致，请等待1分钟以上时间后重新使用本程序。
+
 - 目前本程序只支持windows。
+
 - `sm.ms`国内访问不算友好，生成的markdown拷贝立即在网络上发布可能会显示防盗链等图片挂掉的情况，那是因为国内CDN比较慢，等一段时间就好了。
 
 ## 安装
@@ -27,7 +39,15 @@ pip install markdown-img-icexmoon
 pip install --upgrade markdown-img-icexmoon
 ```
 
-## 使用方式
+## 功能
+
+### 查看帮助文档
+
+1. 执行`python -m markdown_img -h`。
+
+### 生成图床markdown副本
+
+本程序的主要功能，将扫描命令行工作目录下的markdown文件，会将其中的本地图片替换为图床图片后生成一个图床副本，生成的副本会存储在工作目录下的`markdown_img`文件夹中。
 
 1. 使用CMD定位到将要处理的markdown文件目录。
 2. 执行`python -m markdown_img`。
@@ -36,9 +56,47 @@ pip install --upgrade markdown-img-icexmoon
 5. 等待处理。
 6. 完毕后查看`工作目录/markdown_img`目录。
 
-> - 图床token存储在程序所在目录的`smms_token.config`文件中，如果需要修改的可以自行修改，也可以删除该配置文件后重新运行程序输入。
+> - 图床token存储在程序所在目录的`main.config`文件中，如果需要修改的可以自行修改，也可以删除该配置文件后重新运行程序输入。
+> - 目前仍不支持根据文件新旧程度重新生成markdown副本的功能，如果原文件改变，需要手动删除副本后重新生成。
 > - 稍后会丰富并完善程序的相关命令参数。
 
+### 从图床恢复本地图库
+
+如果图床副本完好，但本体markdown文件关联的本地图片丢失的，可以利用此功能尝试恢复。
+
+1. 切换命令行工作目录到要恢复的markdown文件目录。
+2. 执行`python -m markdown_img -m img_recove`
+3. 等待处理。
+4. 完毕后查看本地markdown文件图片是否恢复。
+
+> - 因为网络图床不稳定，虽然程序本身有超时重连机制，但如果恢复的图片过多，很可能处理中断，只需要重新运行程序即可。
+> - 恢复逻辑为对比副本和原本中的图片出现顺序，1对1恢复，所以务必保证两者没有差异，程序会在两者数量不同时中断并提示用户手动确认。
+
+### 切换图床服务
+
+可以切换图床服务，以备某个图床不可用或者访问不稳定。
+
+目前支持的图床有[**sm.ms**](https://sm.ms/)、阿里、[**如优**](https://img.rruu.net/)、[**Vim-CN**](https://img.vim-cn.com/)。
+
+1. 执行`python -m markdown_img -i ali`
+
+   > - 具体的图床标识可以查看帮助文档。
+   > - 需要访问令牌的图床服务切换后使用中会提示输入相应的访问令牌。
+
+## 更新日志
+
+### 0.0.6
+
+新增命令行参数功能，具体情况可使用`python -m markdown_img --help`进行查看。
+
+新增使用图床备份恢复本地图片库的功能，具体使用方法见帮助信息。
+
+### 0.1.1
+
+加入阿里图床、Vim-CN图床、如优图床的调用支持。
 
+其中阿里图床和如优图床通过如优的API调用实现，需要在如优图床官网申请账号并获取token。
 
+> 如优图床的容量为单账号10G，不过注册只需要邮箱，可以注册多个账号。
 
+加入从图床副本恢复本地markdown图片的功能，具体使用方式见帮助文档。

build/lib/markdown_img/__init__.py

@@ -1,8 +1,33 @@
-import os
+import getopt
 import sys
-part = __file__.rpartition('\\')
-packageDirPath = part[0]
-sys.path.append(packageDirPath)
-from main import Main
+import os
+from .main import Main
+try:
+    opts, args = getopt.gnu_getopt(sys.argv[1:], 'm:hi:', ['mode=', 'help','img_service='])
+except getopt.GetoptError as e:
+    print("获取参数信息出错，错误提示：", e.msg)
+    exit()
 mainProcess = Main()
-mainProcess.main()
+if len(opts) == 0:
+    mainProcess.main()
+else:
+    # print(opts)
+    for opt in opts:
+        argKey = opt[0]
+        argVal = opt[1]
+        if argKey == '--help' or argKey == '-h':
+            mainProcess.outputHelpInfo()
+        elif argKey == '--mode' or argKey == '-m':
+            mode = argVal
+            if mode == 'img_recove':
+                mainProcess.imgRecovery()
+            elif mode == 'normal':
+                mainProcess.main()
+            else:
+                print("错误的mode参数")
+        elif argKey == '--img_service' or argKey == '-i':
+            mainProcess.changeImgService(argVal)
+        else:
+            mainProcess.main()
+        break
+exit()

build/lib/markdown_img/config.py

@@ -1,9 +1,19 @@
 import os
-from user_exception import UserException
+from .user_exception import UserException
+import json
 
 
 class Config():
+    PARAM_SMMS_TOKEN = 'smms_token'
+    PARAM_RRUU_TOKEN = 'rruu_token'
+    PARAM_IMG_SERVICE = 'img_service'
+    IMG_SERVICE_SMMS = 'smms'
+    IMG_SERVICE_RRUU = 'rruu'
+    IMG_SERVICE_ALI = 'ali'
+    IMG_SERVICE_VIMCN = 'vimcn'
     smmsTokenFile = ""
+    configFile = ""
+    mainConfig = {}
 
     def __init__(self):
         pass
@@ -27,16 +37,60 @@ def getSmmsTokenFile(self):
             Config.smmsTokenFile = self.getCurrentDirPath()+'\\smms_token.config'
         return Config.smmsTokenFile
 
+    def __getConfigFile(self):
+        if Config.configFile == "":
+            Config.configFile = self.getCurrentDirPath()+'\\main.config'
+        return Config.configFile
+
+    def __getMainConfig(self, real=False):
+        if len(Config.mainConfig) == 0 or real == True:
+            if os.path.exists(self.__getConfigFile()):
+                with open(file=self.__getConfigFile(), mode='r', encoding='UTF-8') as fopen:
+                    Config.mainConfig = json.loads(fopen.read())
+            else:
+                # 不存在配置文件的，给予默认配置
+                Config.mainConfig['img_service'] = 'smms'
+        return Config.mainConfig
+
+    def getConfigParam(self, param):
+        '''获取配置参数'''
+        mainConfig = self.__getMainConfig()
+        if param in mainConfig:
+            return mainConfig[param]
+        else:
+            return ''
+
+    def setConfigParam(self, param, value):
+        '''设置配置参数'''
+        mainConfig = self.__getMainConfig()
+        mainConfig[param] = value
+
+    def writeMainConfig(self):
+        fopen = open(file=self.__getConfigFile(), mode='w', encoding='UTF-8')
+        mainConfig = self.__getMainConfig()
+        print(json.dumps(mainConfig), file=fopen)
+        fopen.close()
+
+    def getRruuToken(self):
+        token = self.getConfigParam(Config.PARAM_RRUU_TOKEN)
+        if token == '':
+            raise UserException(UserException.CODE_NO_RRUU_TOKEN)
+        return token
+
     def getSmmsToken(self):
-        if not os.path.exists(self.getSmmsTokenFile()):
-            raise UserException(UserException.CODE_NO_CONFIG)
-        configFileOpen = open(file=self.getSmmsTokenFile(), mode='r')
-        token = configFileOpen.readline()
-        configFileOpen.close()
-        if len(token) <= 4:
-            raise UserException(UserException.CODE_NO_CONFIG)
-        token = token[0:len(token)-1]
+        token = self.getConfigParam(Config.PARAM_SMMS_TOKEN)
+        if token == '':
+            raise UserException(UserException.CODE_NO_SMMS_TOKEN)
         return token
+        # if not os.path.exists(self.getSmmsTokenFile()):
+        #     raise UserException(UserException.CODE_NO_CONFIG)
+        # configFileOpen = open(file=self.getSmmsTokenFile(), mode='r')
+        # token = configFileOpen.readline()
+        # configFileOpen.close()
+        # if len(token) <= 4:
+        #     raise UserException(UserException.CODE_NO_CONFIG)
+        # token = token[0:len(token)-1]
+        # return token
 
     def writeSmmsToken(self, token: str):
         with open(file=self.getSmmsTokenFile(), mode='w') as configFileOpen:

build/lib/markdown_img/download_help.py

@@ -0,0 +1,34 @@
+import sys
+from urllib import request
+import requests
+from .user_exception import UserException
+
+
+class DownloadHelp():
+    def processReport(self, a, b, c):
+        per = 100.0*a*b/c
+        if per > 100:
+            per = 1
+        # sys.stdout.write(" "+"%.2f%% 已经下载的大小：%1d 文件大小：%1d"%(per,a*b,c)+'\r')
+        # sys.stdout.flush()
+
+    def download(self, url, fileName):
+        request.urlretrieve(url, filename=fileName,
+                            reporthook=self.processReport)
+
+    def chunkDownload(self, url, fileName, timeout=5):
+        i = 0
+        while True:
+            try:
+                r = requests.get(url, stream=True, timeout=timeout)
+                with open(fileName, 'wb') as f:
+                    for chunk in r.iter_content(chunk_size=32):
+                        f.write(chunk)
+                return
+            except requests.exceptions.RequestException:
+                #链接超时重试三次
+                i += 1
+                if i < 3:
+                    continue
+                else:
+                    raise UserException(UserException.CODE_TIMEOUT,"下载"+url+"超时，请重新再试")

build/lib/markdown_img/help.info

@@ -0,0 +1,9 @@
+-h  --help  输出帮助信息
+-m  --mode  选择运行模式
+    img_recove  使用图床的备份修复本地图片库
+    normal      正常模式
+-i  --img_service 选择图床服务商
+    smms    sm.ms图床，默认图床服务，稳定性好，访问速度一般。
+    ali     阿里图床，访问速度快（非正常提供服务，稳定性不保证）。
+    rruu    如优图床。
+    vimcn   Vim-CN图床，访问速度偏慢。