I'll bail now

james-gray · Feb 22, 2015 · 7353bcf · 7353bcf
1 parent eb73a77
commit 7353bcf
Showing 1 changed file with 21 additions and 8 deletions.
diff --git a/app/main.py b/app/main.py
@@ -160,23 +160,36 @@ def move_edge(nagini, bounds):
 def move():
     data = bottle.request.json
 
+    width = len(data['board'])
+    height = len(data['board'][0])
+
     bounds = {
         "up": 1,
-        "down": len(data['board'][0]) - 2,
-        "right": len(data['board']) - 2,
+        "down": height - 2,
+        "right": width - 2,
         "left": 1,
     }
 
     nagini = [s for s in data['snakes'] if s['name'] == 'nagini'][0]
 
     directions = get_possible_directions(nagini, data['board'], bounds)
     if not directions:
-        # Commit suicide honorably so as not to give any victories to
-        # the other inferior snakes!
-        return json.dumps({
-            'move': seppuku(nagini, data['board']),
-            'taunt': 'You will always remember this as the day you almost caught Captain Jack Sparrow!'
-        })
+        bounds = {
+            "up": 0,
+            "down": height - 1,
+            "right": width - 1,
+            "left": 0,
+        }
+
+        directions = get_possible_directions(nagini, data['board'], bounds)
+        if not directions:
+            # Commit suicide honorably so as not to give any victories to
+            # the other inferior snakes!
+
+            return json.dumps({
+                'move': seppuku(nagini, data['board']),
+                'taunt': 'You will always remember this as the day you almost caught Captain Jack Sparrow!'
+            })
 
     direction = move_edge(nagini, bounds)