main.py

import pygame
from grid import Grid, NodeState
from pathfind import find_nearest_route, edges


class Game:
  def __init__(self, rows=11, columns=11, start=(5, 5)):
    self.rows = rows
    self.cols = columns
    
    self.grid = Grid(rows, columns, start)
    
    self.player_turn = True

    self.screen = pygame.display.set_mode((800, 600))
    self.clock = pygame.time.Clock()
    self.run = True
    self.tick = 60

  def loop(self):
    while self.run:
      for event in pygame.event.get():
        if event.type == pygame.QUIT:
          self.run = False

        if self.player_turn:
          if event.type == pygame.MOUSEBUTTONDOWN:
            if self.grid.node_pressed(event.pos):
              self.player_turn = False

      if not self.player_turn:
        route = find_nearest_route(self.grid.get_raw(), self.grid.player_pos, self.cols, self.rows)

        if route == [] or route.path == []:
          self.run = False
          print('Player won!')
        else:
          self.grid.move_player(route.path[1][0], route.path[1][1])

          if self.grid.player_pos in edges(self.cols, self.rows):
            self.run = False
            print('AI won!')

          self.player_turn = True

      self.screen.fill((0, 0, 0))
      self.grid.draw_grid(self.screen)

      pygame.display.update()
      self.clock.tick(self.tick)

  pygame.quit()

if __name__ == '__main__':
  game = Game()
  game.loop()