59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
# -*- coding: utf-8 -*- #
|
|
# Copyright 2022 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.
|
|
"""Base classes for abstracting away common logic."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
import enum
|
|
|
|
from googlecloudsdk.api_lib.util import apis
|
|
|
|
API_NAME = 'recommender'
|
|
|
|
RECOMMENDER_MESSAGE_PREFIX = {
|
|
'v1': 'GoogleCloudRecommenderV1',
|
|
'v1beta1': 'GoogleCloudRecommenderV1beta1',
|
|
'v1alpha2': 'GoogleCloudRecommenderV1alpha2'
|
|
}
|
|
|
|
|
|
class EntityType(enum.Enum):
|
|
"""Cloud Entity Types."""
|
|
ORGANIZATION = 1
|
|
FOLDER = 2
|
|
PROJECT = 3
|
|
BILLING_ACCOUNT = 4
|
|
|
|
|
|
class ClientBase(object):
|
|
"""Base client class for all versions."""
|
|
|
|
def __init__(self, api_version):
|
|
self._client = apis.GetClientInstance(API_NAME, api_version)
|
|
self._api_version = api_version
|
|
self._messages = self._client.MESSAGES_MODULE
|
|
self._message_prefix = RECOMMENDER_MESSAGE_PREFIX[api_version]
|
|
|
|
def _GetMessage(self, message_name):
|
|
"""Returns the API messages class by name."""
|
|
return getattr(self._messages, message_name, None)
|
|
|
|
def _GetVersionedMessage(self, message_name):
|
|
"""Returns the versioned API messages class by name."""
|
|
return self._GetMessage('{prefix}{name}'.format(
|
|
prefix=self._message_prefix, name=message_name))
|