Changeset 29:af9337dd3cf1

Timestamp:

12/19/10 15:43:53 (5 years ago)

Author:

Alex Martynov <martiran@òÀæ>

Branch:

default

Message:

corrected UI.load(), corrected UI.step(), UI.end(), root made self.root

File:

• main.py (modified) (3 diffs)

main.py

@@ -7 +7 @@
 class UI(object):
     def __init__ (self):
-        root = tk.Tk()
-        root.title("Python Battle")
-        self.canvas = tk.Canvas(root, background = "black")
+        self.root = tk.Tk()
+        self.root.title("Python Battle")
+        self.canvas = tk.Canvas(self.root, background = "black")
         self.canvas.pack(side ="top", fill="both", expand="yes")
-        buttons = tk.Frame(root)
+        buttons = tk.Frame(self.root)
         buttons.pack(side ="bottom", fill="both", expand="yes")
         self.buttons_pack(buttons)
@@ -31 +31 @@
         step_b.pack(side="bottom", fill="both", expand = "yes")
         return
+    
     def load (self, snake_number):
-        file_name = tkfd.askopenfilename(title="Open file")
-        snake = self.engine.create_snake(snake_number)
-        snake.load(file_name)
+        if self.step_id >= 200:
+            self.step_id = 0
+            pass
+        elif self.step_id == 0:
+            file_name = tkfd.askopenfilename(title="Open file")
+            snake = self.engine.create_snake(snake_number)
+            snake.load(file_name)
+            pass
+        else:
+            pass
         return
 
@@ -47 +55 @@
         return
     def step (self):
-        self.canvas.after_cancel(self.after_id)
-        self.id = self.id+1
-        self.engine.step()
-        pass
+        if self.id <= 200:
+            self.canvas.after_cancel(self.after_id)
+            self.id = self.id+1
+            self.engine.step()
+            pass
+        else:
+            self.end()
+            pass
+        return
 
     def end (self):
+        root = tk.Tk()
+        end_label = tk.Label(root, text="End")
+        end_label.pack()
+        root.mainloop()
         pass