feat: Add new gcloud commands, API clients, and third-party libraries across various services.

gcloud auth application-default login/google-cloud-sdk/lib/surface/sql/operations/__init__.py

@@ -0,0 +1,35 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2013 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.
+
+"""Provide commands for working with Cloud SQL instance operations."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.calliope import base
+from googlecloudsdk.calliope import exceptions
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA, base.ReleaseTrack.BETA,
+                    base.ReleaseTrack.ALPHA)
+class Operations(base.Group):
+  """Provide commands for working with Cloud SQL instance operations.
+
+  Provide commands for working with Cloud SQL instance operations, including
+  listing and getting information about instance operations of a Cloud SQL
+  instance.
+  """
+  category = base.DATABASES_CATEGORY

gcloud auth application-default login/google-cloud-sdk/lib/surface/sql/operations/cancel.py

@@ -0,0 +1,84 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2023 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.
+"""Cancels a Cloud SQL instance operation."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.sql import api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.core import log
+from googlecloudsdk.core import properties
+from googlecloudsdk.core.console import console_io
+
+DETAILED_HELP = {
+    'EXAMPLES':
+        """\
+        To cancel an operation with the id "prod-operation-id",
+        like "acb40108-a483-4a8b-8a5c-e27100000032", run:
+
+          $ {command} prod-operation-id
+        """,
+}
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA, base.ReleaseTrack.BETA,
+                    base.ReleaseTrack.ALPHA)
+class Cancel(base.UpdateCommand):
+  """Cancels a Cloud SQL instance operation."""
+
+  detailed_help = DETAILED_HELP
+
+  @staticmethod
+  def Args(parser):
+    """Args is called by calliope to gather arguments for this command.
+
+    Args:
+      parser: An argparse parser that you can use it to add arguments that go on
+        the command line after this command. Positional arguments are allowed.
+    """
+    parser.add_argument(
+        'operation', help='Name that uniquely identifies the operation.')
+
+  def Run(self, args):
+    """Cancels a Cloud SQL instance operation.
+
+    Args:
+      args: argparse.Namespace, The arguments that this command was invoked
+        with.
+
+    Returns:
+      An empty response.
+    """
+    client = api_util.SqlClient(api_util.API_VERSION_DEFAULT)
+    sql_client = client.sql_client
+    sql_messages = client.sql_messages
+
+    operation_ref = client.resource_parser.Parse(
+        args.operation,
+        collection='sql.operations',
+        params={'project': properties.VALUES.core.project.GetOrFail})
+
+    log.warning('Warning: You are about to cancel [{operation}].'.format(
+        operation=operation_ref.operation))
+    console_io.PromptContinue(cancel_on_no=True)
+
+    empty = sql_client.operations.Cancel(
+        sql_messages.SqlOperationsCancelRequest(
+            project=operation_ref.project, operation=operation_ref.operation))
+    log.status.write('Cancellation issued on [{operation}].\n'.format(
+        operation=operation_ref.operation))
+    return empty

