Changes in main.py [26:a29fb8b305d7:25:2881ede732a7]

File:

• main.py (modified) (3 diffs)

main.py

@@ -3 +3 @@
 import engine
 
+class UI(object):
 
-
-class UI(object):
     def __init__ (self):
         root = tk.Tk()
@@ -11 +10 @@
         self.canvas = tk.Canvas(root, background = "black")
         self.canvas.pack(side ="top", fill="both", expand="yes")
-        buttons = tk.Frame(root)
+        buttons = self.create_buttons(root)
         buttons.pack(side ="bottom", fill="both", expand="yes")
-        self.buttons_pack(buttons)
         self.id = 0
         self.engine = engine.Engine(self.canvas)
         return
-    def buttons_pack(self, frame):
+
+    def create_buttons(self, root):
+        frame = tk.Frame(root)
         load_1 = tk.Button(frame, text="Load", command=self.load(1))
         load_1.pack(side="top", fill="both", expand = "yes")
@@ -30 +30 @@
         step_b = tk.Button(frame, text="Step", command=self.step())
         step_b.pack(side="bottom", fill="both", expand = "yes")
-        return
+        return frame
+
     def load (self, snake_number):
         file_name = tkfd.askopenfilename(title="Open file")
         snake_file = open(file_name,'r')
-        
+
+    def run (self):
         pass
 
-    def run (self):
+    def step (self):
+        pass
 
-        pass
-    def step (self):
-
-        pass
-