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

gcloud auth application-default login/google-cloud-sdk/lib/surface/datastream/connection_profiles/__init__.py

@@ -0,0 +1,30 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2020 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.
+
+"""Connection profiles command group for Datastream."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.calliope import base
+
+
+@base.ReleaseTracks(base.ReleaseTrack.BETA, base.ReleaseTrack.GA)
+class ConnectionProfiles(base.Group):
+  """Manage Datastream connection profiles.
+
+  Commands for managing Datastream connection profiles.
+  """

gcloud auth application-default login/google-cloud-sdk/lib/surface/datastream/connection_profiles/create.py

@@ -0,0 +1,201 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2020 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 to create connection profiles for a datastream."""
+
+
+from googlecloudsdk.api_lib.datastream import connection_profiles
+from googlecloudsdk.api_lib.datastream import util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.datastream import flags
+from googlecloudsdk.command_lib.datastream import resource_args
+from googlecloudsdk.command_lib.datastream.connection_profiles import flags as cp_flags
+from googlecloudsdk.core.console import console_io
+
+DESCRIPTION = ('Create a Datastream connection profile')
+EXAMPLES = """\
+    To create a connection profile for Oracle:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=oracle --oracle-password=fakepassword --oracle-username=fakeuser --display-name=my-profile --oracle-hostname=35.188.150.50 --oracle-port=1521 --database-service=ORCL --static-ip-connectivity
+
+    To create a connection profile for MySQL:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=mysql --mysql-password=fakepassword --mysql-username=fakeuser --display-name=my-profile --mysql-hostname=35.188.150.50 --mysql-port=3306 --static-ip-connectivity
+
+    To create a connection profile for PostgreSQL:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=postgresql --postgresql-password=fakepassword --postgresql-username=fakeuser --display-name=my-profile --postgresql-hostname=35.188.150.50 --postgresql-port=5432 --postgresql-database=db --static-ip-connectivity
+
+    To create a connection profile for SQL Server:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=sqlserver --sqlserver-password=fakepassword --sqlserver-username=fakeuser --display-name=my-profile --sqlserver-hostname=35.188.150.50 --sqlserver-port=1433 --sqlserver-database=db --static-ip-connectivity
+
+    To create a connection profile for Salesforce using Username, Password and Security Token:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=salesforce --salesforce-password=fakepassword --salesforce-username=fakeuser --salesforce-security-token=fakesecuritytoken --display-name=my-profile --salesforce-hostname=35.188.150.50 --salesforce-port=1433 --salesforce-database=db --static-ip-connectivity
+
+    To create a connection profile for Salesforce using OAuth:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=salesforce --salesforce-client-secret=fakesecret --salesforce-client-id=fake-client-id --display-name=my-profile --salesforce-domain=fakecompany.my.salesforce.com --static-ip-connectivity
+
+    To create a connection profile for Google Cloud Storage:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=google-cloud-storage --bucket=fake-bucket --root-path=/root/path --display-name=my-profile
+
+    To create a connection profile for BigQuery:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=bigquery --display-name=my-profile
+
+    To create a connection profile for MongoDB:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=mongodb --mongodb-password=fakepassword --mongodb-username=fakeuser --display-name=my-profile --mongodb-host-addresses=35.188.150.50:27017
+
+   """
+EXAMPLES_BETA = """\
+    To create a connection profile for Oracle:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=oracle --oracle-password=fakepassword --oracle-username=fakeuser --display-name=my-profile --oracle-hostname=35.188.150.50 --oracle-port=1521 --database-service=ORCL --static-ip-connectivity
+
+    To create a connection profile for MySQL:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=mysql --mysql-password=fakepassword --mysql-username=fakeuser --display-name=my-profile --mysql-hostname=35.188.150.50 --mysql-port=3306 --static-ip-connectivity
+
+    To create a connection profile for Google Cloud Storage:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=google-cloud-storage --bucket-name=fake-bucket --root-path=/root/path --display-name=my-profile --no-connectivity
+   """
+
+
+@base.DefaultUniverseOnly
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Create(base.Command):
+  """Create a Datastream connection profile."""
+  detailed_help = {'DESCRIPTION': DESCRIPTION, 'EXAMPLES': EXAMPLES}
+
+  @staticmethod
+  def CommonArgs(parser, release_track):
+    """Common arguments for all release tracks.
+
+    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.
+      release_track: Some arguments are added based on the command release
+        track.
+    """
+    resource_args.AddConnectionProfileResourceArg(parser, 'to create',
+                                                  release_track)
+
+    cp_flags.AddTypeFlag(parser)
+    cp_flags.AddDisplayNameFlag(parser)
+    if release_track == base.ReleaseTrack.GA:
+      cp_flags.AddValidationGroup(parser, 'Create')
+
+    profile_flags = parser.add_group(mutex=True)
+    cp_flags.AddMysqlProfileGroup(profile_flags)
+    cp_flags.AddOracleProfileGroup(profile_flags)
+    cp_flags.AddPostgresqlProfileGroup(profile_flags)
+    cp_flags.AddSqlServerProfileGroup(profile_flags)
+    cp_flags.AddSalesforceProfileGroup(profile_flags)
+    cp_flags.AddGcsProfileGroup(profile_flags, release_track)
+    cp_flags.AddMongodbProfileGroup(profile_flags)
+    flags.AddLabelsCreateFlags(parser)
+
+  @staticmethod
+  def Args(parser):
+    """Args is called by calliope to gather arguments for this command."""
+    Create.CommonArgs(parser, base.ReleaseTrack.GA)
+
+  def Run(self, args):
+    """Create a Datastream connection profile.
+
+    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.
+    """
+    connection_profile_ref = args.CONCEPTS.connection_profile.Parse()
+    parent_ref = connection_profile_ref.Parent().RelativeName()
+
+    if args.oracle_prompt_for_password:
+      args.oracle_password = console_io.PromptPassword(
+          'Please Enter Password: ')
+
+    if args.mysql_prompt_for_password:
+      args.mysql_password = console_io.PromptPassword('Please Enter Password: ')
+
+    if args.postgresql_prompt_for_password:
+      args.postgresql_password = console_io.PromptPassword(
+          'Please Enter Password: ')
+
+    if args.sqlserver_prompt_for_password:
+      args.sqlserver_password = console_io.PromptPassword(
+          'Please Enter Password: '
+      )
+
+    if args.salesforce_prompt_for_password:
+      args.salesforce_password = console_io.PromptPassword(
+          'Please Enter Password: '
+      )
+
+    if args.salesforce_prompt_for_security_token:
+      args.salesforce_security_token = console_io.PromptPassword(
+          'Please Enter Security Token: '
+      )
+
+    if args.salesforce_prompt_for_oauth2_client_secret:
+      args.salesforce_oauth2_client_secret = console_io.PromptPassword(
+          'Please Enter OAuth 2.0 Client Secret: '
+      )
+    if args.mongodb_prompt_for_password:
+      args.mongodb_password = console_io.PromptPassword(
+          'Please Enter Password: '
+      )
+
+    cp_type = (args.type).upper()
+    cp_client = connection_profiles.ConnectionProfilesClient()
+    result_operation = cp_client.Create(
+        parent_ref, connection_profile_ref.connectionProfilesId, cp_type,
+        self.ReleaseTrack(), args)
+
+    client = util.GetClientInstance()
+    messages = util.GetMessagesModule()
+    resource_parser = util.GetResourceParser()
+
+    operation_ref = resource_parser.Create(
+        'datastream.projects.locations.operations',
+        operationsId=result_operation.name,
+        projectsId=connection_profile_ref.projectsId,
+        locationsId=connection_profile_ref.locationsId)
+
+    return client.projects_locations_operations.Get(
+        messages.DatastreamProjectsLocationsOperationsGetRequest(
+            name=operation_ref.operationsId))
+
+
+@base.Deprecate(
+    is_removed=False,
+    warning=('Datastream beta version is deprecated. Please use`gcloud '
+             'datastream connection-profiles create` command instead.'))
+@base.ReleaseTracks(base.ReleaseTrack.BETA)
+class CreateBeta(Create):
+  """Create a Datastream connection profile."""
+  detailed_help = {'DESCRIPTION': DESCRIPTION, 'EXAMPLES': EXAMPLES_BETA}
+
+  @staticmethod
+  def Args(parser):
+    """Args is called by calliope to gather arguments for this command."""
+    Create.CommonArgs(parser, base.ReleaseTrack.BETA)

