xk6-frostfs/scenarios/preset/resolve_containers_in_preset.py

#!/usr/bin/python3

import argparse
import json
import http.client
import ssl

parser = argparse.ArgumentParser()
parser.add_argument('--endpoint', help='Endpoint of the S3 gateway')
parser.add_argument('--preset_file', help='JSON file path with s3 preset')

args = parser.parse_args()

def main():
    with open(args.preset_file) as f:
        preset_text = f.read()

    preset = json.loads(preset_text)

    conn = http.client.HTTPSConnection(args.endpoint, context = ssl._create_unverified_context())
    containers = []
    for bucket in preset.get('buckets'):
        conn.request("HEAD", f'/{bucket}')
        response = conn.getresponse()
        containers.append(response.getheader('X-Container-Id'))
        response.read()

    preset['containers'] = containers
    with open(args.preset_file, 'w+') as f:
        json.dump(preset, f, ensure_ascii=False, indent=2)

if __name__ == "__main__":
    main()