python - 七牛实用工具包

最近在搞资源上传，用脚本去上传180g的资源，以及后台要做资源之间的映射匹配什么鬼的。真的是有趣呢。

一边写脚本，一边总结出这个七牛工具包，才4个小函数。。

github：https://github.com/emaste-r/MyQiniuUtil/tree/master

# coding=utf-8
from qiniu import Auth, put_file, etag
from qiniu import BucketManager  # 构建鉴权对象


class MyQiniuUtil(object):
    """
        处理七牛的工具类
    """
    access_key = "your-access-key"
    secret_key = "your-secret-key"
    bucket_name = 'your-bucket-name'

    @classmethod
    def upload(cls, file_path):
        """
        上传文件
        :param file_path: 文件路径
        :return: key
        """
        # 构建鉴权对象
        q = Auth(cls.access_key, cls.secret_key)

        # 上传到七牛后保存的文件名
        key = None

        # 生成上传 Token，可以指定过期时间等
        token = q.upload_token(cls.bucket_name, key, 3600)

        # 要上传文件的本地路径
        ret, info = put_file(token, key, file_path)

        assert ret['hash'] == etag(file_path)
        return ret['key']

    @classmethod
    def get_img_info(cls, img_url):
        """
        获取七牛上图片的基本信息：长宽、后缀
        :param img_url:
        :return: tuple(长、宽、后缀)
        """
        info_url = img_url + '?imageInfo'  # 基本信息
        import requests
        r = requests.get(info_url)
        body = r.content
        r.close()
        import json
        info_dic = json.loads(body)
        height = info_dic['height']
        width = info_dic['width']
        suffix = info_dic['format']
        return height, width, suffix

    @classmethod
    def get_key(cls, file_path):
        """
        获取一个文件的key，七牛的算法是获取文件的hash值，使用的是 qiniu.etag()
        :param file_path:
        :return: key (str)
        """
        key = etag(file_path)
        return key

    @classmethod
    def check_file_exist(cls, file_path):
        """
        判断key在你空间中存在
        :param file_path:
        :param bucket_name:
        :return: True False
        """
        key = cls.get_key(file_path)

        # 初始化Auth状态
        q = Auth(cls.access_key, cls.secret_key)

        # 初始化BucketManager
        bucket = BucketManager(q)

        ret, info = bucket.stat(cls.bucket_name, key)

        text_body = info.text_body
        if 'error' in text_body:
            return False
        return True


if __name__ == '__main__':
    # 文件
    file_path = u"/Users/ouyang/Desktop/新增测试/净空法师/封面.jpg"

    # 上传, 返回key
    key = MyQiniuUtil.upload(file_path)
    print key

    # 获取文件key
    key = MyQiniuUtil.get_key(file_path)
    print key

    # 判断文件是否存在
    is_exist = MyQiniuUtil.check_file_exist(file_path)
    print is_exist