ybw/utils/aliyun.py

import oss2


# 远程bucket配置
oss_conf = {
    "key_id": "LTAI4G5x9aoZx8dDamQ7vfZi",
    "key_secret": "Bk98FsbPYXcJe72n1bG3Ssf73acuNh",
    # "endpoint": "oss-cn-beijing-internal.aliyuncs.com",
    "endpoint": "oss-cn-beijing.aliyuncs.com",
    "bucket_name": "jy-datafile"
}


class AliYunService:

    def __init__(self):
        self.__acc_key_id = oss_conf['key_id']
        self.__acc_key_secret = oss_conf['key_secret']
        self.__endpoint = oss_conf['endpoint']
        self.__bucket_name = oss_conf['bucket_name']

    def push_oss_from_local(self, key, filename):
        """
        上传一个本地文件到OSS的普通文件

        :param str key: 上传到OSS的文件名
        :param str filename: 本地文件名，需要有可读权限
        """
        auth = oss2.Auth(self.__acc_key_id, self.__acc_key_secret)
        bucket = oss2.Bucket(auth, self.__endpoint, self.__bucket_name)
        bucket.put_object_from_file(key, filename)

    def push_oss_from_stream(self, key, data):
        """
        流式上传oss

        :param str key: 上传到OSS的文件名
        :param data: 待上传的内容。
        :type data: bytes，str或file-like object
        """
        auth = oss2.Auth(self.__acc_key_id, self.__acc_key_secret)
        bucket = oss2.Bucket(auth, self.__endpoint, self.__bucket_name)
        bucket.put_object(key, data)