diff --git a/main.py b/main.py index cebd520..bdad40e 100644 --- a/main.py +++ b/main.py @@ -99,8 +99,18 @@ class Game: y = self.infantry_ork.y mob_image = self.infantry_ork.INFANTRY_ORK_IMG prediction = self.prediction_road(x,y,mob_image) - else: - self.bfs.bfs(self.bfs.get_cell_number(x,y)) + if prediction == "ORK_INFANTRY": + self.bfs.bfs(self.bfs.get_cell_number(x,y)) + x = self.archer_ork.x + y = self.archer_ork.y + mob_image = self.archer_ork.ARCHER_ORK_IMG + prediction = self.prediction_road(x,y,mob_image) + if prediction == "ORK_ARCHER": + self.bfs.bfs(self.bfs.get_cell_number(x,y)) + x = self.sauron.x + y = self.sauron.y + mob_image = self.sauron.SAURON_IMG + def prediction_road(self,x,y,mob_image):