# -*- coding: utf-8 -*- # # Copyright 2017 Google LLC. All Rights Reserved. # # 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. """Creates a database for a Cloud SQL instance.""" from googlecloudsdk.api_lib.sql import api_util from googlecloudsdk.api_lib.sql import exceptions from googlecloudsdk.api_lib.sql import operations from googlecloudsdk.api_lib.sql import validate from googlecloudsdk.calliope import base from googlecloudsdk.command_lib.sql import flags from googlecloudsdk.core import log from googlecloudsdk.core import properties @base.ReleaseTracks( base.ReleaseTrack.GA, base.ReleaseTrack.BETA, base.ReleaseTrack.ALPHA ) @base.UniverseCompatible class AddDatabase(base.Command): """Creates a database for a Cloud SQL instance.""" @staticmethod def Args(parser): """Args is called by calliope to gather arguments for this command. Args: parser: An argparse parser that you can use to add arguments that go on the command line after this command. Positional arguments are allowed. """ flags.AddDatabaseName(parser) flags.AddCharset(parser) flags.AddCollation(parser) flags.AddInstance(parser) base.ASYNC_FLAG.AddToParser(parser) def Run(self, args): """Creates a database for a Cloud SQL instance. Args: args: argparse.Namespace, The arguments that this command was invoked with. Returns: A dict object representing the operations resource describing the create operation if the create was successful. """ client = api_util.SqlClient(api_util.API_VERSION_DEFAULT) sql_client = client.sql_client sql_messages = client.sql_messages validate.ValidateInstanceName(args.instance) instance_ref = client.resource_parser.Parse( args.instance, params={'project': properties.VALUES.core.project.GetOrFail}, collection='sql.instances') new_database = sql_messages.Database( kind='sql#database', project=instance_ref.project, instance=instance_ref.instance, name=args.database, charset=args.charset, collation=args.collation) # TODO(b/35386183): Move this API call logic. result_operation = sql_client.databases.Insert( sql_messages.SqlDatabasesInsertRequest( database=new_database, instance=instance_ref.instance, project=instance_ref.project, ) ) operation_ref = client.resource_parser.Create( 'sql.operations', operation=result_operation.name, project=instance_ref.project) if args.async_: result = sql_client.operations.Get( sql_messages.SqlOperationsGetRequest( project=operation_ref.project, operation=operation_ref.operation)) else: try: operations.OperationsV1Beta4.WaitForOperation( sql_client, operation_ref, 'Creating Cloud SQL database') except exceptions.OperationError: log.Print('Database creation failed. Check if a database named {0} ' 'already exists.'.format(args.database)) # Must fail with non-zero exit code on API request failure. # TODO(b/35156765): Refactor. raise result = new_database result.kind = None log.CreatedResource(args.database, kind='database', is_async=args.async_) return result