1 # 这是一个示例 Python 脚本。

  2

  3 # 按 ⌃R 执行或将其替换为您的代码。

  4 # 按 双击 ⇧ 在所有地方搜索类、文件、工具窗口、操作和设置。

  5 import sys

  6 import pygame

  7 import random

  8

  9 game = None

 10 BOMB_COUNT = 1

 11

 12 # 空间

 13 class Button(object):

 14     pass

 15

 16 class Game(object):

 17     screen = None

 18     row_count = 0

 19     bomb_location = []

 20     squares_list = []

 21     squares_val_dict = {}

 22     win = False

 23

 24     def __init__(self, count):

 25         pygame.init()

 26         total_width = count * 18 + 100 * 2

 27         total_height = count * 18 + 100 * 2

 28         self.row_count = count

 29         self.screen = pygame.display.set_mode((total_width, total_height))

 30         self.win = False

 31         self.bomb_location = []

 32         self.squares_list = []

 33         self.squares_val_dict = {}

 34

 35     def __del__(self):

 36         print('del game')

 37         pygame.display.quit()

 38

 39     def is_over(self):

 40         return self.win

 41

 42     def get_screen(self):

 43         return self.screen

 44

 45     def random_bomb(self):

 46         # 随机产生炸弹

 47         for i in range(BOMB_COUNT):

 48             while 1:

 49                 x = random.randint(0, 29)

 50                 y = random.randint(0, 29)

 51                 if (x, y) in self.bomb_location:

 52                     continue

 53                 else:

 54                     self.bomb_location.append((x, y))

 55                 break

 56

 57         # 计算某个位置范围的数字

 58         for x in range(self.row_count):

 59             for y in range(self.row_count):

 60                 count = 0

 61                 if (x - 1, y) in self.bomb_location:

 62                     count += 1

 63                 if (x - 1, y - 1) in self.bomb_location:

 64                     count += 1

 65                 if (x - 1, y + 1) in self.bomb_location:

 66                     count += 1

 67

 68                 if (x , y - 1) in self.bomb_location:

 69                     count += 1

 70                 if (x , y) in self.bomb_location:

 71                     count += 1

 72                 if (x , y + 1) in self.bomb_location:

 73                     count += 1

 74

 75                 if (x + 1 , y - 1) in self.bomb_location:

 76                     count += 1

 77                 if (x + 1 , y) in self.bomb_location:

 78                     count += 1

 79                 if (x + 1, y + 1) in self.bomb_location:

 80                     count += 1

 81

 82                 # print('%s,%s %d' % (x, y, count))

 83                 self.squares_val_dict[(x, y)] = count

 84

 85     def init_square(self):

 86         for i in range(self.row_count):

 87             self.squares_list.append([])

 88             top = 100 + i * 18

 89             for j in range(self.row_count):

 90                 left = 100 + j * 18

 91                 width = 18

 92                 height = 18

 93                 exist = False

 94                 if (i, j) in self.bomb_location:

 95                     # print('%s,%s exists' % (j, i))

 96                     exist = True

 97

 98                 # 周围的炸弹数量

 99                 around_count = self.squares_val_dict.get((i, j), 0)

100                 # print('init square %s,%s %d' % (i, j, around_count))

101                 self.squares_list[i].append(Square(exist, around_count, self.screen, left, top, width, height))

102                 self.squares_list[i][j].draw()

103         pygame.display.update()

104

105     def start_game(self):

106         pass

107

108     def display_win(self):

109         font = pygame.font.SysFont("Andale Mono", 32)

110         txt = font.render("Winner Winner Winner", True, 'Red')

111         self.get_screen().blit(txt, (200, 0))

112

113         font = pygame.font.SysFont("Andale Mono", 16)

114         txt = font.render("uploading to dashboard...", True, 'green')

115         self.get_screen().blit(txt, (260, 40))

116

117         font = pygame.font.SysFont("Andale Mono", 16)

118         txt = font.render("click to continue...", True, 'gray')

119         self.get_screen().blit(txt, (280, 60))

120

121         self.win = True

122

123     def display_flag(self):

124         for (x, y) in self.bomb_location:

125             square = self.squares_list[x][y]

126             square.right_click()

127             square.draw()

128

129     # 根据所有的旗帜来判断胜利

130     def check_win_by_flag(self):

131         for (x, y) in self.bomb_location:

132             square = self.squares_list[x][y]

133             if square.state == 'flag' and square.exist:

134                 continue

135             return False

136         self.display_win()

137         return True

138

139     # 根据已经没有格子点击了来判断胜利

140     def check_win_by_click(self):

141         # print('check by click')

142         for x in range(self.row_count):

143             for y in range(self.row_count):

144                 square = self.squares_list[x][y]

145                 if square.state == 'blank' or square.exist:

146                     # print(1)

147                     continue

148                 return False

149         self.display_flag()

150         self.display_win()

151         return True

152

153     def right_clicked(self, pos):

154         left = pos[0]

155         top = pos[1]

