snake: af59540d48a9 main.py

snake

view main.py @ 165:af59540d48a9

UI.start() function added (still need to check for bags) UI.next_round() function added - passes

author	Alex Martynov
date	Thu, 23 Dec 2010 20:00:59 +0300
parents	de9d0b9071da
children	f99b094a3d5d

line source

1 import Tkinter as tk

2 import tkFileDialog as tkfd

3 import engine

4 import snake

8 class UI(object):

9 """User Interface:

11 Atributes:

13 - 'root' - root Window game placed at

14 - 'engine' - engine of the game

15 - 'canvas' - Widget field is pictured at

16 - 'step_id' - current step of the game

17 - 'after_id' - identificator of runing game process

18 - 'step_legth' - fime of the step"""

19 def __init__ (self):

20 """Create Python Battle game window.

21 Initialyze engige of the game."""

22 self.root = tk.Tk()

23 self.root.title("Python Battle")

24 self.canvas = tk.Canvas(self.root, background = "black")

25 self.canvas.pack(side ="top", fill="both", expand="yes")

26 self.buttons_pack(self.root).pack(side ="bottom", fill="both", expand="no")

27 self.step_id = 0

28 self.engine = engine.Engine(self.canvas)

29 self.after_id = None

30 self.step_length = 150

31 return

33 def buttons_pack(self, root):

34 """Packing the buttons in root frame.

35 Definition of button functions."""

36 buttons = tk.Frame(root)

37 load_1 = tk.Button(buttons, text="Load 1", command=lambda: self.load(0))

38 load_1.grid(row=1, column=2, stick="news")

39 load_2 = tk.Button(buttons, text="Load 2", command=lambda: self.load(1))

40 load_2.grid(row=2, column=3, stick="news")

41 run_b = tk.Button(buttons, text="Run", command=lambda: self.start())

42 run_b.grid(row=2, column=2, stick="news")

43 restart_b = tk.Button(buttons, text="Restart", command=lambda: self.restart())

44 restart_b.grid(row=1, column=5, stick="news")

45 load_3 = tk.Button(buttons, text="Load 3", command=lambda: self.load(2))

46 load_3.grid(row=3, column=2, stick="news")

47 load_4 = tk.Button(buttons, text="Load 4", command=lambda: self.load(3))

48 load_4.grid(row=2, column=1, stick="news")

49 step_b = tk.Button(buttons, text="Step", command=lambda: self.step())

50 step_b.grid(row=2, column=5, stick="news")

51 end_b = tk.Button(buttons, text="End", command=lambda: self.end())

52 end_b.grid(row=3, column=5, stick="news")

53 for column in range(1, 6):

54 buttons.grid_columnconfigure(column, weight=1)

55 return buttons

57 def load (self, snake_number):

58 """Ask for snake file loading.

59 Initialyzing snake and draw it on the field.

60 Return field back to default (without snakes) after end of the game."""

61 if self.step_id == 666:

62 self.step_id = 0

63 self.engine.snakes = [None, None, None, None]

64 pass

65 if self.step_id == 0:

66 file = tkfd.askopenfile(title="Open file")

67 if file == None:

68 return

69 snake = self.engine.create_snake(snake_number)

70 snake.load(file)

71 pass

72 self.engine.refill()

73 self.engine.redraw()

74 return

76 def start (self):

77 """Init running of the game."""

78 if self.step_id == 666:

79 self.next_round()

80 if self.dead_snake_check() == False:

81 return

82 if self.step_id > 200:

83 self.end()

84 return

85 self.engine.psnakes = self.engine.snakes[:]

86 self.run()

88 def run (self):

89 """Run the game with 'step_length' ms step

90 After the end of the game - restarts it with snakes survived in

91 previous game"""

92 self.step_id = self.step_id+1

93 self.engine.step()

94 self.after_id = self.canvas.after(self.step_length, self.run)

95 return

97 def step (self):

98 """Do the next game step"""

99 if self.dead_snake_check() == False:

100 return

101 if self.step_id <= 200:

102 if self.after_id != None:

103 self.canvas.after_cancel(self.after_id)

104 pass

105 self.step_id = self.step_id+1

106 self.engine.step()

107 pass

108 else:

109 self.end()

110 pass

111 return

112

113 def dead_snake_check(self):

114 """Check the number of snakes alive.

115 End the game if alive snake number is less than two."""

116 dead_snakes = 0

117 for snake in self.engine.snakes:

118 if snake == None:

119 dead_snakes=dead_snakes+1

120 pass

121 if dead_snakes >= 3:

122 self.end()

123 return False

124

125 def restart(self):

126 """"Restarts the game after the end of the game with snakes survived"""

127 self.step_id = 0

128 for i, snake in enumerate(self.engine.psnakes):

129 if self.engine.psnakes[i] != None:

130 self.engine.snakes[i] = snake

131 self.engine.create_snake(i, snake)

132 self.engine.refill()

133 self.engine.redraw()

134

135 def next_round(self):

136 pass

137

138 def end (self):

139 """End the game and raise the window that tels about it."""

140 if self.after_id != None:

141 self.canvas.after_cancel(self.after_id)

142 pass

143 self.step_id = 666

144 root = tk.Tk()

145 end_label = tk.Label(root, text="End")

146 end_label.pack()

147 root.mainloop()

148 pass

149

150 if __name__ == "__main__":

151 snake_batle = UI()

152 snake_batle.root.mainloop()