134 lines
4.4 KiB
Python
134 lines
4.4 KiB
Python
# -*- coding: utf-8 -*- #
|
|
# Copyright 2025 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 a new investigation."""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import unicode_literals
|
|
|
|
import datetime
|
|
|
|
from googlecloudsdk.api_lib.gemini_cloud_assist import args as gca_args
|
|
from googlecloudsdk.api_lib.gemini_cloud_assist import util as gca_util
|
|
from googlecloudsdk.calliope import arg_parsers
|
|
from googlecloudsdk.calliope import base
|
|
from googlecloudsdk.core.util import times
|
|
|
|
|
|
@base.DefaultUniverseOnly
|
|
class Create(base.Command):
|
|
r"""Update an existing investigation.
|
|
|
|
Shows metadata for the investigation after updating.
|
|
|
|
This command can fail for the following reasons:
|
|
* The investigation ID doesn't exist.
|
|
* The active account does not have permission to update the investigation.
|
|
* The investigation is currently running.
|
|
|
|
## EXAMPLES
|
|
|
|
The following command updates title, text description, resources, and start
|
|
time of the existing investigation with the name`example-foo-bar-1`:
|
|
|
|
$ {command} example-foo-bar-1 --title="Example Investigation" \
|
|
--issue="I have a problem" --start-time=2025-07-10
|
|
"""
|
|
|
|
@staticmethod
|
|
def Args(parser):
|
|
gca_args.AddInvestigationResourceArg(
|
|
parser, verb="to update", required=True, allow_no_id=False
|
|
)
|
|
parser.display_info.AddFormat("value(investigation_markdown_detailed())")
|
|
parser.add_argument(
|
|
"--issue",
|
|
required=False,
|
|
help=(
|
|
"A description of the issue you are investigating, or an error log."
|
|
),
|
|
)
|
|
|
|
start_time_group = parser.add_mutually_exclusive_group(sort_args=False)
|
|
start_time_group.add_argument(
|
|
"--start-time",
|
|
required=False,
|
|
type=times.ParseDateTime,
|
|
help="The estimated start time of the issue you are investigating.",
|
|
)
|
|
start_time_group.add_argument(
|
|
"--clear-start-time",
|
|
required=False,
|
|
action="store_true",
|
|
help="Clear the start time of the investigation.",
|
|
)
|
|
|
|
end_time_group = parser.add_mutually_exclusive_group(sort_args=False)
|
|
end_time_group.add_argument(
|
|
"--end-time",
|
|
required=False,
|
|
type=times.ParseDateTime,
|
|
help="The estimated end time of the issue you are investigating.",
|
|
)
|
|
end_time_group.add_argument(
|
|
"--clear-end-time",
|
|
required=False,
|
|
action="store_true",
|
|
help="Clear the end time of the investigation.",
|
|
)
|
|
|
|
title_group = parser.add_mutually_exclusive_group(sort_args=False)
|
|
title_group.add_argument(
|
|
"--title",
|
|
required=False,
|
|
help="The desired title of the resulting investigation.",
|
|
)
|
|
title_group.add_argument(
|
|
"--clear-title",
|
|
required=False,
|
|
action="store_true",
|
|
help="Clear the title of the investigation.",
|
|
)
|
|
|
|
resources_group = parser.add_mutually_exclusive_group(sort_args=False)
|
|
resources_group.add_argument(
|
|
"--resources",
|
|
required=False,
|
|
metavar="RESOURCE",
|
|
type=arg_parsers.ArgList(),
|
|
help="A list of resources relevant to the investigation",
|
|
)
|
|
resources_group.add_argument(
|
|
"--clear-resources",
|
|
required=False,
|
|
action="store_true",
|
|
help="Clear the resources of the investigation.",
|
|
)
|
|
|
|
def Run(self, args):
|
|
investigation_ref = args.CONCEPTS.investigation.Parse()
|
|
updated_investigation = gca_util.UpdateInvestigation(
|
|
investigation_ref,
|
|
"" if args.clear_title else args.title,
|
|
args.issue,
|
|
datetime.datetime.min if args.clear_start_time else args.start_time,
|
|
datetime.datetime.max if args.clear_end_time else args.end_time,
|
|
[] if args.clear_resources else args.resources,
|
|
)
|
|
return gca_util.RunInvestigationRevisionBlocking(
|
|
updated_investigation.revision
|
|
)
|