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

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/__init__.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 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.
+"""Command group for Dataplex Asset Resource."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.calliope import base
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Assets(base.Group):
+  """Manage Dataplex Asset resources."""
+
+  category = base.DATA_ANALYTICS_CATEGORY

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/actions/__init__.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 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.
+"""Command group for Dataplex Asset Resource."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.calliope import base
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class Actions(base.Group):
+  """Manage Dataplex asset resource actions."""
+
+  category = base.DATA_ANALYTICS_CATEGORY

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/actions/list.yaml

@@ -0,0 +1,34 @@
+- release_tracks: [ALPHA, GA]
+  help_text:
+    brief: |
+      List Dataplex asset actions.
+    description: |
+      List all Dataplex Actions under a specific asset.
+    examples: |
+       To list all actions of a Dataplex asset `test-asset` defined in location `us-central1` with
+       lake `test-lake`, zone `test-zone`, run:
+
+         $ {command} --project=test-project --location=us-central1 --lake=test-lake --zone=test-zone --asset=test-asset
+
+  command_type: LIST
+  request:
+    ALPHA:
+      api_version: v1
+    collection: dataplex.projects.locations.lakes.zones.assets.actions
+  arguments:
+    resource:
+      help_text: |
+        Arguments and flags that define the Dataplex asset Actions you want to list.
+      override_resource_collection: true
+      spec: !REF googlecloudsdk.command_lib.dataplex.resources:asset
+
+  response:
+    id_field: name
+  output:
+    format: |
+      table(
+        name.basename():label=NAME,
+        data_locations:label=DATA_LOCATION,
+        category: label=CATEGORY,
+        issue:label=ISSUE
+      )

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/add_iam_policy_binding.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google Inc. 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.
+"""Command to add-iam-policy-binding to a Dataplex asset resource."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.dataplex import asset
+from googlecloudsdk.api_lib.util import exceptions as gcloud_exception
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.dataplex import resource_args
+from googlecloudsdk.command_lib.iam import iam_util
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class AddIamPolicyBinding(base.Command):
+  """Adds IAM policy binding to a Dataplex asset resource."""
+
+  detailed_help = {
+      'EXAMPLES':
+          """\
+          To add an IAM policy binding for the role of `roles/dataplex.viewer`
+          for the user `test-user@gmail.com` to asset `test-asset` within zone
+          `test-zone` in lake `test-lake` in location `us-central`, run:
+
+            $ {command} test-asset --location=us-central1 --lake=test-lake --zone=test-zone --role=roles/dataplex.viewer --member=user:foo@gmail.com
+
+          See https://cloud.google.com/dataplex/docs/iam-roles for details of
+          policy role and member types.
+          """,
+  }
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAssetResourceArg(parser, 'to add IAM policy binding to.')
+
+    iam_util.AddArgsForAddIamPolicyBinding(parser)
+
+  @gcloud_exception.CatchHTTPErrorRaiseHTTPException(
+      'Status code: {status_code}. {status_message}.')
+  def Run(self, args):
+    asset_ref = args.CONCEPTS.asset.Parse()
+
+    result = asset.AddIamPolicyBinding(asset_ref, args.member, args.role)
+    return result

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/create.py

@@ -0,0 +1,228 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google Inc. 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.
+"""Command to create a Dataplex asset resource."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.dataplex import asset
+from googlecloudsdk.api_lib.dataplex import util as dataplex_util
+from googlecloudsdk.api_lib.util import exceptions as gcloud_exception
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.dataplex import flags
+from googlecloudsdk.command_lib.dataplex import resource_args
+from googlecloudsdk.command_lib.util.apis import arg_utils
+from googlecloudsdk.command_lib.util.args import labels_util
+from googlecloudsdk.core import log
+
+
+def _CommonArgs(parser):
+  """Create a common args."""
+  resource_args.AddAssetResourceArg(parser, 'to create.')
+  parser.add_argument(
+      '--validate-only',
+      action='store_true',
+      default=False,
+      help="Validate the create action, but don't actually perform it.",
+  )
+  parser.add_argument('--description', help='Description of the asset')
+  parser.add_argument('--display-name', help='Display name of the asset')
+  flags.AddDiscoveryArgs(parser)
+  base.ASYNC_FLAG.AddToParser(parser)
+  labels_util.AddCreateLabelsFlags(parser)
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Create(base.Command):
+  """Create a Dataplex asset resource.
+
+  An asset represents a cloud resource that is being managed within a lake as a
+  member of a zone.
+
+  This asset ID will be used to generate names such as table names when
+  publishing metadata to Hive Metastore and BigQuery.
+   * Must contain only lowercase letters, numbers, and hyphens.
+   * Must start with a letter.
+   * Must end with a number or a letter.
+   * Must be between 1-63 characters.
+   * Must be unique within the zone.
+  """
+
+  detailed_help = {
+      'EXAMPLES':
+          """\
+          To create a Dataplex asset with name `test-asset`, within zone
+          `test-zone`, in lake `test-lake`, in location `us-central1`, with
+          resource type `STORAGE_BUCKET`, with resource name
+          `projects/test-project/buckets/test-bucket`, run:
+
+            $ {command} test-asset --location=us-central --lake=test-lake --zone=test-zone --resource-type=STORAGE_BUCKET --resource-name=projects/test-project/buckets/test-bucket
+
+          To create a Dataplex asset with name `test-asset`, within zone
+          `test-zone`, in lake `test-lake`, in location `us-central1`, with
+          resource type `STORAGE_BUCKET`, with resource name
+          `projects/test-project/buckets/test-bucket`, with discovery-enabled,
+          and discovery schedule `0 * * * *`, run:
+
+            $ {command} test-asset --location=us-central --lake=test-lake --zone=test-zone --resource-type=STORAGE_BUCKET --resource-name=projects/test-project/buckets/test-bucket --discovery-enabled --discovery-schedule="0 * * * *"
+          """,
+  }
+
+  @staticmethod
+  def Args(parser):
+    _CommonArgs(parser)
+    resource_spec = parser.add_group(
+        required=True,
+        help='Specification of the resource that is referenced by this asset.',
+    )
+    resource_spec.add_argument(
+        '--resource-name',
+        help=""""Relative name of the cloud resource that contains the data that
+                 is being managed within a lake. For example:
+                 `projects/{project_number}/buckets/{bucket_id}` or  `projects/{project_number}/datasets/{dataset_id}`""",
+    )
+    resource_spec.add_argument(
+        '--resource-type',
+        required=True,
+        choices={
+            'STORAGE_BUCKET': 'Cloud Storage Bucket',
+            'BIGQUERY_DATASET': 'BigQuery Dataset',
+        },
+        type=arg_utils.ChoiceToEnumName,
+        help='Type',
+    )
+    resource_spec.add_argument(
+        '--resource-read-access-mode',
+        required=False,
+        choices={
+            'DIRECT': 'Data is accessed directly using storage APIs',
+            'MANAGED': (
+                'Data is accessed through a managed interface using BigQuery'
+                ' APIs.'
+            ),
+        },
+        type=arg_utils.ChoiceToEnumName,
+        help='Read access mode',
+    )
+
+  def GenerateRequest(self, args):
+    return asset.GenerateAssetForCreateRequest(args)
+
+  @gcloud_exception.CatchHTTPErrorRaiseHTTPException(
+      'Status code: {status_code}. {status_message}.')
+  def Run(self, args):
+    asset_ref = args.CONCEPTS.asset.Parse()
+    dataplex_client = dataplex_util.GetClientInstance()
+    create_req_op = dataplex_client.projects_locations_lakes_zones_assets.Create(
+        dataplex_util.GetMessageModule().DataplexProjectsLocationsLakesZonesAssetsCreateRequest(
+            assetId=asset_ref.Name(),
+            parent=asset_ref.Parent().RelativeName(),
+            validateOnly=args.validate_only,
+            googleCloudDataplexV1Asset=self.GenerateRequest(args),
+        )
+    )
+    validate_only = getattr(args, 'validate_only', False)
+    if validate_only:
+      log.status.Print('Validation complete.')
+      return
+
+    async_ = getattr(args, 'async_', False)
+    if not async_:
+      asset.WaitForOperation(create_req_op)
+      log.CreatedResource(
+          asset_ref.Name(),
+          details='Asset created in [{0}]'
+          .format(asset_ref.Parent().RelativeName()))
+      return
+
+    log.status.Print('Creating [{0}] with operation [{1}].'.format(
+        asset_ref, create_req_op.name))
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class CreateAlpha(Create):
+  """Create a Dataplex asset resource.
+
+  An asset represents a cloud resource that is being managed within a lake as a
+  member of a zone.
+
+  This asset ID will be used to generate names such as table names when
+  publishing metadata to Hive Metastore and BigQuery.
+   * Must contain only lowercase letters, numbers, and hyphens.
+   * Must start with a letter.
+   * Must end with a number or a letter.
+   * Must be between 1-63 characters.
+   * Must be unique within the zone.
+  """
+
+  detailed_help = {
+      'EXAMPLES': """\
+          To create a Dataplex asset with name `test-asset`, within zone
+          `test-zone`, in lake `test-lake`, in location `us-central1`, with
+          resource type `STORAGE_BUCKET`, with resource name
+          `projects/test-project/buckets/test-bucket`, run:
+
+            $ {command} test-asset --location=us-central --lake=test-lake --zone=test-zone --resource-type=STORAGE_BUCKET --resource-name=projects/test-project/buckets/test-bucket
+
+          To create a Dataplex asset with name `test-asset`, within zone
+          `test-zone`, in lake `test-lake`, in location `us-central1`, with
+          resource type `STORAGE_BUCKET`, with resource name
+          `projects/test-project/buckets/test-bucket`, with discovery-enabled,
+          and discovery schedule `0 * * * *`, run:
+
+            $ {command} test-asset --location=us-central --lake=test-lake --zone=test-zone --resource-type=STORAGE_BUCKET --resource-name=projects/test-project/buckets/test-bucket --discovery-enabled --discovery-schedule="0 * * * *"
+          """,
+  }
+
+  def GenerateRequest(self, args):
+    return asset.GenerateAssetForCreateRequestAlpha(args)
+
+  @staticmethod
+  def Args(parser):
+    _CommonArgs(parser)
+    resource_spec = parser.add_group(
+        required=True,
+        help='Specification of the resource that is referenced by this asset.',
+    )
+    resource_spec.add_argument(
+        '--resource-name',
+        help=""""Relative name of the cloud resource that contains the data that
+                 is being managed within a lake. For example:
+                 `projects/{project_number}/buckets/{bucket_id}` or  `projects/{project_number}/datasets/{dataset_id}`""",
+    )
+    resource_spec.add_argument(
+        '--resource-type',
+        required=True,
+        choices={
+            'STORAGE_BUCKET': 'Cloud Storage Bucket',
+            'BIGQUERY_DATASET': 'BigQuery Dataset',
+        },
+        type=arg_utils.ChoiceToEnumName,
+        help='Type',
+    )
+    resource_spec.add_argument(
+        '--resource-read-access-mode',
+        required=False,
+        choices={
+            'DIRECT': 'Data is accessed directly using storage APIs',
+            'MANAGED': (
+                'Data is accessed through a managed interface using BigQuery'
+                ' APIs.'
+            ),
+        },
+        type=arg_utils.ChoiceToEnumName,
+        help='Read access mode',
+    )

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/delete.yaml

