class Bucket(object):
def __init__(self, connection):
log.debug('class: %s' % self.__class__.__name__)
self.connection = connection
self.add_io_info = AddIOInfo()
self.test_op_code = 'create'
def create(self, bucket_name, json_file):
log.debug('function: %s' % self.create.__name__)
log.info('in create bucket')
"""
:param bucket_name: string
:rtype: dict
:return: create_bucket_stack:
args:
1.status: True, bucket created or False if bucket creation failed
2.bucket: bucket objects
3.msgs: error messages
"""
try:
bucket = self.connection.create_bucket(bucket_name)
create_bucket_stack = {'status': True,
'bucket': bucket}
add_bucket_to_json = JBucket(json_file)
add_bucket_to_json.add(bucket_name)
self.add_io_info.add_bucket_info(self.connection.access_key, **{'bucket_name': bucket_name,
'test_op_code': self.test_op_code})
except (exception.AWSConnectionError, exception.BotoClientError, exception.S3ResponseError,
exception.S3CreateError, IOError), e:
log.error(e)
create_bucket_stack = {'status': False,
'msgs': e}
return create_bucket_stack
class Bucket(object):
def __init__(self, connection):
log.debug("class: %s" % self.__class__.__name__)
self.connection = connection
self.add_io_info = AddIOInfo()
self.test_op_code = "create"
def create(self, bucket_name, json_file):
log.debug("function: %s" % self.create.__name__)
log.info("in create bucket")
"""
:param bucket_name: string
:rtype: dict
:return: create_bucket_stack:
args:
1.status: True, bucket created or False if bucket creation failed
2.bucket: bucket objects
3.msgs: error messages
"""
try:
bucket = self.connection.create_bucket(bucket_name)
create_bucket_stack = {"status": True, "bucket": bucket}
add_bucket_to_json = JBucket(json_file)
add_bucket_to_json.add(bucket_name)
self.add_io_info.add_bucket_info(
self.connection.access_key, **{
"bucket_name": bucket_name,
"test_op_code": self.test_op_code
})
except (
exception.AWSConnectionError,
exception.BotoClientError,
exception.S3ResponseError,
exception.S3CreateError,
IOError,
) as e:
log.error(e)
create_bucket_stack = {"status": False, "msgs": e}
return create_bucket_stack
def get(self, bucket_name, json_file=None):
log.debug("function: %s" % self.get.__name__)
log.info("in get bucket")
"""
:param bucket_name:string
:rtype: dict
:return: get_bucket_stack:
args:
status: True, if got bucket or False, no get bucket
attribs: bucket object
msgs: error messges
"""
try:
bucket = self.connection.get_bucket(bucket_name)
if json_file is not None:
add_bucket_to_json = JBucket(json_file)
add_bucket_to_json.add(bucket_name)
get_bucket_stack = {"status": True, "bucket": bucket}
except (exception.S3ResponseError, exception.AWSConnectionError) as e:
log.error(e)
get_bucket_stack = {"status": False, "msgs": e}
return get_bucket_stack
def delete(self, bucket_name):
log.debug("function: %s" % self.delete.__name__)
log.info("in delete bucket")
"""
:param bucket_name: string
:rtype: dict
:return: delete_bucket_stack:
args:
status: True, if bucket is deleted or False if not deleted
msgs: error messages
"""
try:
self.connection.delete_bucket(bucket_name)
delete_bucket_stack = {"status": True}
except exception.S3ResponseError as e:
log.error(e)
delete_bucket_stack = {"status": False, "msgs": e}
return delete_bucket_stack
def enable_disable_versioning(self, bucket, enabled):
try:
bucket.configure_versioning(enabled)
versioning_status = bucket.get_versioning_status()
log.info(versioning_status)
return True
except (exception.S3ResponseError, exception.BotoClientError) as e:
log.error(e)
return False
def set_user_grant(self, bucket, grants):
"""
:param acls:
send acls in form of {'permission' : <permission type>, 'user_id' : canonical_user_id, 'recursive' : bool }
persmission type : (READ, WRITE, READ_ACP, WRITE_ACP, FULL_CONTROL)
:param bucket: buckect object
"""
if grants is not None:
try:
log.debug("setting grants %s" % grants)
bucket.add_user_grant(
permission=grants["permission"],
user_id=grants["user_id"],
recursive=grants["recursive"],
)
acp = bucket.get_acl()
for grant in acp.acl.grants:
log.info("grants set: %s on %s" %
(grant.permission, grant.id))
return True
except (exception.S3ResponseError, exception.BotoClientError) as e:
log.error(e)
return False
else:
log.info("not setting any acls")
def set_acls(self, bucket, acls):
"""
:param bucket: bucket objects
:param acls: canned acls : private, public-read, public-read-write, authenticated-read
:return:
"""
if acls is not None:
try:
log.info("got acl: %s" % acls)
bucket.set_acl(acls)
acp = bucket.get_acl()
for grant in acp.acl.grants:
log.info("canned acls set: %s on %s" %
(grant.permission, grant.id))
return True
except (exception.S3ResponseError, exception.BotoClientError) as e:
log.error(e)
return False
else:
log.info("not setting any acls")