async def _test_movement(self):
async with ClientSession() as session:
ws_url = f'{self.live_server_url}/{GAME_WS_URL}'
async with session.ws_connect(ws_url) as ws:
response = await ws.receive()
data = json.loads(response.data)
self.assertEqual(data['event'],
MazeConsumer.NEW_MAZE_EVENT,
msg='First event not new maze')
# create maze on data from response
maze = Maze(is_generate=False)
maze.matrix_walls = data['content']['maze']['matrix']
maze.x = data['content']['maze']['current_x']
maze.y = data['content']['maze']['current_y']
maze.max_depth_x = data['content']['maze']['current_x']
maze.max_depth_y = data['content']['maze']['max_depth_y']
# go to some free path direction
free_path = maze.get_access_paths()[0]
await ws.send_json({
'event': MazeConsumer.MOVE_EVENT,
'content': free_path
})
maze.go_to(free_path)
while True:
# possible events: TIMER_EVENT and NEW_COORD_EVENT
response = await ws.receive()
data = json.loads(response.data)
if data['event'] == MazeConsumer.TIMER_EVENT:
continue
elif data['event'] == MazeConsumer.NEW_COORD_EVENT:
self.assertEqual(
(maze.x, maze.y),
(data['content'][0], data['content'][1]),
msg='Bad coordinates')
break
else:
self.assertTrue(False, msg='Unexpected event')
def new_game(self, difficulty):
self.maze = Maze(MAZE_WIDTH, MAZE_HEIGHT, difficulty)
self.set_state('status', self.STARTING)
#TODO: base this off FPS, or actual clock time
self.starting_ticks = 180 - 1
class MainScene(Scene):
STARTING, PLAYING, DEAD, WIN = range(4)
def load(self):
# state variables
self.set_state('running', True)
self.font = pygame.font.Font(MENU_FONT, 64)
#TODO: it's pretty jumpy...
self.timer_fonts = [
pygame.font.Font(MENU_FONT, 128 + size*10)
for size in range(60)
]
self.dead_print = self.font.render("You're Dead, Bro", True, WHITE)
self.win_print = self.font.render("WINNING!!!", True, WHITE)
def setup(self, first_time, new_game=True, game_mode=Maze.EASY):
if self.get_state('status') != self.DEAD:
self.set_state('status', self.STARTING)
if new_game:
self.new_game(game_mode)
def new_game(self, difficulty):
self.maze = Maze(MAZE_WIDTH, MAZE_HEIGHT, difficulty)
self.set_state('status', self.STARTING)
#TODO: base this off FPS, or actual clock time
self.starting_ticks = 180 - 1
def render(self, screen):
screen.fill(BLACK)
self.maze.render(screen)
if self.get_state('status') == self.DEAD:
screen.blit(self.dead_print, center(self.dead_print))
elif self.get_state('status') == self.WIN:
screen.blit(self.win_print, center(self.win_print))
status = self.get_state('status')
if status == self.PLAYING:
self.maze.player.update()
elif status == self.STARTING:
self.do_starting(screen)
cur_pixel = screen.get_at(self.maze.player.current)
if cur_pixel == self.maze.wall_color:
self.set_state('status', self.DEAD)
elif cur_pixel == END_COLOR:
#TODO: there's a 1 in 17 million chance for this to falsely win because
# maze color matches winning marker color
self.set_state('status', self.WIN)
def do_starting(self, screen):
seconds = self.starting_ticks / 60 + 1
font = self.timer_fonts[(self.starting_ticks % 60)]
countdown = font.render(str(seconds), True, WHITE)
screen.blit(countdown, center(countdown))
#TODO: base this on FPS
self.starting_ticks -= 2
if self.starting_ticks <= 0:
self.set_state('status', self.PLAYING)
def do_event(self, event):
if event.type == pygame.KEYUP:
if event.key == pygame.K_ESCAPE:
self.manager.switch_scene('menu')
elif event.key == pygame.K_LEFT:
self.maze.player.change_dir('left')
elif event.key == pygame.K_RIGHT:
self.maze.player.change_dir('right')
elif event.key == pygame.K_UP:
self.maze.player.change_dir('up')
elif event.key == pygame.K_DOWN:
self.maze.player.change_dir('down')
elif event.key == pygame.K_q:
self.maze.DEBUG = not self.maze.DEBUG
elif event.key == pygame.K_w:
self.maze.player.speed += 1
elif event.key == pygame.K_d:
self.maze.player.speed -= 1
def possible(maze: Maze, location: Location,
visited: List[Location]) -> List['Move']:
neighbors = maze.neighbors(location)
return [Move(*n.sub(location)) for n in neighbors if n not in visited]