Implement key pair quotas.

Fixes LP Bug #987058.

Change-Id: Ibefcdc448cb60754d5358fd08d74f7d279c8b16e

2 files changed, 12 insertions, 0 deletions

diff --git a/nova/db/api.py b/nova/db/api.py
index 5de921667..1d4103c2d 100644
--- a/nova/db/api.py
+++ b/nova/db/api.py
@@ -703,6 +703,11 @@ def key_pair_get_all_by_user(context, user_id):
     return IMPL.key_pair_get_all_by_user(context, user_id)
 
 
+def key_pair_count_by_user(context, user_id):
+    """Count number of key pairs for the given user ID."""
+    return IMPL.key_pair_count_by_user(context, user_id)
+
+
 ####################
 
 
diff --git a/nova/db/sqlalchemy/api.py b/nova/db/sqlalchemy/api.py
index 1d7509aef..105bce004 100644
--- a/nova/db/sqlalchemy/api.py
+++ b/nova/db/sqlalchemy/api.py
@@ -1816,6 +1816,13 @@ def key_pair_get_all_by_user(context, user_id):
                    all()
 
 
+def key_pair_count_by_user(context, user_id):
+    authorize_user_context(context, user_id)
+    return model_query(context, models.KeyPair, read_deleted="no").\
+                   filter_by(user_id=user_id).\
+                   count()
+
+
 ###################