From fbf287cc59c2e13734432420f35792c7daa357d5 Mon Sep 17 00:00:00 2001 From: Tony Mack Date: Tue, 9 Apr 2013 23:16:28 -0400 Subject: [PATCH] Fix NameError --- plstackapi/core/api/slices.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/plstackapi/core/api/slices.py b/plstackapi/core/api/slices.py index 5e63df8..7ebeb1e 100644 --- a/plstackapi/core/api/slices.py +++ b/plstackapi/core/api/slices.py @@ -26,6 +26,13 @@ def lookup_site(fields): sites = Site.objects.filter(login_base=login_base) if sites: site = sites[0] + elif 'site' in fields: + if isinstance(fields['site'], int): + sites = Site.objects.filter(id=fields['site']) + else: + sites = Site.objects.filter(login_base=fields['site']) + if sites: + site = sites[0] if not site: raise Exception, "No such site: %s" % login_base return site @@ -87,8 +94,8 @@ def delete_slice(auth, filter={}): def get_slices(auth, filter={}): client = auth_check(auth) - site = lookup_site(fields) - if site: fields['site'] = site + site = lookup_site(filter) + if site: filter['site'] = site slices = Slice.objects.filter(**filter) return slices -- 2.47.0