fix pk lookups

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

diff --git a/plstackapi/core/api/flavors.py b/plstackapi/core/api/flavors.py
index ee55f73..94c38eb 100644 (file)
--- a/plstackapi/core/api/flavors.py
+++ b/plstackapi/core/api/flavors.py
@@ -1,14 +1,17 @@
+from types import StringTypes
 from plstackapi.openstack.client import OpenStackClient
 from plstackapi.openstack.driver import OpenStackDriver
 from plstackapi.core.api.auth import auth_check
 from plstackapi.core.models import Flavor
  
 def _get_flavors(filter):
+    if isinstance(filter, StringTypes) and filter.isdigit():
+        filter = int(filter)
     if isinstance(filter, int):
         flavors = Flavor.objects.filter(id=filter)
     elif isinstance(filter, StringTypes):
         flavors = Flavor.objects.filter(name=filter)
-    elif isinstance(filer, dict):
+    elif isinstance(filter, dict):
         flavors = Flavor.objects.filter(**filter)
     else:
         flavors = []