merge latest code

[plstackapi.git] / plstackapi / core / api / images.py

diff --git a/plstackapi/core/api/images.py b/plstackapi/core/api/images.py
index b773b0b..0b14043 100644 (file)
--- a/plstackapi/core/api/images.py
+++ b/plstackapi/core/api/images.py
@@ -3,6 +3,16 @@ from plstackapi.openstack.driver import OpenStackDriver
 from plstackapi.core.api.auth import auth_check
 from plstackapi.core.models import Image
  
+def _get_images(filter):
+    if isinstance(filter, int):
+        images = image.objects.filter(id=filter)
+    elif isinstance(filter, StringTypes):
+        images = image.objects.filter(name=filter)
+    elif isinstance(filer, dict):
+        images = image.objects.filter(**filter)
+    else:
+        images = []
+    return images
 
 def add_image(auth, fields={}):
     """not implemented"""
@@ -14,7 +24,7 @@ def delete_image(auth, filter={}):
 
 def get_images(auth, filter={}):
     auth_check(auth)   
-    images = Image.objects.filter(**filter)
+    images = _get_images(filter)
     return images