from google.cloud import gkehub_v1
from google.oauth2 import service_account
from google.cloud.gkehub_v1.types import MembershipFeatureSpec
from google.cloud.gkehub_v1.types import feature
from google.cloud.gkehub_v1 import configmanagement_v1

projectId = 'xxxxxx'
location = 'global'


def create_feature(client, parent, name):
    git_config = configmanagement_v1.GitConfig(
        sync_repo="https://github.com/GoogleCloudPlatform/anthos-config-management-samples",
        sync_branch="main",
        policy_dir="multi-environments-kustomize/config-source/overlays/dev",
        secret_type=None
    )
    config_sync = configmanagement_v1.ConfigSync(
        git=git_config,
        source_format='unstructured'
    )
    configmanagement = configmanagement_v1.MembershipSpec(
        config_sync=config_sync,
        version='1.11.2'
    )
    member_spec = MembershipFeatureSpec(
        configmanagement=configmanagement
    )
    feature_spec = feature.Feature(
        name=name,
        membership_specs={
            name: member_spec

        }
    )
    # Initialize request argument(s)
    request = gkehub_v1.CreateFeatureRequest(
        parent=parent,
        feature_id='configmanagement',
        resource=feature_spec
    )

    # Make the request
    operation = client.create_feature(request=request)

    print("Waiting for operation to complete...")

    response = operation.result()
    print(response)

    # Handle the response
    print(response.pb)

cluster_name = 'xxxxxxxxxx'
credentials = service_account.Credentials.from_service_account_file('credentials.json')
client = gkehub_v1.GkeHubClient(credentials=credentials)
parent = 'projects/%s/locations/%s' % (projectId, location)
membership_name = '%s/memberships/%s' % (parent, cluster_name)
create_feature(client, parent, membership_name)

Traceback (most recent call last):
  File "list_config.py", line 60, in <module>
    create_feature(client, parent, membership_name)
  File "list_config.py", line 26, in create_feature
    member_spec = MembershipFeatureSpec(
  File "\pythonProjectAnthos\venv\lib\site-packages\proto\message.py", line 585, in __init__
    super().__setattr__("_pb", self._meta.pb(**params))
TypeError: Parameter to MergeFrom() must be instance of same class: expected google.cloud.gkehub.configmanagement.v1.MembershipSpec got MembershipSpec.