From 27024d70af4756cb6e4b210b025ed7427541f773 Mon Sep 17 00:00:00 2001
From: Richard Theis <rtheis@us.ibm.com>
Date: Thu, 21 Apr 2016 11:33:24 -0500
Subject: Support quota show for current project

The "os quota show" command "<project/class>" argument is now
optional. If not specified, the user's current project is used.
This allows non-admin users to show quotas for their current
project.

Change-Id: I602d4cc09c9d29ce84271eff78137f8810cb1a47
Closes-Bug: #1572733
---
 openstackclient/common/quota.py            | 29 ++++++++++++++++++-----------
 openstackclient/tests/common/test_quota.py | 11 +++++++++++
 2 files changed, 29 insertions(+), 11 deletions(-)

(limited to 'openstackclient')

diff --git a/openstackclient/common/quota.py b/openstackclient/common/quota.py
index b3d4c3b6..b497a44d 100644
--- a/openstackclient/common/quota.py
+++ b/openstackclient/common/quota.py
@@ -145,7 +145,8 @@ class ShowQuota(command.ShowOne):
         parser.add_argument(
             'project',
             metavar='<project/class>',
-            help='Show this project or class (name/ID)',
+            nargs='?',
+            help='Show quotas for this project or class (name or ID)',
         )
         type_group = parser.add_mutually_exclusive_group()
         type_group.add_argument(
@@ -164,12 +165,22 @@ class ShowQuota(command.ShowOne):
         )
         return parser
 
+    def _get_project(self, parsed_args):
+        if parsed_args.project is not None:
+            identity_client = self.app.client_manager.identity
+            project = utils.find_resource(
+                identity_client.projects,
+                parsed_args.project,
+            ).id
+        elif self.app.client_manager.auth_ref:
+            # Get the project from the current auth
+            project = self.app.client_manager.auth_ref.project_id
+        else:
+            project = None
+        return project
+
     def get_compute_volume_quota(self, client, parsed_args):
-        identity_client = self.app.client_manager.identity
-        project = utils.find_resource(
-            identity_client.projects,
-            parsed_args.project,
-        ).id
+        project = self._get_project(parsed_args)
 
         try:
             if parsed_args.quota_class:
@@ -189,11 +200,7 @@ class ShowQuota(command.ShowOne):
         if parsed_args.quota_class or parsed_args.default:
             return {}
         if self.app.client_manager.is_network_endpoint_enabled():
-            identity_client = self.app.client_manager.identity
-            project = utils.find_resource(
-                identity_client.projects,
-                parsed_args.project,
-            ).id
+            project = self._get_project(parsed_args)
             return self.app.client_manager.network.get_quota(project)
         else:
             return {}
diff --git a/openstackclient/tests/common/test_quota.py b/openstackclient/tests/common/test_quota.py
index edf29c9b..ba7ee469 100644
--- a/openstackclient/tests/common/test_quota.py
+++ b/openstackclient/tests/common/test_quota.py
@@ -59,6 +59,7 @@ class TestQuota(compute_fakes.TestComputev2):
         self.service_catalog_mock = \
             self.app.client_manager.auth_ref.service_catalog
         self.service_catalog_mock.reset_mock()
+        self.app.client_manager.auth_ref.project_id = identity_fakes.project_id
 
 
 class TestQuotaSet(TestQuota):
@@ -304,3 +305,13 @@ class TestQuotaShow(TestQuota):
             identity_fakes.project_id)
         self.volume_quotas_class_mock.get.assert_called_with(
             identity_fakes.project_id)
+
+    def test_quota_show_no_project(self):
+        parsed_args = self.check_parser(self.cmd, [], [])
+
+        self.cmd.take_action(parsed_args)
+
+        self.quotas_mock.get.assert_called_with(identity_fakes.project_id)
+        self.volume_quotas_mock.get.assert_called_with(
+            identity_fakes.project_id)
+        self.network.get_quota.assert_called_with(identity_fakes.project_id)
-- 
cgit v1.2.1