コード例 #1
0
ファイル: oss.py プロジェクト: zj793039327/django-aliyun-oss
    def __init__(
        self,
        bucket=OSS_STORAGE_BUCKET_NAME,
        access_key=None,
        secret_key=None,
        acl=DEFAULT_ACL,
        # calling_format=CALLING_FORMAT,
        encrypt=False,
        # gzip=IS_GZIPPED,
        # gzip_content_types=GZIP_CONTENT_TYPES,
        # preload_metadata=PRELOAD_METADATA
    ):
        self.bucket = bucket
        self.acl = acl

        if not access_key and not secret_key:
            access_key, secret_key = self._get_access_keys()

        self.connection = OssAPI(ACCESS_ADDRESS, access_key, secret_key)
        self.headers = HEADERS
コード例 #2
0
    def __init__(self, bucket=OSS_STORAGE_BUCKET_NAME,
                access_key=None,
                secret_key=None,
                acl=DEFAULT_ACL,
                # calling_format=CALLING_FORMAT,
                encrypt=False,
                # gzip=IS_GZIPPED,
                # gzip_content_types=GZIP_CONTENT_TYPES,
                # preload_metadata=PRELOAD_METADATA
            ):
        self.bucket = bucket
        self.acl = acl

        if not access_key and not secret_key:
            access_key, secret_key = self._get_access_keys()

        self.connection = OssAPI(ACCESS_ADDRESS, access_key, secret_key)
        self.headers = HEADERS
コード例 #3
0
ファイル: oss.py プロジェクト: KleeTaurus/django-aliyun-oss
class OSSStorage(Storage):
    """Aliyun Open Storage Service"""

    def __init__(self, bucket=OSS_STORAGE_BUCKET_NAME,
                 access_key=None,
                 secret_key=None,
                 acl=DEFAULT_ACL,
                 # calling_format=CALLING_FORMAT,
                 encrypt=False,
                 # gzip=IS_GZIPPED,
                 # gzip_content_types=GZIP_CONTENT_TYPES,
                 # preload_metadata=PRELOAD_METADATA
                 ):

        self.bucket = bucket
        self.acl = acl

        if not access_key and not secret_key:
            access_key, secret_key = self._get_access_keys()

        self.connection = OssAPI(ACCESS_ADDRESS, access_key, secret_key)
        self.headers = HEADERS

    def _get_access_keys(self):
        access_key = ACCESS_KEY_NAME
        secret_key = SECRET_KEY_NAME
        if (access_key or secret_key) and (not access_key or not secret_key):
            access_key = os.environ.get(ACCESS_KEY_NAME)
            secret_key = os.environ.get(SECRET_KEY_NAME)

        if access_key and secret_key:
            return access_key, secret_key

        return None, None

    def _clean_name(self, name):
        # Useful for windows' paths
        return os.path.join(BUCKET_PREFIX, os.path.normpath(name).replace('\\', '/'))

    def _put_file(self, name, content, content_type=None):
        if content_type:
            pass
        else:
            content_type = mimetypes.guess_type(name)[0] or 'application/x-octet-stream'

        self.headers.update({
            'x-oss-acl': self.acl,
            'Content-Type': content_type,
            'Content-Length': str(len(content)),
        })
        fp = StringIO(content)
        response = self.connection.put_object_from_fp(self.bucket, name, fp, content_type, self.headers)
        if (response.status / 100) != 2:
            raise IOError('OSSStorageError: %s' % response.read())

    def _open(self, name, mode='rb'):
        name = self._clean_name(name)
        remote_file = OSSStorageFile(name, self, mode=mode)
        return remote_file

    def _read(self, name, start_range=None, end_range=None):
        name = self._clean_name(name)
        if start_range is None:
            headers = {}
        else:
            headers = {'Range': 'bytes=%s-%s' % (start_range, end_range)}
        response = self.connection.get_object(self.bucket, name, headers)
        if (response.status / 100) != 2:
            raise IOError('OSSStorageError: %s' % response.read())

        header_map = convert_header2map(response.getheaders())
        content_len = safe_get_element('content-length', header_map)
        etag = safe_get_element('etag', header_map).upper()
        return response.read(), etag, content_len

    def _save(self, name, content):
        name = self._clean_name(name)
        content.open()
        if hasattr(content, 'chunks'):
            content_str = ''.join(chunk for chunk in content.chunks())
        else:
            content_str = content.read()
        self._put_file(name, content_str)
        return name

    def delete(self, name):
        name = self._clean_name(name)
        response = self.connection.delete_object(self.bucket, name)
        if response.status != 204:
            raise IOError('OSSStorageError: %s' % response.read())

    def exists(self, name):
        name = self._clean_name(name)
        response = self.connection.head_object(self.bucket, name)
        return response.status == 200

    def size(self, name):
        name = self._clean_name(name)
        response = self.connection.head_object(self.bucket, name)
        header_map = convert_header2map(response.getheaders())
        content_len = safe_get_element('content-length', header_map)
        return content_len and int(content_len) or 0

    def url(self, name):
        name = self._clean_name(name)
        return '%s%s' % (settings.MEDIA_URL, name)

    def modified_time(self, name):
        try:
            from dateutil import parser, tz
        except ImportError:
            raise NotImplementedError()
        name = self._clean_name(name)
        response = self.connection.head_object(self.bucket, name)
        header_map = convert_header2map(response.getheaders())
        last_modified = response.getheader('Last-Modified')
        # convert to string to date
        last_modified_date = parser.parse(last_modified)
        # if the date has no timzone, assume UTC
        if last_modified_date.tzinfo is None:
            last_modified_date = last_modified_date.replace(tzinfo=tz.tzutc())
        # convert date to local time w/o timezone
        return last_modified_date.astimezone(tz.tzlocal()).replace(tzinfo=None)

    ## UNCOMMENT BELOW IF NECESSARY
    #def get_available_name(self, name):
    #    """ Overwrite existing file with the same name. """
    #    name = self._clean_name(name)
    #    return name

    def copy_to_file(self, name, target):
        name = self._clean_name(name)
        response = self.connection.get_object_to_file(self.bucket, name, target)
        if response.status / 100 != 2:
            raise IOError('OSSStorageError: %s' % response.read())

    def save_file(self, filename, name):
        name = self._clean_name(name)
        response = self.connection.put_object_from_file(self.bucket, name, filename=filename, headers=self.headers)
        if response.status / 100 != 2:
            raise IOError('OSSStorageError: %s' % response.read())
コード例 #4
0
ファイル: oss.py プロジェクト: zj793039327/django-aliyun-oss
class OSSStorage(Storage):
    """Aliyun Open Storage Service"""
    def __init__(
        self,
        bucket=OSS_STORAGE_BUCKET_NAME,
        access_key=None,
        secret_key=None,
        acl=DEFAULT_ACL,
        # calling_format=CALLING_FORMAT,
        encrypt=False,
        # gzip=IS_GZIPPED,
        # gzip_content_types=GZIP_CONTENT_TYPES,
        # preload_metadata=PRELOAD_METADATA
    ):
        self.bucket = bucket
        self.acl = acl

        if not access_key and not secret_key:
            access_key, secret_key = self._get_access_keys()

        self.connection = OssAPI(ACCESS_ADDRESS, access_key, secret_key)
        self.headers = HEADERS

    def _get_access_keys(self):
        access_key = ACCESS_KEY_NAME
        secret_key = SECRET_KEY_NAME
        if (access_key or secret_key) and (not access_key or not secret_key):
            access_key = os.environ.get(ACCESS_KEY_NAME)
            secret_key = os.environ.get(SECRET_KEY_NAME)

        if access_key and secret_key:
            return access_key, secret_key

        return None, None

    def _clean_name(self, name):
        # Useful for windows' paths
        return os.path.join(BUCKET_PREFIX,
                            os.path.normpath(name).replace('\\', '/'))

    def _put_file(self, name, content, content_type=None):
        if content_type:
            pass
        else:
            content_type = mimetypes.guess_type(
                name)[0] or "application/x-octet-stream"

        self.headers.update({
            'x-oss-acl': self.acl,
            'Content-Type': content_type,
            'Content-Length': str(len(content)),
        })
        fp = BytesIO(content)
        logger.debug('')
        response = self.connection.put_object_from_fp(self.bucket, name, fp,
                                                      content_type,
                                                      self.headers)
        if (response.status / 100) != 2:
            raise IOError("OSSStorageError: %s" % response.read())

    def _open(self, name, mode='rb'):
        name = self._clean_name(name)
        remote_file = OSSStorageFile(name, self, mode=mode)
        return remote_file

    def _read(self, name, start_range=None, end_range=None):
        name = self._clean_name(name)
        if start_range is None:
            headers = {}
        else:
            headers = {'Range': 'bytes=%s-%s' % (start_range, end_range)}
        response = self.connection.get_object(self.bucket, name, headers)
        if int(response.status / 100) != 2:
            raise IOError("OSSStorageError: %s, %s" %
                          (response.status, response.read()))

        header_map = convert_header2map(response.getheaders())
        # content_len = safe_get_element("content-length", header_map)
        content_len = safe_get_element("content-range", header_map)
        etag = safe_get_element("etag", header_map).upper()
        return response.read(), etag, content_len

    def _save(self, name, content):
        name = self._clean_name(name)
        content.open()
        if hasattr(content, 'chunks'):
            content_str = b''.join(chunk for chunk in content.chunks())
        else:
            content_str = content.read()
        self._put_file(name, content_str)
        return name

    def delete(self, name):
        name = self._clean_name(name)
        response = self.connection.delete_object(self.bucket, name)
        if response.status != 204:
            raise IOError("OSSStorageError: %s" % response.read())

    def exists(self, name):
        name = self._clean_name(name)
        response = self.connection.head_object(self.bucket, name)
        return response.status == 200

    def size(self, name):
        name = self._clean_name(name)
        response = self.connection.head_object(self.bucket, name)
        header_map = convert_header2map(response.getheaders())
        content_len = safe_get_element("content-length", header_map)
        return content_len and int(content_len) or 0

    def url(self, name):
        name = self._clean_name(name)
        return '%s%s' % (settings.MEDIA_URL, name)

    def modified_time(self, name):
        try:
            from dateutil import parser, tz
        except ImportError:
            raise NotImplementedError()
        name = self._clean_name(name)
        response = self.connection.head_object(self.bucket, name)
        # header_map = convert_header2map(response.getheaders())
        last_modified = response.getheader('Last-Modified')
        # convert to string to date
        last_modified_date = parser.parse(last_modified)
        # if the date has no timzone, assume UTC
        if last_modified_date.tzinfo == None:
            last_modified_date = last_modified_date.replace(tzinfo=tz.tzutc())
        # convert date to local time w/o timezone
        return last_modified_date.astimezone(tz.tzlocal()).replace(tzinfo=None)

    ## UNCOMMENT BELOW IF NECESSARY
    # def get_available_name(self, name):
    #    """ Overwrite existing file with the same name. """
    #    name = self._clean_name(name)
    #    return name

    def copy_to_file(self, name, target):
        name = self._clean_name(name)
        response = self.connection.get_object_to_file(self.bucket, name,
                                                      target)
        if response.status / 100 != 2:
            raise IOError("OSSStorageError: %s" % response.read())

    def save_file(self, filename, name):
        name = self._clean_name(name)
        response = self.connection.put_object_from_file(self.bucket,
                                                        name,
                                                        filename=filename,
                                                        headers=self.headers)
        if response.status / 100 != 2:
            raise IOError("OSSStorageError: %s" % response.read())