From: Royce Lv <lvroyce(a)linux.vnet.ibm.com>
Add support for passing params for GET method,
we will call it like:
GET /vms?state=running
Signed-off-by: Royce Lv <lvroyce(a)linux.vnet.ibm.com>
---
src/kimchi/control/base.py | 26 +++++++++++++++++++-------
src/kimchi/control/storagepools.py | 4 ++--
src/kimchi/control/utils.py | 2 ++
3 files changed, 23 insertions(+), 9 deletions(-)
diff --git a/src/kimchi/control/base.py b/src/kimchi/control/base.py
index 185c8d8..73b70fb 100644
--- a/src/kimchi/control/base.py
+++ b/src/kimchi/control/base.py
@@ -28,7 +28,7 @@ import urllib2
import kimchi.template
from kimchi.control.utils import get_class_name, internal_redirect, model_fn
-from kimchi.control.utils import parse_request, validate_method
+from kimchi.control.utils import parse_request, parse_param, validate_method
from kimchi.control.utils import validate_params
from kimchi.exception import InvalidOperation, InvalidParameter
from kimchi.exception import MissingParameter, NotFoundError, OperationFailed
@@ -212,10 +212,11 @@ class Collection(object):
return res.get()
- def _get_resources(self):
+ def _get_resources(self, filter_params):
try:
+ validate_params(filter_params, self, 'get_list')
get_list = getattr(self.model, model_fn(self, 'get_list'))
- idents = get_list(*self.model_args)
+ idents = get_list(*self.model_args, **filter_params)
res_list = []
for ident in idents:
# internal text, get_list changes ident to unicode for sorted
@@ -234,15 +235,26 @@ class Collection(object):
args = self.resource_args + [ident.decode("utf-8")]
return self.resource(self.model, *args)
- def get(self):
- resources = self._get_resources()
+ def filter_data(self, resources, filter_params):
data = []
for res in resources:
- data.append(res.data)
+ valid = True
+ for key, val in filter_params.items():
+ if not ( key in res.data and res.data[key] == val):
+ valid = False
+ break
+ if valid:
+ data.append(res.data)
+ return data
+
+ def get(self):
+ filter_params = parse_param()
+ resources = self._get_resources(filter_params)
+ data = self.filter_data(resources, filter_params)
return kimchi.template.render(get_class_name(self), data)
@cherrypy.expose
- def index(self, *args):
+ def index(self, *args, **kwargs):
method = validate_method(('GET', 'POST'))
if method == 'GET':
try:
diff --git a/src/kimchi/control/storagepools.py b/src/kimchi/control/storagepools.py
index 466b4b6..0df14c5 100644
--- a/src/kimchi/control/storagepools.py
+++ b/src/kimchi/control/storagepools.py
@@ -61,9 +61,9 @@ class StoragePools(Collection):
return resp
- def _get_resources(self):
+ def _get_resources(self, filter_params):
try:
- res_list = super(StoragePools, self)._get_resources()
+ res_list = super(StoragePools, self)._get_resources(filter_params)
# Append reserved pools
isos = getattr(self, ISO_POOL_NAME)
isos.lookup()
diff --git a/src/kimchi/control/utils.py b/src/kimchi/control/utils.py
index c3c5f8e..4b3c4b0 100644
--- a/src/kimchi/control/utils.py
+++ b/src/kimchi/control/utils.py
@@ -81,6 +81,8 @@ def parse_request():
raise cherrypy.HTTPError(415, "This API only supports"
" 'application/json'")
+def parse_param():
+ return cherrypy.request.params
def internal_redirect(url):
raise cherrypy.InternalRedirect(url.encode("utf-8"))
--
1.8.1.2