fix pk lookups

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

diff --git a/plstackapi/core/api/sites.py b/plstackapi/core/api/sites.py
index 540f815..daf8fa1 100644 (file)
--- a/plstackapi/core/api/sites.py
+++ b/plstackapi/core/api/sites.py
@@ -1,3 +1,4 @@
+from types import StringTypes
 from plstackapi.openstack.client import OpenStackClient
 from plstackapi.openstack.driver import OpenStackDriver
 from plstackapi.core.api.auth import auth_check
@@ -5,11 +6,13 @@ from plstackapi.core.models import Site
 
 
 def _get_sites(filter):
+    if isinstance(filter, StringTypes) and filter.isdigit():
+        filter = int(filter)
     if isinstance(filter, int):
         sites = Site.objects.filter(id=filter)
     elif isinstance(filter, StringTypes):
         sites = Site.objects.filter(name=filter)
-    elif isinstance(filer, dict):
+    elif isinstance(filter, dict):
         sites = Site.objects.filter(**filter)
     else:
         sites = []