From d1c404fe8eac3c743004a9a48a683e9361c8f7b3 Mon Sep 17 00:00:00 2001 From: Joel Kronqvist Date: Mon, 3 Nov 2025 23:10:04 +0200 Subject: fix: added typing --- main.py | 27 +++++++++++++++------------ 1 file changed, 15 insertions(+), 12 deletions(-) (limited to 'main.py') diff --git a/main.py b/main.py index a0659f3..688589c 100755 --- a/main.py +++ b/main.py @@ -15,18 +15,21 @@ running = True while running: - for event in pygame.event.get(): - if event.type == pygame.QUIT: - running = False - elif event.type == pygame.KEYDOWN: - if event.key == pygame.K_w and view.game.snake.heading != Vec.up.neg(): - view.nextControlDirection = Vec.up - elif event.key == pygame.K_a and view.game.snake.heading != Vec.left.neg(): - view.nextControlDirection = Vec.left - elif event.key == pygame.K_r and view.game.snake.heading != Vec.down.neg(): - view.nextControlDirection = Vec.down - elif event.key == pygame.K_s and view.game.snake.heading != Vec.right.neg(): - view.nextControlDirection = Vec.right + match view.game.snake: + case None: raise ValueError("no snake") + case _: + for event in pygame.event.get(): + if event.type == pygame.QUIT: + running = False + elif event.type == pygame.KEYDOWN: + if event.key == pygame.K_w and view.game.snake.heading != Vec.up.neg(): + view.nextControlDirection = Vec.up + elif event.key == pygame.K_a and view.game.snake.heading != Vec.left.neg(): + view.nextControlDirection = Vec.left + elif event.key == pygame.K_r and view.game.snake.heading != Vec.down.neg(): + view.nextControlDirection = Vec.down + elif event.key == pygame.K_s and view.game.snake.heading != Vec.right.neg(): + view.nextControlDirection = Vec.right if view.isRunning(): view.update(pygame.time.get_ticks()) -- cgit v1.2.3