snake.py

import pygame
import time
import random

# Initializing pygame
pygame.init()

# Define Colors
white= (255,255,255)
black= (0,0,0)
red = (255,0,0)
orange = (255,165,0)

width = 800
height = 600
screen = pygame.display.set_mode((width,height))
pygame.display.set_caption("Snake Game")

clock = pygame.time.Clock()

snake_size = 10
snake_speed = 15

message_font = pygame.font.SysFont('ubuntu',30)
score_font = pygame.font.SysFont('ubuntu',25)

def print_score(score):
    text = score_font.render("Score: "+str(score),True,orange)
    screen.blit(text,[0,0])

def draw_snake(snake_size,snake_pixels):
    for pixel in snake_pixels:
        pygame.draw.rect(screen,white,[pixel[0],pixel[1],snake_size,snake_size])

def run_game():

    game_over = False
    game_close = False

    x = width/2
    y = height/2

    x_speed = 0
    y_speed = 0
    
    snake_pixels = []
    snake_length = 1

    food_x = round(random.randrange(0,width-snake_size)/10.0)*10.0
    food_y = round(random.randrange(0,height-snake_size)/10.0)*10.0


    while not game_over:

        while game_close:
            screen.fill(black)
            game_over_msg = message_font.render("Game Over !",True,red)
            quit_msg=message_font.render("Press 1 to Quit",True,red)
            restart_msg=message_font.render("Press 2 to Restart",True,red)
            screen.blit(game_over_msg,[300,200])
            screen.blit(quit_msg,[300,300])
            screen.blit(restart_msg,[300,400])
            print_score(snake_length-1)
            pygame.display.update()

            for event in pygame.event.get():
                if event.type == pygame.KEYDOWN:
                    if event.key == pygame.K_1:
                        game_over = True
                        game_close = False
                    if event.key ==pygame.K_2:
                        run_game()
                if event.type == pygame.QUIT:
                    game_over = True
                    game_close = False


        for event in pygame.event.get():
            if event.type == pygame.QUIT:
                game_over = True
            if event.type == pygame.KEYDOWN:
                if event.key == pygame.K_LEFT:
                    x_speed = -snake_size
                    y_speed = 0
                if event.key == pygame.K_RIGHT:
                    x_speed = snake_size
                    y_speed = 0
                if event.key == pygame.K_UP:
                    x_speed = 0
                    y_speed = -snake_size
                if event.key == pygame.K_DOWN:
                    x_speed = 0
                    y_speed = snake_size

        if x>= width or x<0 or y>=height or y<0:
            game_close = True

        x+=x_speed
        y+=y_speed

        screen.fill(black)
        pygame.draw.rect(screen,orange,[food_x,food_y,snake_size,snake_size])

        snake_pixels.append([x,y])

        if len(snake_pixels)> snake_length:
            del snake_pixels[0]

        for pixel in snake_pixels[:-1]:
            if pixel == [x,y]:
                game_close = True

        draw_snake(snake_size,snake_pixels)
        print_score(snake_length - 1)

        pygame.display.update()

        if x == food_x and y==food_y:
            food_x = round(random.randrange(0,width-snake_size)/10.0)*10.0
            food_y = round(random.randrange(0,height-snake_size)/10.0)*10.0
            snake_length+=1

        clock.tick(snake_speed)

    pygame.quit()
    quit()

run_game()