Machine_learning_2023/Interface/movement.py

import pygame
import sys


def movement_key_press(board_size, x, y, cat_x=None, cat_y=None):
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            sys.exit()
        if event.type == pygame.KEYDOWN:
            # go left
            if event.key == pygame.K_LEFT and x > 0:
                x -= 1

                # go right
            if event.key == pygame.K_RIGHT and x < board_size - 1:
                x += 1

                # go up
            if event.key == pygame.K_UP and y > 0:
                y -= 1

                # go down
            if event.key == pygame.K_DOWN and y < board_size - 1:
                y += 1
    return (x, y)