@@ -0,0 +1,22 @@
+- release_tracks: [ALPHA, GA]
+  help_text:
+    brief: |
+      Delete a Dataplex asset resource.
+    description: |
+      Delete a Dataplex asset resource.
+    examples: |
+      To delete a Dataplex asset `test-asset` within zone `test-zone` in lake `test-lake` in
+      location `us-central1`, run:
+
+          $ {command} test-asset --location=us-central1 --lake=test-lake --zone=test-zone
+  request:
+    ALPHA:
+      api_version: v1
+    collection: dataplex.projects.locations.lakes.zones.assets
+  arguments:
+    resource:
+      help_text: |
+        Arguments and flags that define the Dataplex asset you want to delete.
+      spec: !REF googlecloudsdk.command_lib.dataplex.resources:asset
+  async:
+    collection: dataplex.projects.locations.operations

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/describe.yaml

@@ -0,0 +1,21 @@
+- release_tracks: [ALPHA, GA]
+  help_text:
+    brief: |
+      Describe a Dataplex asset resource.
+    description: |
+      Displays all details of a Dataplex asset resource given a valid asset ID.
+    examples: |
+      To describe a Dataplex asset `test-asset` within zone `test-zone` in lake `test-lake` in
+      location `us-central1`, run:
+
+          $ {command} test-asset --location=us-central1 --lake=test-lake --zone=test-zone
+  request:
+    ALPHA:
+      api_version: v1
+    collection: dataplex.projects.locations.lakes.zones.assets
+    method: get
+  arguments:
+    resource:
+      help_text: |
+        Arguments and flags that define the Dataplex asset you want to retrieve.
+      spec: !REF googlecloudsdk.command_lib.dataplex.resources:asset

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/get_iam_policy.yaml

