main.py

'''
Greedy Snake Game
'''
import pygame
import random
# Initialize the game
pygame.init()
# Set up the game window
window_width = 800
window_height = 600
window = pygame.display.set_mode((window_width, window_height))
pygame.display.set_caption("Greedy Snake Game")
# Define colors
BLACK = (0, 0, 0)
WHITE = (255, 255, 255)
GREEN = (0, 255, 0)
RED = (255, 0, 0)
# Define game variables
snake_size = 20
snake_speed = 10
clock = pygame.time.Clock()
# Define the Snake class
class Snake:
    def __init__(self):
        self.x = window_width // 2
        self.y = window_height // 2
        self.direction = "RIGHT"
        self.length = 1
        self.body = []
    def move(self):
        if self.direction == "UP":
            self.y -= snake_size
        elif self.direction == "DOWN":
            self.y += snake_size
        elif self.direction == "LEFT":
            self.x -= snake_size
        elif self.direction == "RIGHT":
            self.x += snake_size
    def draw(self):
        for part in self.body:
            pygame.draw.rect(window, GREEN, (part[0], part[1], snake_size, snake_size))
    def check_collision(self):
        if self.x < 0 or self.x >= window_width or self.y < 0 or self.y >= window_height:
            return True
        for part in self.body[1:]:
            if self.x == part[0] and self.y == part[1]:
                return True
        return False
# Define the Food class
class Food:
    def __init__(self):
        self.x = random.randint(0, (window_width - snake_size) // snake_size) * snake_size
        self.y = random.randint(0, (window_height - snake_size) // snake_size) * snake_size
    def draw(self):
        pygame.draw.rect(window, RED, (self.x, self.y, snake_size, snake_size))
# Initialize the snake and food
snake = Snake()
food = Food()
# Game loop
running = True
while running:
    # Handle events
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_UP and snake.direction != "DOWN":
                snake.direction = "UP"
            elif event.key == pygame.K_DOWN and snake.direction != "UP":
                snake.direction = "DOWN"
            elif event.key == pygame.K_LEFT and snake.direction != "RIGHT":
                snake.direction = "LEFT"
            elif event.key == pygame.K_RIGHT and snake.direction != "LEFT":
                snake.direction = "RIGHT"
    # Move the snake
    snake.move()
    # Check collision with food
    if snake.x == food.x and snake.y == food.y:
        snake.length += 1
        food = Food()
    # Update the snake's body
    snake.body.insert(0, (snake.x, snake.y))
    if len(snake.body) > snake.length:
        snake.body.pop()
    # Check collision with snake's body or boundaries
    if snake.check_collision():
        running = False
    # Clear the window
    window.fill(BLACK)
    # Draw the snake and food
    snake.draw()
    food.draw()
    # Update the display
    pygame.display.update()
    # Set the game speed
    clock.tick(snake_speed)
# Quit the game
pygame.quit()