defuse mechanic added'

Engine/Board.py

@@ -1,5 +1,6 @@
 import pygame
 from Constants import ROWS, COLS, SQUARE_SIZE, GREEN, BLACK, WIDTH
+from Engine.Bomb import Bomb
 
 
 class Board:
@@ -31,3 +32,9 @@ class Board:
                 rect = image.get_rect()
                 rect.center = (key.getX() * SQUARE_SIZE + SQUARE_SIZE / 2, key.getY() * SQUARE_SIZE + SQUARE_SIZE / 2)
                 self.win.blit(image, rect)
+
+    def getBomb(self, point):
+        if point in self.bombMap:
+            return self.bombMap[point]
+        return Bomb(0,0,"")
+

Engine/Game.py

@@ -24,6 +24,7 @@ class Game:
         self.goingDown = True
 
     def update(self):
+        self.agent.defuse(self.board.getBomb(self.agent.getPoint()))
         self.board.drawSquares(self.win)
         self.board.drawBombs()
         self.board.drawAgent(self.win, self.agent)
@@ -46,11 +47,15 @@ class Game:
         self.agent.point = point
 
     def randomizeObject(self):
-        for x in range(10):
+        i = 0
+        while i < 11:
             point = Point(random.randint(0, 7), random.randint(0, 7))
+            if(point.getX() == 0 and point.getY() == 0):
+                continue;
             if point not in self.board.bombMap:
                 object = self.pickObject(random.randint(0, 4))
                 self.board.bombMap[point] = object
+                i += 1
 
     def pickObject(self, rand):
         if rand == 0: