def __init__(self, bucket=settings.AWS_STORAGE_BUCKET_NAME,
access_key=None, secret_key=None, acl='public-read',
is_secure=settings.AWS_IS_SECURE,
calling_format=CallingFormat.PATH):
self.bucket = bucket
self.acl = acl
if not access_key and not secret_key:
access_key, secret_key = self._get_access_keys()
host = getattr(settings, AWS_HOST, S3.DEFAULT_HOST)
self.connection = AWSAuthConnection(access_key, secret_key,
is_secure=is_secure, calling_format=calling_format,
server=host)
self.generator = QueryStringAuthGenerator(access_key, secret_key,
calling_format=calling_format, is_secure=False,
server=host)
self.headers = getattr(settings, AWS_HEADERS, {})
class S3Storage(Storage):
"""Amazon Simple Storage Service"""
def __init__(self, bucket=settings.AWS_STORAGE_BUCKET_NAME,
access_key=None, secret_key=None, acl='public-read',
is_secure=settings.AWS_IS_SECURE,
calling_format=CallingFormat.PATH):
self.bucket = bucket
self.acl = acl
if not access_key and not secret_key:
access_key, secret_key = self._get_access_keys()
host = getattr(settings, AWS_HOST, S3.DEFAULT_HOST)
self.connection = AWSAuthConnection(access_key, secret_key,
is_secure=is_secure, calling_format=calling_format,
server=host)
self.generator = QueryStringAuthGenerator(access_key, secret_key,
calling_format=calling_format, is_secure=False,
server=host)
self.headers = getattr(settings, AWS_HEADERS, {})
def _get_access_keys(self):
access_key = getattr(settings, ACCESS_KEY_NAME, None)
secret_key = getattr(settings, SECRET_KEY_NAME, None)
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:
# Both were provided, so use them
return access_key, secret_key
return None, None
def _get_connection(self):
return AWSAuthConnection(*self._get_access_keys())
def _put_file(self, name, content):
content_type = guess_type(name)[0] or "application/x-octet-stream"
self.headers.update({'x-amz-acl': self.acl, 'Content-Type': content_type})
response = self.connection.put(self.bucket, name, content, self.headers)
if response.http_response.status != 200:
raise IOError("File was not opened with write access. %s", response.message)
def _open(self, name, mode='rb'):
remote_file = S3StorageFile(name, self, mode=mode)
return remote_file
def _read(self, name, start_range=None, end_range=None):
if start_range is None:
headers = {}
else:
headers = {'Range': 'bytes=%s-%s' % (start_range, end_range)}
response = self.connection.get(self.bucket, name, headers)
headers = response.http_response.msg
return response.object.data, headers['etag'], headers.get('content-range', None)
def _save(self, name, content):
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):
self.connection.delete(self.bucket, name)
def exists(self, name):
response = self.connection._make_request('HEAD', self.bucket, name)
return response.status == 200
def size(self, name):
response = self.connection._make_request('HEAD', self.bucket, name)
content_length = response.getheader('Content-Length')
return content_length and int(content_length) or 0
def url(self, name):
return self.generator.make_bare_url(self.bucket, name)
