fix bug

Author: naderzare

src/World.py

@@ -63,13 +63,14 @@ def update(self, message):
                     self.walls.append(Vector2D(i, j))
 
         for s in self.snakes:
-            self.snakes[s].set_head(Vector2D(message.world['heads'][self.snakes[s].name][0], message.world['heads'][self.snakes[s].name][1]))
+            id = message.name_id[self.snakes[s].name]
+            self.snakes[id].set_head(Vector2D(message.world['heads'][self.snakes[s].name][0], message.world['heads'][self.snakes[s].name][1]))
 
     def get_self(self):
         return self.snakes[self.self_id]
 
     def get_snake(self, id):
-        return self.snakes[self.id]
+        return self.snakes[id]
 
     def get_walls(self):
         return self.walls

src/base/Message.py

@@ -124,14 +124,15 @@ def parse(coded_msg):
 
 
 class MessageClientWorld(Message):
-    def __init__(self, cycle, world, score):
+    def __init__(self, cycle, world, score, name_id):
         self.type = "MessageClientWorld"
         self.cycle = cycle
         self.world = world
         self.score = score
+        self.name_id = name_id
 
     def build(self):
-        msg = {"message_type": self.type, "value": {"cycle": self.cycle, "score": self.score, "world": self.world}}
+        msg = {"message_type": self.type, "value": {"cycle": self.cycle, "score": self.score, "world": self.world, "name_id": self.name_id}}
         str_msg = str.encode(str(msg))
         return str_msg
 
@@ -142,7 +143,8 @@ def parse(coded_msg):
             cycle = msg['value']['cycle']
             world = msg['value']['world']
             score = msg['value']['score']
-            message = MessageClientWorld(cycle, world, score)
+            name_id = msg['value']['name_id']
+            message = MessageClientWorld(cycle, world, score, name_id)
             return True, message
         return False, None