使用七牛接口实现图片和文件一键上传

Posted 2020-12-26

本文主要介绍使用七牛提供的接口把七牛云作为图床，用 Python 实现图片上传和文件上传的脚本实现，可以支持按文件夹批量上传。

搭建过网站的同学，一定都涉及到图片上传和文件上传的问题，最近为了弄图片资源接触了一下云存储，还挺挺好用。

之前如果我们把图片资源存储到本机服务器，那就每次写文章的时候都需要准备一堆的图片，并且要上传到对应的相对目录上去，如果用了云存储(不限七牛)，好处就是需要的资源直接上传，然后生成一个链接来使用就行了，Markdown 语法对链接的支持也很好，恩，可以搞一下。

不过，又有了新问题，就是是用云存储，得先经过登录->打开对象存储->打开内容管理->上传，然后填入上传后的相对目录，并手动选择本地文件，是不是很麻烦？

所以本次是提供了一个一键上传资源的脚本(仅针对七牛云)，实现的效果就是，输入项目目录和文件本地地址后，自动上传所有文件到云端，并生成可使用 URL。

来，直接上代码：

# -*- coding: utf-8 -*-

import os
import sys
from qiniu import Auth, put_file

if 3 != len(sys.argv):
    print(‘[Usage] %s [dir_set] [filepath]‘ % os.path.basename(sys.argv[0]))
    sys.exit(0)
else:
    # dir_set 的格式为 image/upload-qiniu/ ，注意末尾带反斜杠/
    dir_set = sys.argv[1]
    file_path = sys.argv[2]

# 个人中心->密匙管理->AK
access_key = ‘你的AccessKey‘
# 个人中心->密匙管理->SK
secret_key = ‘你的SecretKey‘
# 七牛空间名
bucket_name = ‘你的存储空间名‘

qiniu_auth = Auth(access_key, secret_key)

def upload_qiniu(input_path):
    #upload single file to qiniu
    filename = os.path.basename(input_path)
    key = ‘%s%s‘ % (dir_set, filename)

    token = qiniu_auth.upload_token(bucket_name, key)
    ret, info = put_file(token, key, input_path, check_crc=True)
    if ret and ret[‘key‘] == key:
        print(‘%s done‘ % (‘http://www.sylan215.com/‘ + dir_set + filename))
    else:
        print(‘%s error‘ % (‘http://www.sylan215.com/‘ + dir_set + filename))

def upload_all_files(input_path):
    if os.path.isfile(input_path):
        upload_qiniu(input_path)
    elif os.path.isdir(input_path):
        dirlist = os.walk(input_path)
        for root, dirs, files in dirlist:
            for filename in files:
                upload_qiniu(os.path.join(root, filename))
    else:
        print(‘Please input the exists file path!‘)

if __name__ == "__main__":
    upload_all_files(file_path)

使用说明

1. 请先安装 qiniu 的依赖库：

python -m pip install qiniu

2. 修改绑定域名

修改代码中的 http://www.sylan215.com 为自己在七牛云存储绑定的域名

3. 带两个参数执行脚本

第一个参数是上传后的目录，第二个参数是文件在本地的地址，支持文件夹。

其中第一个参数的目录格式为：image/test/。

比如如下的命令:

upload_to_qiniu.py image/test/ d:	est.png

运行后，上传后的文件地址就类似：http://www.sylan215.com/image/test/test.png

PS：已使用 Python2.7 和 Python3.4 亲测有效，其他 Python 版本如果有异常，请反馈。

更详细的说明请跳转到项目地址：https://github.com/sylan215/upload-to-qiniu，欢迎大家 star，并一起丰富这个脚本的功能。