python实现图片上传到图床

需求背景:

因为最近写blog用markdown,本地写图片不知上传到哪里合适,有以下几个备选方案:

1. 自己搭一个图床服务器(pass, 因为要考虑持久性)
2. github(pass, 因为项目必须开源,挂载git上实在太难看了)
3. 简书留一个图床的文章(pass, 操作过于繁琐, 还要切浏览器)
4. 公有图床(应用方案)

最后选用的图床api是: SM.MS, api简单易用, 可以自主上传和删除, 自15年开始运营至现在,持久性还不错

功能实现:

• 封装api
  没什么好说的,python的requests包,分别封装了上传,最近上传列表和删除接口,代码如下:

LIST_URL = "https://sm.ms/api/list"
UPLOAD_URL = "https://sm.ms/api/upload"
CLEAR_URL = "https://sm.ms/api/clear"

def get_history():
    params = {'ssl': 0, 'format': 'json'}
    r = requests.get(LIST_URL, params)
    print(r.json())

def upload(filePath):
    if not os.path.isfile(filePath):
        print('要上传的文件不存在')
        return
    data = {'smfile': open(filePath, 'rb')}
    r = requests.post(UPLOAD_URL, files=data)
    print(r.json())
    log(r.json())

def clear_history():
    r = requests.get(CLEAR_URL);
    print(r.json())
    log(r.json())

• 日志记录:
  因为文件的链接和删除链接都在上传成功后返回的json中,需要记录上传后的结果, 这里做的比较简单, 做了一个日志文件追加写入.

• 参数解析:
  因为要做成脚本运行,需要对参数进行解析,python原生的获取参数的方法是sys.argv,没有参数解析和校验,这里使用自带的getopt包解析参数.

getopt包的使用方法:

语法格式: opt, arg = getopt.getopt(args, options [,long_options])
参数说明:

• args: 要解析的命令行参数, 用sys获取时需要从第一个参数获取, 因为第0个参数是启动的文件名, 既args = sys.argv[1:]
• options: 以字符串格式定义的-短参数,option后的(:)表示选项需要有附件参数, 不带(:) 表示选项不需要附加参数
• long_options: 已列表形式定义的--长参数, long_options后的(=) 表示选项需要附加参数
• 返回一个由(opt,value)元组的列表和一个参数列表

脚本运行方法:

• 上传文件:
  python3 gallery.py -u [filepath]
  运行结果:
  image
• 查看最近上传列表:
  python3 gallery.py -l
  image
• 删除上传文件
  python3 gallery.py -d
  image

完整的脚本代码:

#!/usr/bin/env python3
import os
import sys
import getopt
from datetime import date, datetime

import requests
LIST_URL = "https://sm.ms/api/list"
UPLOAD_URL = "https://sm.ms/api/upload"
CLEAR_URL = "https://sm.ms/api/clear"
LOG_FILE_PATH = "gallery.log"

def get_history():
    params = {'ssl': 0, 'format': 'json'}
    r = requests.get(LIST_URL, params)
    print(r.json())

def upload(filePath):
    if not os.path.isfile(filePath):
        print('要上传的文件不存在')
        return
    data = {'smfile': open(filePath, 'rb')}
    r = requests.post(UPLOAD_URL, files=data)
    print(r.json())
    log(r.json())

def clear_history():
    r = requests.get(CLEAR_URL);
    print(r.json())
    log(r.json())

def log(message):
    logFile = os.path.join(os.path.curdir, LOG_FILE_PATH)
    with open(logFile, 'w+') as f:
        f.write(datetime.now().strftime('%Y-%m-%d %H:%M:%S') + '%s%s' % (message, '\r\n'))

if __name__ == '__main__':
    args = sys.argv[1:]
    try:
        opts, arg = getopt.getopt(args, "u:ld", ["upload=",  "list", "delete"])
    except getopt.GetoptError:
        print('参数解析错误')

    for opt, arg in opts:
        if opt in ('-u', '--upload'):
            upload(arg)
        elif opt in ('-l', '--list'):
            get_history()
        elif opt in ('-d', '--delete'):
            clear_history()
        else:
            print('''参数错误, 参数选项为: 
            上传文件: -u | --upload=[filePath] 
            查看最近上传列表: -l | --list 
            删除最近上传: -d | --delete ''')

附:

参数解析的参考用例

import sys
import getopt


def main(argv):
    """
        通过 getopt模块 来识别参数demo,  http://blog.csdn.net/ouyang_peng/
    """
    
    username = ""
    password = ""

    try:
        """
            options, args = getopt.getopt(args, shortopts, longopts=[])
            
            参数args：一般是sys.argv[1:]。过滤掉sys.argv[0]，它是执行脚本的名字，不算做命令行参数。
            参数shortopts：短格式分析串。例如："hp:i:"，h后面没有冒号，表示后面不带参数；p和i后面带有冒号，表示后面带参数。
            参数longopts：长格式分析串列表。例如：["help", "ip=", "port="]，help后面没有等号，表示后面不带参数；ip和port后面带冒号，表示后面带参数。
            
            返回值options是以元组为元素的列表，每个元组的形式为：(选项串, 附加参数)，如：('-i', '192.168.0.1')
            返回值args是个列表，其中的元素是那些不含'-'或'--'的参数。
        """
        opts, args = getopt.getopt(argv, "hu:p:", ["help", "username=", "password="])
    except getopt.GetoptError:
        print('Error: test_arg.py -u <username> -p <password>')
        print('   or: test_arg.py --username=<username> --password=<password>')
        sys.exit(2)

    # 处理 返回值options是以元组为元素的列表。
    for opt, arg in opts:
        if opt in ("-h", "--help"):
            print('test_arg.py -u <username> -p <password>')
            print('or: test_arg.py --username=<username> --password=<password>')
            sys.exit()
        elif opt in ("-u", "--username"):
            username = arg
        elif opt in ("-p", "--password"):
            password = arg
    print('username为：', username)
    print('password为：', password)

    # 打印 返回值args列表，即其中的元素是那些不含'-'或'--'的参数。
    for i in range(0, len(args)):
        print('参数 %s 为：%s' % (i + 1, args[i]))


if __name__ == "__main__":
    # sys.argv[1:]为要处理的参数列表，sys.argv[0]为脚本名，所以用sys.argv[1:]过滤掉脚本名。
    main(sys.argv[1:])