gcloud auth application-default login/google-cloud-sdk/lib/surface/datastream/connection_profiles/delete.yaml

@@ -0,0 +1,26 @@
+- release_tracks: [BETA, GA]
+  BETA:
+    deprecate:
+      is_removed: false
+      warning: |
+        This command has been deprecated.
+        Use `gcloud datastream connection_profiles delete` instead.
+  help_text:
+    brief: |
+      Delete a Datastream connection profile.
+    description: |
+      Deletes a connection profile.
+    examples: |
+      To delete a connection profile:
+
+          $ {command} CONNECTION_PROFILE --location=us-central1
+
+  request:
+    collection: datastream.projects.locations.connectionProfiles
+    api_version: v1
+
+  arguments:
+    resource:
+      help_text: |
+        Connection profile resource - Connection profile to delete.
+      spec: !REF googlecloudsdk.command_lib.datastream.resources:connection_profile

gcloud auth application-default login/google-cloud-sdk/lib/surface/datastream/connection_profiles/describe.yaml

@@ -0,0 +1,23 @@
+- release_tracks: [BETA, GA]
+  BETA:
+    deprecate:
+      is_removed: false
+      warning: |
+        This command has been deprecated.
+        Use `gcloud datastream connection_profiles describe` instead.
+  help_text:
+    brief: Show details about a Datastream connection profile.
+    description: Show details about a connection profile.
+    examples: |
+      To show details about a connection profile, run:
+
+        $ {command} my-connection-profile --location=us-central1
+
+  request:
+    collection: datastream.projects.locations.connectionProfiles
+    api_version: v1
+
+  arguments:
+    resource:
+      help_text: The connection profile you want to get the details of.
+      spec: !REF googlecloudsdk.command_lib.datastream.resources:connection_profile

