80 lines
3.2 KiB
Python
80 lines
3.2 KiB
Python
# -*- 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.
|
|
"""Flags module for Essential Contacts commands."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
from googlecloudsdk.calliope import arg_parsers
|
|
from googlecloudsdk.command_lib.util.apis import arg_utils
|
|
|
|
|
|
def AddContactIdArg(parser, help_text='id of contact'):
|
|
"""Adds an arg for the contact id to the parser."""
|
|
parser.add_argument('CONTACT_ID', type=str, help=help_text)
|
|
|
|
|
|
def AddParentArgs(parser):
|
|
"""Adds args for the parent resource of a contact to the parser."""
|
|
parent_group = parser.add_mutually_exclusive_group(required=False)
|
|
parent_group.add_argument(
|
|
'--project',
|
|
help='project number or id where contacts are set. If neither --project, --folder, nor --organization are provided then the config property [core/project] will be used as the resource.'
|
|
)
|
|
parent_group.add_argument(
|
|
'--folder',
|
|
help='folder number where contacts are set. If neither --project, --folder, nor --organization are provided then the config property [core/project] will be used as the resource.'
|
|
)
|
|
parent_group.add_argument(
|
|
'--organization',
|
|
help='organization number where contacts are set. If neither --project, --folder, nor --organization are provided then the config property [core/project] will be used as the resource.'
|
|
)
|
|
|
|
|
|
def _NotificationCategoryEnumMapper(notification_category_enum_message):
|
|
return arg_utils.ChoiceEnumMapper('--notification-categories',
|
|
notification_category_enum_message)
|
|
|
|
|
|
def AddNotificationCategoriesArg(
|
|
parser,
|
|
notification_category_enum_message,
|
|
required=False,
|
|
help_text='list of notification categories contact is subscribed to.'):
|
|
"""Adds the arg for specifying a list of notification categories to the parser."""
|
|
parser.add_argument(
|
|
'--notification-categories',
|
|
metavar='NOTIFICATION_CATEGORIES',
|
|
type=arg_parsers.ArgList(
|
|
choices=_NotificationCategoryEnumMapper(
|
|
notification_category_enum_message).choices),
|
|
help=help_text,
|
|
required=required)
|
|
|
|
|
|
def AddEmailArg(parser, required=False, help_text='email address of contact.'):
|
|
"""Adds an arg for the contact's email to the parser."""
|
|
parser.add_argument('--email', help=help_text, required=required)
|
|
|
|
|
|
def AddLanugageArg(
|
|
parser,
|
|
required=False,
|
|
help_text='preferred language of contact. Must be a valid ISO 639-1 '
|
|
'language code.'):
|
|
"""Adds an arg for the contact's preferred language to the parser."""
|
|
parser.add_argument('--language', help=help_text, required=required)
|