def Run(self, args):
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
instance_service = client.projects_locations_instances
if args.IsSpecified('accelerator_type') or args.IsSpecified(
'accelerator_core_count'):
operation = instance_service.SetAccelerator(
instance_util.CreateSetAcceleratorRequest(args, messages))
instance_util.HandleLRO(
operation,
args,
instance_service,
release_track,
operation_type=instance_util.OperationType.UPDATE)
if args.IsSpecified('labels'):
operation = instance_service.SetLabels(
instance_util.CreateSetLabelsRequest(args, messages))
instance_util.HandleLRO(
operation,
args,
instance_service,
release_track,
operation_type=instance_util.OperationType.UPDATE)
if args.IsSpecified('machine_type'):
operation = instance_service.SetMachineType(
instance_util.CreateSetMachineTypeRequest(args, messages))
instance_util.HandleLRO(
operation,
args,
instance_service,
release_track,
operation_type=instance_util.OperationType.UPDATE)
def Run(self, args):
instance_service = util.GetClient().projects_locations_instances
if args.IsSpecified('accelerator_type') or args.IsSpecified(
'accelerator_core_count'):
operation = instance_service.SetAccelerator(
instance_util.CreateSetAcceleratorRequest(args))
instance_util.HandleLRO(
operation,
args,
instance_service,
operation_type=instance_util.OperationType.UPDATE)
if args.IsSpecified('labels'):
operation = instance_service.SetLabels(
instance_util.CreateSetLabelsRequest(args))
instance_util.HandleLRO(
operation,
args,
instance_service,
operation_type=instance_util.OperationType.UPDATE)
if args.IsSpecified('machine_type'):
operation = instance_service.SetMachineType(
instance_util.CreateSetMachineTypeRequest(args))
instance_util.HandleLRO(
operation,
args,
instance_service,
operation_type=instance_util.OperationType.UPDATE)
def Run(self, args):
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
environment_service = client.projects_locations_environments
result = environment_service.Get(
env_util.CreateEnvironmentDescribeRequest(args, messages))
return result
def Run(self, args):
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
instance_service = client.projects_locations_instances
result = instance_service.Get(
instance_util.CreateInstanceDescribeRequest(args, messages))
return result
def Run(self, args):
"""This is what gets called when the user runs this command."""
location_service = util.GetClient().projects_locations
return list_pager.YieldFromList(
location_service,
loc_util.CreateLocationListRequest(args),
field='locations',
limit=args.limit,
batch_size_attribute='pageSize')
def Run(self, args):
instance_service = util.GetClient().projects_locations_instances
operation = instance_service.Reset(
instance_util.CreateInstanceResetRequest(args))
return instance_util.HandleLRO(
operation,
args,
instance_service,
operation_type=instance_util.OperationType.RESET)
def Run(self, args):
"""This is what gets called when the user runs this command."""
environment_service = util.GetClient().projects_locations_environments
operation = environment_service.Delete(
env_util.CreateEnvironmentDeleteRequest(args))
return env_util.HandleLRO(operation,
args,
environment_service,
is_delete=True)
def Run(self, args):
"""This is what gets called when the user runs this command."""
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
environment_service = client.projects_locations_environments
operation = environment_service.Create(
env_util.CreateEnvironmentCreateRequest(args, messages))
return env_util.HandleLRO(operation, args, environment_service,
release_track)
def Run(self, args):
"""This is what gets called when the user runs this command."""
instance_service = util.GetClient().projects_locations_instances
operation = instance_service.Create(
instance_util.CreateInstanceCreateRequest(args))
return instance_util.HandleLRO(
operation,
args,
instance_service,
operation_type=instance_util.OperationType.CREATE)
def Run(self, args):
"""This is what gets called when the user runs this command."""
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
location_service = client.projects_locations
return list_pager.YieldFromList(
location_service,
loc_util.CreateLocationListRequest(args, messages),
field='locations',
limit=args.limit,
batch_size_attribute='pageSize')
def Run(self, args):
"""This is what gets called when the user runs this command."""
if (not args.IsSpecified('location')) and (
not properties.VALUES.notebooks.location.IsExplicitlySet()):
raise parser_errors.RequiredError(argument='--location')
instance_service = util.GetClient().projects_locations_instances
return list_pager.YieldFromList(
instance_service,
instance_util.CreateInstanceListRequest(args),
field='instances',
limit=args.limit,
batch_size_attribute='pageSize')
def Run(self, args):
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
instance_service = client.projects_locations_instances
operation = instance_service.Reset(
instance_util.CreateInstanceResetRequest(args, messages))
return instance_util.HandleLRO(
operation,
args,
instance_service,
release_track,
operation_type=instance_util.OperationType.RESET)
def Run(self, args):
"""This is what gets called when the user runs this command."""
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
instance_service = client.projects_locations_instances
operation = instance_service.Create(
instance_util.CreateInstanceCreateRequest(args, client, messages))
return instance_util.HandleLRO(
operation,
args,
instance_service,
release_track,
operation_type=instance_util.OperationType.CREATE)
def Run(self, args):
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
instance_service = client.projects_locations_instances
if args.IsSpecified('target_snapshot'):
operation = instance_service.Rollback(
instance_util.CreateInstanceRollbackRequest(args, messages))
return instance_util.HandleLRO(
operation,
args,
instance_service,
release_track,
operation_type=instance_util.OperationType.ROLLBACK)
def Run(self, args):
"""This is what gets called when the user runs this command."""
release_track = self.ReleaseTrack()
client = util.GetClient(release_track)
messages = util.GetMessages(release_track)
if (not args.IsSpecified('location')) and (
not properties.VALUES.notebooks.location.IsExplicitlySet()):
raise parser_errors.RequiredError(argument='--location')
environment_service = client.projects_locations_environments
return list_pager.YieldFromList(environment_service,
env_util.CreateEnvironmentListRequest(
args, messages),
field='environments',
limit=args.limit,
batch_size_attribute='pageSize')
def Run(self, args):
environment_service = util.GetClient().projects_locations_environments
result = environment_service.Get(
env_util.CreateEnvironmentDescribeRequest(args))
return result
def Run(self, args):
instance_service = util.GetClient().projects_locations_instances
result = instance_service.Get(
instance_util.CreateInstanceDescribeRequest(args))
return result
def GetVmImageFromExistingEnvironment():
environment_service = util.GetClient().projects_locations_environments
result = environment_service.Get(
env_util.CreateEnvironmentDescribeRequest(args))
return result.vmImage
def Run(self, args):
instance_service = util.GetClient().projects_locations_instances
result = instance_service.IsUpgradeable(
instance_util.CreateInstanceIsUpgradeableRequest(args))
return result