gcloud auth application-default login/google-cloud-sdk/lib/surface/datastream/connection_profiles/discover.py

@@ -0,0 +1,97 @@
+# -*- coding: utf-8 -*- #
+# Copyright 2020 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 to discover connection profiles for a datastream."""
+
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import unicode_literals
+
+from googlecloudsdk.api_lib.datastream import connection_profiles
+from googlecloudsdk.api_lib.datastream import util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.datastream import resource_args
+from googlecloudsdk.command_lib.datastream.connection_profiles import flags as cp_flags
+from googlecloudsdk.command_lib.util.concepts import concept_parsers
+from googlecloudsdk.core import properties
+
+DESCRIPTION = (
+    'Discover data objects accessible from a Datastream connection profile')
+EXAMPLES = """\
+    To discover an existing connection profile:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --connection-profile-name=some-cp --recursive=true
+
+    To discover a non-existing connection profile:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --connection-profile-object-file=path/to/yaml/or/json/file
+
+   """
+
+
+class _Discover:
+  """Base class for discovering Datastream connection profiles."""
+
+  detailed_help = {'DESCRIPTION': DESCRIPTION, 'EXAMPLES': EXAMPLES}
+
+  @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.
+    """
+    concept_parsers.ConceptParser.ForResource(
+        '--location',
+        resource_args.GetLocationResourceSpec(),
+        group_help='The location you want to list the connection profiles for.',
+        required=True).AddToParser(parser)
+    resource_args.AddConnectionProfileDiscoverResourceArg(parser)
+    cp_flags.AddDepthGroup(parser)
+    cp_flags.AddRdbmsGroup(parser)
+    cp_flags.AddHierarchyGroup(parser)
+
+  def Run(self, args):
+    """Discover a Datastream connection profile.
+
+    Args:
+      args: argparse.Namespace, The arguments that this command was invoked
+        with.
+
+    Returns:
+      A dict object representing the operations resource describing the discover
+      operation if the discover was successful.
+    """
+    project = properties.VALUES.core.project.Get(required=True)
+    location = args.location
+    parent_ref = util.ParentRef(project, location)
+
+    cp_client = connection_profiles.ConnectionProfilesClient()
+    return cp_client.Discover(parent_ref, self.ReleaseTrack(), args)
+
+
+@base.Deprecate(
+    is_removed=False,
+    warning=('Datastream beta version is deprecated. Please use`gcloud '
+             'datastream connection-profiles discover` command instead.')
+)
+@base.ReleaseTracks(base.ReleaseTrack.BETA)
+class DiscoverBeta(_Discover, base.Command):
+  """Discover a Datastream connection profile."""
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Discover(_Discover, base.Command):
+  """Discover a Datastream connection profile."""

gcloud auth application-default login/google-cloud-sdk/lib/surface/datastream/connection_profiles/list.py

@@ -0,0 +1,167 @@
+# -*- 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.
+"""Implementation of connection profile list command."""
+
+
+from googlecloudsdk.api_lib.datastream import connection_profiles
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.datastream import resource_args
+from googlecloudsdk.command_lib.util.concepts import concept_parsers
+from googlecloudsdk.core import properties
+
+
+def _GetUri(connection_profile_info):
+  """Gets the resource URI for a connection profile."""
+
+  return connection_profiles.ConnectionProfilesClient().GetUri(
+      connection_profile_info.name)
+
+
+class _ConnectionProfileInfo:
+  """Container for connection profile data using in list display."""
+
+  def __init__(self, message, db_type):
+    self.display_name = message.displayName
+    self.name = message.name
+    self.type = db_type
+    self.create_time = message.createTime
+
+
+class _List:
+  """Base class for listing Datastream connection profiles."""
+
+  @classmethod
+  def Args(cls, parser):
+    """Register flags for this command."""
+
+    concept_parsers.ConceptParser.ForResource(
+        "--location",
+        resource_args.GetLocationResourceSpec(),
+        group_help="The location you want to list the connection profiles for.",
+        required=True).AddToParser(parser)
+
+    parser.display_info.AddFormat("""
+          table(
+            display_name,
+            name:label=ID,
+            type,
+            create_time.date():label=CREATED
+          )
+        """)
+
+  def Run(self, args):
+    """Runs the command.
+
+    Args:
+      args: All the arguments that were provided to this command invocation.
+
+    Returns:
+      An iterator over objects containing connection profile data.
+    """
+    cp_client = connection_profiles.ConnectionProfilesClient()
+    project_id = properties.VALUES.core.project.Get(required=True)
+    profiles = cp_client.List(project_id, args)
+
+    return [
+        _ConnectionProfileInfo(profile, self._GetType(profile))
+        for profile in profiles
+    ]
+
+  def _GetType(self, profile):
+    """Gets DB type of a connection profile.
+
+    Args:
+      profile: A connection configuration type of a connection profile.
+
+    Returns:
+      A String representation of the provided profile DB type.
+      Default is None.
+    """
+    raise NotImplementedError
+
+
+@base.Deprecate(
+    is_removed=False,
+    warning=("Datastream beta version is deprecated. Please use`gcloud "
+             "datastream connection-profiles list` command instead.")
+)
+@base.ReleaseTracks(base.ReleaseTrack.BETA)
+class ListBeta(_List, base.ListCommand):
+  """List Datastream connection profiles.
+
+  List connection profiles.
+
+  ## API REFERENCE
+    This command uses the datastream/v1 API. The full documentation
+    for this API can be found at: https://cloud.google.com/datastream/
+
+  ## EXAMPLES
+    To list all connection profiles in a project and location 'us-central1',
+    run:
+
+        $ {command} --location=us-central1
+
+  """
+
+  def _GetType(self, profile):
+    if profile.mysqlProfile:
+      return "MySQL"
+    elif profile.oracleProfile:
+      return "Oracle"
+    elif profile.gcsProfile:
+      return "Google Cloud Storage"
+    elif profile.sqlServerProfile:
+      return "SQL Server"
+    else:
+      return None
+
+
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class List(_List, base.ListCommand):
+  """List Datastream connection profiles.
+
+  List connection profiles.
+
+  ## API REFERENCE
+    This command uses the datastream/v1 API. The full documentation
+    for this API can be found at: https://cloud.google.com/datastream/
+
+  ## EXAMPLES
+    To list all connection profiles in a project and location 'us-central1',
+    run:
+
+        $ {command} --location=us-central1
+
+  """
+
+  def _GetType(self, profile):
+    if profile.mysqlProfile:
+      return "MySQL"
+    elif profile.oracleProfile:
+      return "Oracle"
+    elif profile.postgresqlProfile:
+      return "PostgreSQL"
+    elif profile.gcsProfile:
+      return "Google Cloud Storage"
+    elif profile.sqlServerProfile:
+      return "SQL Server"
+    elif profile.salesforceProfile:
+      return "Salesforce"
+    elif profile.bigqueryProfile:
+      return "BigQuery"
+    elif profile.mongodbProfile:
+      return "MongoDB"
+    else:
+      return None

gcloud auth application-default login/google-cloud-sdk/lib/surface/datastream/connection_profiles/update.py

@@ -0,0 +1,197 @@
+# -*- 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 to update connection profiles for datastream."""
+
+
+from googlecloudsdk.api_lib.datastream import connection_profiles
+from googlecloudsdk.api_lib.datastream import util
+from googlecloudsdk.calliope import base
+from googlecloudsdk.command_lib.datastream import flags
+from googlecloudsdk.command_lib.datastream import resource_args
+from googlecloudsdk.command_lib.datastream.connection_profiles import flags as cp_flags
+from googlecloudsdk.core.console import console_io
+
+DESCRIPTION = ('Updates a Datastream connection profile')
+EXAMPLES = """\
+    To update a connection profile for Oracle:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=oracle --oracle-password=fakepassword --oracle-username=fakeuser --display-name=my-profile --oracle-hostname=35.188.150.50 --oracle-port=1521 --database-service=ORCL --static-ip-connectivity
+
+    To update a connection profile for MySQL:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=mysql --mysql-password=fakepassword --mysql-username=fakeuser --display-name=my-profile --mysql-hostname=35.188.150.50 --mysql-port=3306 --static-ip-connectivity
+
+    To update a connection profile for PostgreSQL:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=postgresql --postgresql-password=fakepassword --postgresql-username=fakeuser --display-name=my-profile --postgresql-hostname=35.188.150.50 --postgresql-port=5432 --postgresql-database=db --static-ip-connectivity
+
+    To update a connection profile for SQL Server:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=sqlserver --sqlserver-password=fakepassword --sqlserver-username=fakeuser --display-name=my-profile --sqlserver-hostname=35.188.150.50 --sqlserver-port=1433 --sqlserver-database=db --static-ip-connectivity
+
+    To update a connection profile for Salesforce:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=salesforce --salesforce-password=fakepassword --salesforce-username=fakeuser --display-name=my-profile --salesforce-domain=fakecompany.my.salesforce.com --static-ip-connectivity
+
+    To update a connection profile for Google Cloud Storage:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=google-cloud-storage --bucket=fake-bucket --root-path=/root/path --display-name=my-profile
+
+    To update a connection profile for BigQuery:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=bigquery --display-name=my-profile
+
+    To update a connection profile for MongoDB:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=mongodb --mongodb-password=fakepassword --mongodb-username=fakeuser --display-name=my-profile --mongodb-host-addresses=35.188.150.50:27017 --static-ip-connectivity
+
+   """
+
+EXAMPLES_BETA = """\
+    To update a connection profile for Oracle:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=oracle --oracle-password=fakepassword --oracle-username=fakeuser --display-name=my-profile --oracle-hostname=35.188.150.50 --oracle-port=1521 --database-service=ORCL --static-ip-connectivity
+
+    To update a connection profile for MySQL:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=mysql --mysql-password=fakepassword --mysql-username=fakeuser --display-name=my-profile --mysql-hostname=35.188.150.50 --mysql-port=3306 --static-ip-connectivity
+
+    To update a connection profile for Google Cloud Storage:
+
+        $ {command} CONNECTION_PROFILE --location=us-central1 --type=google-cloud-storage --bucket-name=fake-bucket --root-path=/root/path --display-name=my-profile --no-connectivity
+   """
+
+
+@base.DefaultUniverseOnly
+@base.ReleaseTracks(base.ReleaseTrack.GA)
+class Update(base.Command):
+  """Update a Datastream connection profile."""
+  detailed_help = {'DESCRIPTION': DESCRIPTION, 'EXAMPLES': EXAMPLES}
+
+  @staticmethod
+  def CommonArgs(parser, release_track):
+    """Common arguments for all release tracks.
+
+    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.
+      release_track: Some arguments are added based on the command release
+        track.
+    """
+    resource_args.AddConnectionProfileResourceArg(
+        parser, 'to update', release_track, required=False)
+
+    cp_flags.AddTypeFlag(parser)
+    cp_flags.AddDisplayNameFlag(parser, required=False)
+    if release_track == base.ReleaseTrack.GA:
+      cp_flags.AddValidationGroup(parser, 'Update')
+
+    profile_flags = parser.add_group(mutex=True)
+    cp_flags.AddMysqlProfileGroup(profile_flags, required=False)
+    cp_flags.AddOracleProfileGroup(profile_flags, required=False)
+    cp_flags.AddPostgresqlProfileGroup(profile_flags, required=False)
+    cp_flags.AddSqlServerProfileGroup(profile_flags, required=False)
+    cp_flags.AddSalesforceProfileGroup(profile_flags, required=False)
+    cp_flags.AddGcsProfileGroup(profile_flags, release_track, required=False)
+    cp_flags.AddMongodbProfileGroup(profile_flags, required=False)
+    flags.AddLabelsUpdateFlags(parser)
+
+  @staticmethod
+  def Args(parser):
+    """Args is called by calliope to gather arguments for this command."""
+    Update.CommonArgs(parser, base.ReleaseTrack.GA)
+
+  def Run(self, args):
+    """Update a Datastream connection profile.
+
+    Args:
+      args: argparse.Namespace, The arguments that this command was invoked
+        with.
+
+    Returns:
+      A dict object representing the operations resource describing the update
+      operation if the update was successful.
+    """
+    connection_profile_ref = args.CONCEPTS.connection_profile.Parse()
+
+    if args.oracle_prompt_for_password:
+      args.oracle_password = console_io.PromptPassword(
+          'Please Enter Password: ')
+
+    if args.mysql_prompt_for_password:
+      args.mysql_password = console_io.PromptPassword('Please Enter Password: ')
+
+    if args.postgresql_prompt_for_password:
+      args.postgresql_password = console_io.PromptPassword(
+          'Please Enter Password: ')
+
+    if args.sqlserver_prompt_for_password:
+      args.sqlserver_password = console_io.PromptPassword(
+          'Please Enter Password: '
+      )
+
+    if args.salesforce_prompt_for_password:
+      args.salesforce_password = console_io.PromptPassword(
+          'Please Enter Password: '
+      )
+
+    if args.salesforce_prompt_for_security_token:
+      args.salesforce_password = console_io.PromptPassword(
+          'Please Enter Security Token: '
+      )
+
+    if args.salesforce_prompt_for_oauth2_client_secret:
+      args.salesforce_oauth2_client_secret = console_io.PromptPassword(
+          'Please Enter OAuth 2.0 Client Secret: '
+      )
+    if args.mongodb_prompt_for_password:
+      args.mongodb_password = console_io.PromptPassword(
+          'Please Enter Password: '
+      )
+
+    cp_type = (args.type).upper()
+    cp_client = connection_profiles.ConnectionProfilesClient()
+    result_operation = cp_client.Update(connection_profile_ref.RelativeName(),
+                                        cp_type, self.ReleaseTrack(), args)
+
+    client = util.GetClientInstance()
+    messages = util.GetMessagesModule()
+    resource_parser = util.GetResourceParser()
+
+    operation_ref = resource_parser.Create(
+        'datastream.projects.locations.operations',
+        operationsId=result_operation.name,
+        projectsId=connection_profile_ref.projectsId,
+        locationsId=connection_profile_ref.locationsId)
+
+    return client.projects_locations_operations.Get(
+        messages.DatastreamProjectsLocationsOperationsGetRequest(
+            name=operation_ref.operationsId))
+
+
+@base.Deprecate(
+    is_removed=False,
+    warning=('Datastream beta version is deprecated. Please use`gcloud '
+             'datastream connection-profiles update` command instead.')
+)
+@base.ReleaseTracks(base.ReleaseTrack.BETA)
+class UpdateBeta(Update):
+  """Update a Datastream connection profile."""
+  detailed_help = {'DESCRIPTION': DESCRIPTION, 'EXAMPLES': EXAMPLES_BETA}
+
+  @staticmethod
+  def Args(parser):
+    """Args is called by calliope to gather arguments for this command."""
+    Update.CommonArgs(parser, base.ReleaseTrack.BETA)