@@ -0,0 +1,24 @@
+- release_tracks: [ALPHA, GA]
+  help_text:
+    brief: |
+      Get the IAM policy for a Dataplex asset resource.
+    description: |
+      Displays the IAM policy associated with a Dataplex asset resource.
+      If formatted as JSON, the output can be edited and used as
+      a policy file for *set-iam-policy*. The output includes an "etag"
+      field identifying the version emitted and allowing detection of
+      concurrent policy updates.
+    examples: |
+      To print the IAM policy for Dataplex asset `test-asset` within zone `test-zone` in lake
+      `test-lake` in location `us-central1`, run:
+
+          $ {command} test-asset --location=us-central1 --lake=test-lake --zone=test-zone
+  request:
+    ALPHA:
+      api_version: v1
+    collection: dataplex.projects.locations.lakes.zones.assets
+  arguments:
+    resource:
+      help_text: |
+        Arguments and flags that define the Dataplex asset IAM policy you want to retrieve.
+      spec: !REF googlecloudsdk.command_lib.dataplex.resources:asset

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/list.yaml

@@ -0,0 +1,37 @@
+- release_tracks: [ALPHA, GA]
+  help_text:
+    brief: |
+      List Dataplex asset resources.
+    description: |
+      List Dataplex asset resource under a specific project. location, lake, and zone.
+    examples: |
+      To list all Dataplex asset resources within zone `test-zone` in lake `test-lake` in location
+      `us-central`, run:
+
+          $ {command} --location=us-central1 --lake=test-lake --zone=test-zone
+  request:
+    ALPHA:
+      api_version: v1
+    collection: dataplex.projects.locations.lakes.zones.assets
+  arguments:
+    resource:
+      help_text: |
+        Arguments and flags that define the Dataplex asset you want to list.
+      spec: !REF googlecloudsdk.command_lib.dataplex.resources:zone
+
+  response:
+    id_field: name
+  output:
+    format: |
+      table(
+        name.basename():label=NAME,
+        displayName:label=DISPLAY_NAME,
+        state:label=ASSET_STATUS,
+        discoverySpec.enabled:label=DISCOVERY_ENABLED,
+        discoveryStatus.state:label=DISCOVERY_STATUS,
+        resourceSpec.name:label=RESOURCE_NAME,
+        resourceSpec.type:label=RESOURCE_TYPE,
+        resourceStatus.state:label=RESOURCE_STATE,
+        securityStatus.state:label=SECURITY_STATUS,
+        labels:label=LABELS
+      )

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/remove_iam_policy_binding.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google Inc. 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.
+"""Command to remove-iam-policy-binding from a Dataplex asset resource."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.dataplex import asset
+from googlecloudsdk.api_lib.util import exceptions as gcloud_exception
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.dataplex import resource_args
+from googlecloudsdk.command_lib.iam import iam_util
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class RemoveIamPolicyBinding(base.Command):
+  """Remove IAM policy binding from a Dataplex asset resource."""
+
+  detailed_help = {
+      'EXAMPLES':
+          """\
+          To remove an IAM policy binding for the role of `roles/dataplex.viewer`
+          for the user `test-user@gmail.com` from asset `test-asset` in zone
+          `test-zone` in lake `test-lake` in location `us-central`, run:
+
+            $ {command} test-asset --location=us-central1 --lake=test-lake --zone=test-zone --role=roles/dataplex.viewer --member=user:foo@gmail.com
+
+          See https://cloud.google.com/dataplex/docs/iam-roles for details of
+          policy role and member types.
+          """,
+  }
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAssetResourceArg(parser,
+                                      'to remove IAM policy binding from.')
+
+    iam_util.AddArgsForRemoveIamPolicyBinding(parser)
+
+  @gcloud_exception.CatchHTTPErrorRaiseHTTPException(
+      'Status code: {status_code}. {status_message}.')
+  def Run(self, args):
+    asset_ref = args.CONCEPTS.asset.Parse()
+
+    result = asset.RemoveIamPolicyBinding(asset_ref, args.member, args.role)
+    return result

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/set_iam_policy.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google Inc. 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.
+"""Command to set-iam-policy for a Dataplex asset resource."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.dataplex import asset
+from googlecloudsdk.api_lib.util import exceptions as gcloud_exception
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.dataplex import resource_args
+from googlecloudsdk.command_lib.iam import iam_util
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA, base.ReleaseTrack.GA)
+class SetIamPolicy(base.Command):
+  """Set the IAM policy to a Dataplex asset as defined in a JSON or YAML file.
+
+    See https://cloud.google.com/iam/docs/managing-policies for details of
+    the policy file format and contents.
+
+  """
+
+  detailed_help = {
+      'EXAMPLES':
+          """\
+          The following command will read an IAM policy defined in a JSON file
+          `policy.son` and set it for the Dataplex asset `test-asset` in zone
+          `test-zone` in lake `test-lake` in location `us-central1`:
+
+            $ {command} test-asset --location=us-central1 --lake=test-lake --zone=test-zone policy.json
+
+          """,
+  }
+
+  @staticmethod
+  def Args(parser):
+    resource_args.AddAssetResourceArg(parser, 'to set IAM policy binding to.')
+    iam_util.AddArgForPolicyFile(parser)
+
+  @gcloud_exception.CatchHTTPErrorRaiseHTTPException(
+      'Status code: {status_code}. {status_message}.')
+  def Run(self, args):
+    asset_ref = args.CONCEPTS.asset.Parse()
+    result = asset.SetIamPolicyFromFile(asset_ref, args.policy_file)
+    return result

