fixing imports

1 files changed, 6 insertions, 4 deletions

diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py
index 0af217e3..5219a8da 100644
--- a/megapixels/app/server/api.py
+++ b/megapixels/app/server/api.py
@@ -148,13 +148,15 @@ def name_lookup(dataset_name):
     'q': q,
     'timing': time.time() - start,
   }
-  if len(terms) == 1:
-    names = dataset.search_name('%' + term + '%') if term else []
+  if len(terms) == 0:
+    results = []
+  elif len(terms) == 1:
+    names = dataset.search_name('%' + terms[0] + '%') if terms[0] else []
     results = dataset.get_file_records_for_identities(names)
   else:
     lookup = {}
     results_lookup = {}
-    for i, term in enumerate(terms):
+    for i, term in enumerate(terms[0:5]):
       search_term = '%' + term + '%'
       names = dataset.search_name(term) if term else []
       for name in names:
@@ -163,7 +165,7 @@ def name_lookup(dataset_name):
         else:
           lookup[name.id] = 1
           results_lookup[name.id] = name
-    top_names = [results_lookup[item[0]] for item in sorted(lookup.items(), key=operator.itemgetter(1))][0:20]
+    top_names = [results_lookup[item[0]] for item in sorted(lookup.items(), key=operator.itemgetter(1))][0:10]
     results = dataset.get_file_records_for_identities(top_names)
 
   # print(results)