gcloud auth application-default login/google-cloud-sdk/lib/surface/sql/operations/describe.py

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2013 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.
+"""Retrieves information about a Cloud SQL instance operation."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.sql import api_util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.core import properties
+
+DETAILED_HELP = {
+    'EXAMPLES':
+        """\
+        To describe an operation with the name "prod-operation-id", run:
+
+          $ {command} prod-operation-id
+        """,
+}
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA, base.ReleaseTrack.BETA,
+                    base.ReleaseTrack.ALPHA)
+class Get(base.DescribeCommand):
+  """Retrieves information about a Cloud SQL instance operation."""
+
+  detailed_help = DETAILED_HELP
+
+  @staticmethod
+  def Args(parser):
+    """Args is called by calliope to gather arguments for this command.
+
+    Args:
+      parser: An argparse parser that you can use it to add arguments that go
+          on the command line after this command. Positional arguments are
+          allowed.
+    """
+    parser.add_argument(
+        'operation', help='Name that uniquely identifies the operation.')
+
+  def Run(self, args):
+    """Retrieves information about a Cloud SQL instance operation.
+
+    Args:
+      args: argparse.Namespace, The arguments that this command was invoked
+          with.
+
+    Returns:
+      A dict object representing the operations resource if the api request was
+      successful.
+    """
+    client = api_util.SqlClient(api_util.API_VERSION_DEFAULT)
+    sql_client = client.sql_client
+    sql_messages = client.sql_messages
+
+    operation_ref = client.resource_parser.Parse(
+        args.operation,
+        collection='sql.operations',
+        params={'project': properties.VALUES.core.project.GetOrFail})
+
+    return sql_client.operations.Get(
+        sql_messages.SqlOperationsGetRequest(
+            project=operation_ref.project, operation=operation_ref.operation))

gcloud auth application-default login/google-cloud-sdk/lib/surface/sql/operations/list.py

@@ -0,0 +1,90 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2013 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.
+"""List all instance operations.
+
+Lists all instance operations that have been performed on the given
+Cloud SQL instance.
+"""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from apitools.base.py import list_pager
+
+from googlecloudsdk.api_lib.sql import api_util
+from googlecloudsdk.api_lib.sql import validate
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.sql import flags
+from googlecloudsdk.core import properties
+
+DETAILED_HELP = {
+    'EXAMPLES':
+        """\
+        To list operations for instances with ID "prod-instance" , run:
+
+          $ {command} --instance=prod-instance
+
+        To list operations for instances with ID "prod-instance" that have 10 results, run:
+
+          $ {command} --instance=prod-instance --limit=10
+
+        To list operations for instances with ID "prod-instance" that have 10 results in a page, run:
+
+          $ {command} --instance=prod-instance --page-size=10
+        """,
+}
+
+
+@base.DefaultUniverseOnly
+@base.ReleaseTracks(base.ReleaseTrack.GA, base.ReleaseTrack.BETA,
+                    base.ReleaseTrack.ALPHA)
+class List(base.ListCommand):
+  """Lists all instance operations for the given Cloud SQL instance."""
+
+  detailed_help = DETAILED_HELP
+
+  @staticmethod
+  def Args(parser):
+    flags.AddInstance(parser)
+    parser.display_info.AddFormat(flags.OPERATION_FORMAT_BETA)
+    parser.display_info.AddCacheUpdater(None)
+
+  def Run(self, args):
+    """Lists all instance operations that have been performed on an instance.
+
+    Args:
+      args: argparse.Namespace, The arguments that this command was invoked
+          with.
+
+    Returns:
+      A dict object that has the list of operation resources if the command ran
+      successfully.
+    """
+    client = api_util.SqlClient(api_util.API_VERSION_DEFAULT)
+    sql_client = client.sql_client
+    sql_messages = client.sql_messages
+
+    instance_ref = client.resource_parser.Parse(
+        args.instance,
+        params={'project': properties.VALUES.core.project.GetOrFail},
+        collection='sql.instances')
+    validate.ValidateInstanceName(args.instance)
+
+    return list_pager.YieldFromList(
+        sql_client.operations,
+        sql_messages.SqlOperationsListRequest(
+            project=instance_ref.project, instance=instance_ref.instance),
+        limit=args.limit)

gcloud auth application-default login/google-cloud-sdk/lib/surface/sql/operations/wait.py

@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2014 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.
+"""Retrieves information about a Cloud SQL instance operation."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.sql import api_util
+from googlecloudsdk.api_lib.sql import operations
+from googlecloudsdk.calliope import arg_parsers
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.sql import flags
+from googlecloudsdk.core import properties
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA, base.ReleaseTrack.BETA,
+                    base.ReleaseTrack.ALPHA)
+class Wait(base.Command):
+  """Waits for one or more operations to complete."""
+
+  @staticmethod
+  def Args(parser):
+    flags.AddOperationArgument(parser)
+    parser.add_argument(
+        '--timeout',
+        type=arg_parsers.BoundedInt(lower_bound=0, unlimited=True),
+        default=300,
+        help=('Maximum number of seconds to wait for an operation to complete. '
+              'By default, wait for 300s. Set to *unlimited* to wait '
+              'indefinitely.'))
+    parser.display_info.AddFormat(flags.OPERATION_FORMAT_BETA)
+
+  def Run(self, args):
+    """Wait for a Cloud SQL instance operation.
+
+    Args:
+      args: argparse.Namespace, The arguments that this command was invoked
+        with.
+
+    Yields:
+      Operations that were waited for.
+    """
+    client = api_util.SqlClient(api_util.API_VERSION_DEFAULT)
+    sql_client = client.sql_client
+
+    for op in args.operation:
+      operation_ref = client.resource_parser.Parse(
+          op,
+          collection='sql.operations',
+          params={'project': properties.VALUES.core.project.GetOrFail})
+
+      yield operations.OperationsV1Beta4.WaitForOperation(
+          sql_client,
+          operation_ref,
+          'Waiting for [{operation}]'.format(operation=operation_ref),
+          max_wait_seconds=args.timeout)