gcloud auth application-default login/google-cloud-sdk/lib/surface/dataplex/assets/update.py

@@ -0,0 +1,169 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2021 Google Inc. 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.
+"""Command to update a Dataplex asset resource."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.dataplex import asset
+from googlecloudsdk.api_lib.dataplex import util as dataplex_util
+from googlecloudsdk.api_lib.util import exceptions as gcloud_exception
+from googlecloudsdk.calliope import base
+from googlecloudsdk.calliope import exceptions
+from googlecloudsdk.command_lib.dataplex import flags
+from googlecloudsdk.command_lib.dataplex import resource_args
+from googlecloudsdk.command_lib.util.apis import arg_utils
+from googlecloudsdk.command_lib.util.args import labels_util
+from googlecloudsdk.core import log
+
+
+def _CommonArgs(parser):
+  """Create a common args."""
+  resource_args.AddAssetResourceArg(parser, 'to update.')
+  parser.add_argument(
+      '--validate-only',
+      action='store_true',
+      default=False,
+      help="Validate the update action, but don't actually perform it.",
+  )
+  parser.add_argument('--description', help='Description of the asset')
+  parser.add_argument('--display-name', help='Display Name')
+  flags.AddDiscoveryArgs(parser)
+  base.ASYNC_FLAG.AddToParser(parser)
+  labels_util.AddCreateLabelsFlags(parser)
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Update(base.Command):
+  """Update a Dataplex asset resource."""
+
+  detailed_help = {
+      'EXAMPLES':
+          """\
+          To update a Dataplex asset `test-asset` in zone `test-zone` in lake
+          `test-lake` in location `us-central1` to have the display name
+          `first-dataplex-asset` and discovery include patterns `abc`, `def`,
+          run:
+
+            $ {command} test-asset --location=us-central1 --lake=test-lake --zone=test-zone --display-name="first-dataplex-asset" --discovery-include-patterns=abc,def
+          """,
+  }
+
+  @staticmethod
+  def Args(parser):
+    _CommonArgs(parser)
+    resource_spec = parser.add_group(
+        required=False,
+        help='Specification of the resource that is referenced by this asset.',
+    )
+    resource_spec.add_argument(
+        '--resource-read-access-mode',
+        required=False,
+        choices={
+            'DIRECT': 'Data is accessed directly using storage APIs',
+            'MANAGED': (
+                'Data is accessed through a managed interface using BigQuery'
+                ' APIs.'
+            ),
+        },
+        type=arg_utils.ChoiceToEnumName,
+        help='Read access mode',
+    )
+
+  def GenerateUpdateMask(self, args):
+    return asset.GenerateUpdateMask(args)
+
+  def GenerateUpdateRequest(self, args):
+    return asset.GenerateAssetForUpdateRequest(args)
+
+  @gcloud_exception.CatchHTTPErrorRaiseHTTPException(
+      'Status code: {status_code}. {status_message}.'
+  )
+  def Run(self, args):
+    update_mask = self.GenerateUpdateMask(args)
+    if len(update_mask) < 1:
+      raise exceptions.HttpException(
+          'Update commands must specify at least one additional parameter to'
+          ' change.'
+      )
+
+    asset_ref = args.CONCEPTS.asset.Parse()
+    dataplex_client = dataplex_util.GetClientInstance()
+    message = dataplex_util.GetMessageModule()
+    update_req_op = dataplex_client.projects_locations_lakes_zones_assets.Patch(
+        message.DataplexProjectsLocationsLakesZonesAssetsPatchRequest(
+            name=asset_ref.RelativeName(),
+            validateOnly=args.validate_only,
+            updateMask=','.join(update_mask),
+            googleCloudDataplexV1Asset=self.GenerateUpdateRequest(args),
+        )
+    )
+    validate_only = getattr(args, 'validate_only', False)
+    if validate_only:
+      log.status.Print('Validation complete.')
+      return
+
+    async_ = getattr(args, 'async_', False)
+    if not async_:
+      asset.WaitForOperation(update_req_op)
+      log.UpdatedResource(asset_ref, details='Operation was sucessful.')
+      return
+
+    log.status.Print('Updating [{0}] with operation [{1}].'.format(
+        asset_ref, update_req_op.name))
+
+
+@base.ReleaseTracks(base.ReleaseTrack.ALPHA)
+class UpdateAlpha(Update):
+  """Update a Dataplex asset resource."""
+
+  detailed_help = {
+      'EXAMPLES': """\
+          To update a Dataplex asset `test-asset` in zone `test-zone` in lake
+          `test-lake` in location `us-central1` to have the display name
+          `first-dataplex-asset` and discovery include patterns `abc`, `def`,
+          run:
+
+            $ {command} test-asset --location=us-central1 --lake=test-lake --zone=test-zone --display-name="first-dataplex-asset" --discovery-include-patterns=abc,def
+          """,
+  }
+
+  @staticmethod
+  def Args(parser):
+    _CommonArgs(parser)
+    resource_spec = parser.add_group(
+        required=False,
+        help='Specification of the resource that is referenced by this asset.',
+    )
+    resource_spec.add_argument(
+        '--resource-read-access-mode',
+        required=False,
+        choices={
+            'DIRECT': 'Data is accessed directly using storage APIs',
+            'MANAGED': (
+                'Data is accessed through a managed interface using BigQuery'
+                ' APIs.'
+            ),
+        },
+        type=arg_utils.ChoiceToEnumName,
+        help='Read access mode',
+    )
+
+  def GenerateUpdateMask(self, args):
+    return asset.GenerateUpdateMaskAlpha(args)
+
+  def GenerateUpdateRequest(self, args):
+    return asset.GenerateAssetForUpdateRequestAlpha(args)