@@ -78,7 +78,7 @@ def CreateDiskCopy(
7878 try :
7979 if disk_name :
8080 disk_to_copy = src_project .compute .GetDisk (disk_name )
81- elif instance_name :
81+ else :
8282 instance = src_project .compute .GetInstance (instance_name )
8383 disk_to_copy = instance .GetBootDisk ()
8484
@@ -713,19 +713,13 @@ def TriageInstance(project_id: str, instance_name: str) -> Dict[str, Any]:
713713
714714 cpu_usage = project .monitoring .GetCpuUsage (
715715 instance_ids = [instance_info ['id' ]], aggregation_minutes = 1 )
716- if cpu_usage :
717- parsed_cpu = cpu_usage [0 ].get ('cpu_usage' , [])
716+ parsed_cpu = cpu_usage [0 ].get ('cpu_usage' , []) if cpu_usage else None
718717
719718
720719 gce_gpu_usage = project .monitoring .GetInstanceGPUUsage (
721- instance_ids = [instance_info ['id' ]])
722- if gce_gpu_usage :
723- parsed_gce_gpu = gce_gpu_usage
724-
720+ instance_ids = [instance_info ['id' ]])
725721
726722 gke_gpu_usage = project .monitoring .GetNodeAccelUsage ()
727- if gke_gpu_usage :
728- parsed_gke_gpu = gke_gpu_usage
729723
730724 instance_triage = {
731725 'instance_info' : {
@@ -741,25 +735,25 @@ def TriageInstance(project_id: str, instance_name: str) -> Dict[str, Any]:
741735 'data_type' : 'service_accounts' ,
742736 'values' : instance_info ['serviceAccounts' ]
743737 },
744- {
745- 'data_type' : 'firewalls' ,
746- 'values' : instance .GetNormalisedFirewalls ()
747- }, {
748- 'data_type' : 'cpu_usage' , 'values' : parsed_cpu
749- }, {
750- 'data_type' : 'gce_gpu_usage' , 'values' : parsed_gce_gpu
751- }, {
752- 'data_type' : 'gke_gpu_usage' , 'values' : parsed_gke_gpu
753- }, {
754- 'data_type' :
755- 'ssh_auth' ,
756- 'values' :
757- CheckInstanceSSHAuth (
758- project_id , instance_info ['name' ])
759- }, {
760- 'data_type' : 'active_services' ,
761- 'values' : parsed_services
762- }]
738+ {
739+ 'data_type' : 'firewalls' ,
740+ 'values' : instance .GetNormalisedFirewalls ()
741+ }, {
742+ 'data_type' : 'cpu_usage' , 'values' : parsed_cpu
743+ }, {
744+ 'data_type' : 'gce_gpu_usage' , 'values' : gce_gpu_usage
745+ }, {
746+ 'data_type' : 'gke_gpu_usage' , 'values' : gke_gpu_usage
747+ }, {
748+ 'data_type' :
749+ 'ssh_auth' ,
750+ 'values' :
751+ CheckInstanceSSHAuth (
752+ project_id , instance_info ['name' ])
753+ }, {
754+ 'data_type' : 'active_services' ,
755+ 'values' : parsed_services
756+ }]
763757 }
764758
765759 return instance_triage
0 commit comments