@@ -746,21 +746,32 @@ def take_action(self, parsed_args):
746746 # values if the project or class does not exist. This is expected
747747 # behavior. However, we have already checked for the presence of the
748748 # project above so it shouldn't be an issue.
749- if parsed_args .service in {'all' , 'compute' }:
749+ if parsed_args .service == 'compute' or (
750+ parsed_args .service == 'all'
751+ and self .app .client_manager .is_compute_endpoint_enabled ()
752+ ):
750753 compute_quota_info = get_compute_quotas (
751754 self .app ,
752755 project ,
753756 detail = parsed_args .usage ,
754757 default = parsed_args .default ,
755758 )
756- if parsed_args .service in {'all' , 'volume' }:
759+
760+ if parsed_args .service == 'volume' or (
761+ parsed_args .service == 'all'
762+ and self .app .client_manager .is_volume_endpoint_enabled ()
763+ ):
757764 volume_quota_info = get_volume_quotas (
758765 self .app ,
759766 project ,
760767 detail = parsed_args .usage ,
761768 default = parsed_args .default ,
762769 )
763- if parsed_args .service in {'all' , 'network' }:
770+
771+ if parsed_args .service == 'network' or (
772+ parsed_args .service == 'all'
773+ and self .app .client_manager .is_network_endpoint_enabled ()
774+ ):
764775 network_quota_info = get_network_quotas (
765776 self .app ,
766777 project ,
@@ -906,12 +917,18 @@ def take_action(self, parsed_args):
906917 )
907918
908919 # compute quotas
909- if parsed_args .service in {'all' , 'compute' }:
920+ if parsed_args .service == 'compute' or (
921+ parsed_args .service == 'all'
922+ and self .app .client_manager .is_compute_endpoint_enabled ()
923+ ):
910924 compute_client = self .app .client_manager .compute
911925 compute_client .revert_quota_set (project .id )
912926
913927 # volume quotas
914- if parsed_args .service in {'all' , 'volume' }:
928+ if parsed_args .service == 'volume' or (
929+ parsed_args .service == 'all'
930+ and self .app .client_manager .is_volume_endpoint_enabled ()
931+ ):
915932 volume_client = self .app .client_manager .sdk_connection .volume
916933 volume_client .revert_quota_set (project .id )
917934
0 commit comments