Changes in [158:7fc18f23ae89:159:37d7d9c8d975]

Files:

• .hgtags (added)
• engine.py (modified) (4 diffs)
• main.py (modified) (3 diffs)

engine.py

@@ -99 +99 @@
     - 'canvas' - Widget game field is showing on
     - 'snakes' - list of snakes loaded
+    - 'psnakes' - list of snakes loaded in previous match, if other snakes are not loaded
     - 'start_snake_length' - starting length of the snake"""
     
@@ -106 +107 @@
         self.canvas = canvas
         self.snakes = [None, None, None, None]
+        self.psnakes = [None, None, None, None]
         self.init_field()
         self.start_snake_length = 10
@@ -174 +176 @@
             pass
 
-    def create_snake(self, snake_number):
+    def create_snake(self, snake_number, old_snake = None):
         """Create the snake:
         position choice is based on number or placement of 'Load' button
@@ -195 +197 @@
             3:'yellow',
             4:'red',}
-        self.snakes[snake_number-1] = snake.Snake(cells, color_dic[snake_number])
+        if old_snake == None:
+            self.snakes[snake_number-1] = snake.Snake(cells, color_dic[snake_number])
+        else:
+            old_snake.cells = cells
+        self.psnakes[snake_number-1] = self.snakes[snake_number-1]
         return self.snakes[snake_number-1]
     

main.py

@@ -47 +47 @@
         Initialyzing snake and draw it on the field.
         Return field back to default after end of the game."""
-        if self.step_id >= 200:
+        if self.step_id == 666:
             self.step_id = 0
             self.engine.snakes = [None, None, None, None]
+            self.engine.psnakes = [None, None, None, None]
             pass
         if self.step_id == 0:
@@ -64 +65 @@
     def run (self):
         """Run the game with 150 ms step"""
+        if self.step_id == 666:
+            self.step_id = 0
+            for i, snake in enumerate(self.engine.psnakes):
+                self.engine.snakes[i] = snake
+                self.engine.create_snake(i, snake)
         if self.dead_snake_check() == False:
             return
@@ -106 +112 @@
             self.canvas.after_cancel(self.after_id)
             pass
-        self.step_id = 201
+        self.step_id = 666
         root = tk.Tk()
         end_label = tk.Label(root, text="End")