import sys import secrets from src.graphics import * from src.rabbit import Rabbit from src.waiter import * from src.decisionTree import * if __name__ == "__main__": # SETUP pygame.init() clock = pygame.time.Clock() graphics = Graphics() waiter = Waiter(graphics) rabbit = Rabbit() # init functions graphics.drawBackground(waiter.matrix) graphics.update(waiter) goal = None path = '' while True: for event in pygame.event.get(): # rabbit.check(waiter.matrix, waiter.X, waiter.Y) if event.type == pygame.QUIT: pygame.quit() sys.exit() break if event.type == pygame.KEYDOWN: if event.key == pygame.K_ESCAPE: pygame.quit() sys.exit() break if event.key == pygame.K_s: tree.TasksList('check', [10,3]) tree.TasksList('eat', [5,12]) tree.TasksList('order', [12,4]) tree.TasksList('goToBar', [10,10]) tree.TasksList('check', [3,4]) tree.TasksList('eat', [10,5]) tree.print() if event.key == pygame.K_r: temp = False while not temp: x = secrets.randbelow(graphics.width) y = secrets.randbelow(graphics.height) if waiter.matrix.matrix[x][y].walk_through == 1: temp = True goal = (x, y) path = waiter.findPath(goal) path = waiter.translatePath(path) if path != '': nextStep = path[0] path = path[1:] waiter.travel(nextStep, graphics) pygame.display.flip() clock.tick(graphics.fps)