156         x = int((top - 100) / 18)

157         y = int((left - 100) / 18)

158

159         # print('right clicked %s, %s' % (x, y))

160         if x in range(0, self.row_count) and y in range(0, self.row_count):

161             square = self.squares_list[x][y]

162             if not square.right_click():

163                 return

164             # 表示右键生效

165             square.draw()

166

167             if square.state == 'flag' and square.exist:

168                 # 只有当前标记是正确的时候才判断

169                 # 判断是否已经将所有的炸弹标记出来

170                 self.check_win_by_flag()

171             pygame.display.update()

172

173     def clicked(self, pos):

174         left = pos[0]

175         top = pos[1]

176         x = int((top - 100) / 18)

177         y = int((left - 100) / 18)

178

179         def click_square(self, x, y):

180             if x not in range(0, self.row_count) or y not in range(0, self.row_count):

181                 return False

182

183             square = self.squares_list[x][y]

184             if square.state != 'new':

185                 return False

186

187             if not square.click():

188                 return False

189

190             square.draw()

191             if square.around_count == 0:

192                 # print('around is 0')

193                 for (x1, y1) in [

194                     (x - 1, y), (x - 1, y - 1), (x - 1, y + 1),

195                     (x, y - 1), (x, y), (x, y + 1),

196                     (x + 1, y - 1), (x + 1, y), (x + 1, y + 1),

197

198                 ]:

199                     click_square(self, x1, y1)

200             return True

201

202         if x in range(0, self.row_count) and y in range(0, self.row_count):

203             if click_square(self, x, y):

204                 # 判断是否成功

205                 self.check_win_by_click()

206             pygame.display.update()

207

208     def refresh(self):

209         pygame.display.update()

210

211

212 class Square(object):

213     exist = False

214     surface = None

215     rect = None

216     state = '' # new, blank, flag, bomed

217     face = None

218     around_count = 0

219

220     def __init__(self, exist, around_count, surface, left, top, width, height):

221         self.rect = pygame.Rect(left, top, width, height)

222         self.exist = exist

223         self.surface = surface

224         self.state = 'new'

225         self.around_count = around_count

226         # print('%s' % (self.around_count))

227

228     def exists(self):

229         return self.exist

230

231     def draw(self):

232         global game

233         if self.state == 'new':

234             self.face = pygame.Surface((self.rect.width, self.rect.height))

235             self.face.fill('white')

236             game.get_screen().blit(self.face, (self.rect.left, self.rect.top))

237             pygame.draw.rect(self.surface, 'gray', self.rect, 1)

238

239         elif self.state == 'blank':

240             self.face.fill('gray')

241             game.get_screen().blit(self.face, (self.rect.left, self.rect.top))

242             pygame.draw.rect(self.surface, 'white', self.rect, 1)

243

244             # 在格子中间画上数字

245             font = pygame.font.SysFont("Andale Mono", 16)

246             txt = font.render("%s" % (self.around_count if self.around_count > 0 else ''), True, 'blue')

247             # print('%s, %s' % (txt.get_rect(), self.around_count))

248             game.get_screen().blit(txt, (self.rect.left + 4, self.rect.top))

249

250             pass

251         elif self.state == 'flag':

252             # 在格子中间画上 F

253             font = pygame.font.SysFont("Andale Mono", 16)

254             txt = font.render("F", True, 'green')

255             # print('%s, %s' % (txt.get_rect(), self.around_count))

256             game.get_screen().blit(txt, (self.rect.left + 4, self.rect.top))

257

258         elif self.state == 'boom':

259             self.face.fill('red')

260             game.get_screen().blit(self.face, (self.rect.left, self.rect.top))

261             pygame.draw.rect(self.surface, 'white', self.rect, 1)

262             pass

263

264     def click(self):

265         need_update = False

266         if self.state == 'new':

267             if self.exist:

268                 self.state = 'boom'

269                 need_update = True

270             else:

271                 self.state = 'blank'

272                 need_update = True

273         return need_update

274

275     def right_click(self):

276         need_update = False

277         if self.state == 'new':

278             self.state = 'flag'

279             need_update = True

280         elif self.state == 'flag':

281             self.state = 'new'

282             need_update = True

283         return need_update

284

285

286 def init_game(count, x=18, y=18):

287     global game

288     if game:

289         del game

290     game = Game(count)

291     game.random_bomb()

292     game.init_square()

293

294

295 # 按间距中的绿色按钮以运行脚本。

296 if __name__ == '__main__':

297     init_game(30)

298

299     while True:

300         for event in pygame.event.get():

301             if event.type == pygame.MOUSEBUTTONUP:

302                 if game.is_over():

303                     init_game(30)

304                     continue

305                 pos = event.pos

306                 if event.button == 1:

307                     game.clicked(pos)

308                 elif event.button == 3:

309                     game.right_clicked(pos)

310                 # 获取当前那个格子被点击了

311             if event.type == pygame.QUIT:

312                 sys.exit(0)

313             pygame.display.update()