openstack-helm-infra/helm-toolkit/templates/scripts/_create-s3-bucket.py.tpl
Steve Wilkerson a3f444299e HTK: Add s3 user/bucket scripts, snippets, manifests
This proposes adding the following:

Snippets for the environment variables for the s3 admin user and
service users for using rgw's s3 api

Scripts for creating s3 users for use by a particular service and
for creating and linking buckets to those users

Manifest templates for the jobs for creating the s3 users and for
creating and linking the buckets to those users

Change-Id: Ibd5ed0aac49d172c56faffdacd44bdd487978570
2018-09-19 15:52:36 -05:00

95 lines
3.1 KiB
Smarty

{{/*
Copyright 2017 The Openstack-Helm Authors.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/}}
{{- define "helm-toolkit.scripts.create_s3_bucket" }}
#!/usr/bin/env python
import os
import sys
import logging
import rgwadmin
import rgwadmin.exceptions
# Create logger, console handler and formatter
logger = logging.getLogger('OpenStack-Helm S3 Bucket')
logger.setLevel(logging.DEBUG)
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
# Set the formatter and add the handler
ch.setFormatter(formatter)
logger.addHandler(ch)
# Get S3 admin user's access key
if "S3_ADMIN_ACCESS_KEY" in os.environ:
access_key = os.environ['S3_ADMIN_ACCESS_KEY']
logger.info('Found S3 admin access key')
else:
logger.critical('S3 admin access key environment variable not set')
sys.exit(1)
# Get S3 admin user's secret key
if "S3_ADMIN_SECRET_KEY" in os.environ:
secret_key = os.environ['S3_ADMIN_SECRET_KEY']
logger.info('Found S3 admin secret key')
else:
logger.critical('S3 admin secret key environment variable not set')
sys.exit(1)
# Get RGW S3 host endpoint
if "RGW_HOST" in os.environ:
server = os.environ['RGW_HOST']
logger.info('Found RGW S3 host endpoint')
else:
logger.critical('RGW S3 host endpoint environment variable not set')
sys.exit(1)
# Get name of S3 user to link to bucket
if "S3_USERNAME" in os.environ:
s3_user = os.environ['S3_USERNAME']
logger.info('Found S3 user name')
else:
logger.critical('S3 user name environment variable not set')
sys.exit(1)
# Get name of bucket to create for user link
if "S3_BUCKET" in os.environ:
s3_bucket = os.environ['S3_BUCKET']
logger.info('Found S3 bucket name')
else:
logger.critical('S3 bucket name environment variable not set')
sys.exit(1)
try:
rgw_admin = rgwadmin.RGWAdmin(access_key, secret_key, server, secure=False)
try:
rgw_admin.get_bucket(bucket=s3_bucket,uid=s3_user)
except (rgwadmin.exceptions.NoSuchBucket, rgwadmin.exceptions.NoSuchKey), e:
rgw_admin.create_bucket(bucket=s3_bucket)
bucket = rgw_admin.get_bucket(bucket=s3_bucket)
bucket_id = bucket['id']
rgw_admin.link_bucket(bucket=s3_bucket, bucket_id=bucket_id, uid=s3_user)
logger.info("Created bucket {} and linked it to user {}".format(s3_bucket, s3_user))
sys.exit(0)
else:
logger.info("The bucket {} exists for user {}! Exiting without creating a new bucket!".format(s3_bucket, s3_user))
except rgwadmin.exceptions.InvalidArgument:
logger.critical("Invalid arguments supplied for rgwadmin connection. Please check your s3 keys and endpoint")
sys.exit(1)
{{- end }}