Snake.py

# __Author__ __Lencof__
# Snake.py

import os, pygame

pygame.init()
clock = pygame.time.Clock()

def newfood():
    # just generates position of food
    if borders:
        cord = [random.randint(2,34), random.randint(2,34)]
    else:
        cord = [random.randint(1,35), random.randint(1,35)]
    if cord in snake:
        # if the food generates on the snakes body, return another food position
        return newfood()
    else:
        return cord
    
win = pygame.display.set_mode((360, 360))
pygame.display.set_caption('Snake')
running = True
# some initialisation
snake = [[2,10]]
food = [random.randint(1,36), random.randint(1,36)]
key = "right"
### Change this setting to apply borders ###
### it can be either True or False ###
borders = True

# main loop
while running:
    clock.tick(5)
    # get all events
    for event in pygame.event.get():
        if event.type == pygame.QUIT: 
            running = False #this quits the loop
        elif event.type == pygame.KEYDOWN:
            if event.key == pygame.K_LEFT or event.key == pygame.K_a:
                if key != "right":
                    key = "left"
            elif event.key == pygame.K_RIGHT or event.key == pygame.K_d:
                if key != "left":
                    key = "right"
            elif event.key == pygame.K_UP or event.key == pygame.K_w:
                if key != "down":
                    key = "up"
            elif event.key == pygame.K_DOWN or event.key == pygame.K_s:
                if key != "up":
                    key = "down"
    # get direction based on key events
    if key == "up":
        head = [snake[0][0], snake[0][1]-1]
    elif key == "down":
        head = [snake[0][0], snake[0][1]+1]
    elif key == "right":
        head = [snake[0][0]+1, snake[0][1]]
    elif key == "left":
        head = [snake[0][0]-1, snake[0][1]]
    else:
        break
    # handle border cases
    if borders:
        # If borders enabled,
        # If you crash into them, then gameover
        if head[0] not in range(2,36) or head[1] not in range(2,36):
            running = False
            print("Gameover")
        # Draw the borders
        pygame.draw.rect(win, (255,0,255),(0,0,10,360))
        pygame.draw.rect(win, (255,0,255),(0,0,360,10))
        pygame.draw.rect(win, (255,0,255),(350,0,10,360))
        pygame.draw.rect(win, (255,0,255),(0,350,360,10))
    else:
        # If borders are disabled,
        # The snake can pop from other side of the screen
        if head[0] < 2:
            head[0] = 36
        elif head[0] > 36:
            head[0] = 1
        if head[1] < 1:
            head[1] = 36
        elif head[1] > 36:
            head[1] = 1
            
    # If food is eaten, increase length and generate new food after eating
    if head == food:
        snake.append(snake[-1])
        food = newfood()
        
    # check wether it collides with itself
    if snake.count(head) >= 1:
        running = False
        print("Gameover")
    else:
        # else continue by moving a step further
        snake.insert(0, head)
        tail = snake.pop()
        # draw all the stuff
        pygame.draw.rect(win, (255,255,0),(food[0]*10-10, food[1]*10-10,10,10))
        pygame.draw.rect(win, (0,0,0),(tail[0]*10-10, tail[1]*10-10,10,10))
        pygame.draw.rect(win, (255,255,255),(head[0]*10-10, head[1]*10-10,10,10))
    pygame.display.update()#updat
pygame.quit()