@@ -733,21 +733,32 @@ def take_action(self, parsed_args):
733733 # values if the project or class does not exist. This is expected
734734 # behavior. However, we have already checked for the presence of the
735735 # project above so it shouldn't be an issue.
736- if parsed_args .service in {'all' , 'compute' }:
736+ if parsed_args .service == 'compute' or (
737+ parsed_args .service == 'all'
738+ and self .app .client_manager .is_compute_endpoint_enabled ()
739+ ):
737740 compute_quota_info = get_compute_quotas (
738741 self .app ,
739742 project ,
740743 detail = parsed_args .usage ,
741744 default = parsed_args .default ,
742745 )
743- if parsed_args .service in {'all' , 'volume' }:
746+
747+ if parsed_args .service == 'volume' or (
748+ parsed_args .service == 'all'
749+ and self .app .client_manager .is_volume_endpoint_enabled ()
750+ ):
744751 volume_quota_info = get_volume_quotas (
745752 self .app ,
746753 project ,
747754 detail = parsed_args .usage ,
748755 default = parsed_args .default ,
749756 )
750- if parsed_args .service in {'all' , 'network' }:
757+
758+ if parsed_args .service == 'network' or (
759+ parsed_args .service == 'all'
760+ and self .app .client_manager .is_network_endpoint_enabled ()
761+ ):
751762 network_quota_info = get_network_quotas (
752763 self .app ,
753764 project ,
@@ -880,12 +891,18 @@ def take_action(self, parsed_args):
880891 )
881892
882893 # compute quotas
883- if parsed_args .service in {'all' , 'compute' }:
894+ if parsed_args .service == 'compute' or (
895+ parsed_args .service == 'all'
896+ and self .app .client_manager .is_compute_endpoint_enabled ()
897+ ):
884898 compute_client = self .app .client_manager .sdk_connection .compute
885899 compute_client .revert_quota_set (project .id )
886900
887901 # volume quotas
888- if parsed_args .service in {'all' , 'volume' }:
902+ if parsed_args .service == 'volume' or (
903+ parsed_args .service == 'all'
904+ and self .app .client_manager .is_volume_endpoint_enabled ()
905+ ):
889906 volume_client = self .app .client_manager .sdk_connection .volume
890907 volume_client .revert_quota_set (project .id )
891908
0 commit comments