directly show map if only one result comes back from search

1 files changed, 26 insertions, 18 deletions

diff --git a/lib/d2/app/controllers/index.py b/lib/d2/app/controllers/index.py
index 23f7b5e..7febb82 100644
--- a/lib/d2/app/controllers/index.py
+++ b/lib/d2/app/controllers/index.py
@@ -30,25 +30,29 @@ class IndexController(BaseController):
             self._plot_ = PlotAdapter.load()
         return self._plot_
 
+    def _show_map(self, id_type, id):
+        if id_type == 'occupant':
+            forges = self._forge.get_forge_by_occupant_id(id)
+            details = self._detail.occupant_details(id)
+            plots = []
+            if len(forges) > 0:
+                plot_ids = [forge.plot_id for forge in forges]
+                plots = self._plot.get_all(plot_ids)
+        else: # plot
+            forges = self._forge.get_forge_by_plot_id(id)
+            details = self._detail.plot_details(id)
+            plots = self._plot.get_all([id])
+
+        self.render('index.html', 
+                mode="map", 
+                details=details,
+                plots=plots,
+                forges=forges)
+
+
     def get(self, id_type=None, id=None):
         if id_type and id:
-            if id_type == 'occupant':
-                forges = self._forge.get_forge_by_occupant_id(id)
-                details = self._detail.occupant_details(id)
-                plots = []
-                if len(forges) > 0:
-                    plot_ids = [forge.plot_id for forge in forges]
-                    plots = self._plot.get_all(plot_ids)
-            else: # plot
-                forges = self._forge.get_forge_by_plot_id(id)
-                details = self._detail.plot_details(id)
-                plots = self._plot.get_all([id])
-
-            self.render('index.html', 
-                    mode="map", 
-                    details=details,
-                    plots=plots,
-                    forges=forges)
+            self._show_map(id_type, id)
         else:
             self.render('index.html', mode="get")
         
@@ -57,5 +61,9 @@ class IndexController(BaseController):
         name = self.get_argument('name')
         details = self._search.search(name)
 
-        self.render('index.html', mode='post',
+        if len(details) == 1:
+            self._show_map(details[0].id_type,
+                           details[0].id)
+        else:
+            self.render('index.html', mode='post',
                    details=details)