forked from TrueCloudLab/xk6-frostfs
46 lines
1.6 KiB
Python
46 lines
1.6 KiB
Python
import uuid
|
|
|
|
from helpers.cmd import execute_cmd
|
|
|
|
|
|
def create_bucket(endpoint, versioning, location):
|
|
if location:
|
|
location = f"--create-bucket-configuration 'LocationConstraint={location}'"
|
|
bucket_name = str(uuid.uuid4())
|
|
|
|
cmd_line = f"aws --no-verify-ssl s3api create-bucket --bucket {bucket_name} " \
|
|
f"--endpoint http://{endpoint} {location}"
|
|
cmd_line_ver = f"aws --no-verify-ssl s3api put-bucket-versioning --bucket {bucket_name} " \
|
|
f"--versioning-configuration Status=Enabled --endpoint http://{endpoint} "
|
|
|
|
out, success = execute_cmd(cmd_line)
|
|
|
|
if not success and "succeeded and you already own it" not in out:
|
|
print(f" > Bucket {bucket_name} has not been created:\n{out}")
|
|
return False
|
|
|
|
print(f"cmd: {cmd_line}")
|
|
|
|
if versioning == "True":
|
|
out, success = execute_cmd(cmd_line_ver)
|
|
if not success:
|
|
print(f" > Bucket versioning has not been applied for bucket {bucket_name}:\n{out}")
|
|
else:
|
|
print(f" > Bucket versioning has been applied.")
|
|
|
|
print(f"Created bucket: {bucket_name} via endpoint {endpoint}")
|
|
return bucket_name
|
|
|
|
|
|
def upload_object(bucket, payload_filepath, endpoint):
|
|
object_name = str(uuid.uuid4())
|
|
|
|
cmd_line = f"aws --no-verify-ssl s3api put-object --bucket {bucket} --key {object_name} " \
|
|
f"--body {payload_filepath} --endpoint http://{endpoint}"
|
|
out, success = execute_cmd(cmd_line)
|
|
|
|
if not success:
|
|
print(f" > Object {object_name} has not been uploaded.")
|
|
return False
|
|
|
|
return bucket, endpoint, object_name
|