snake.py

import curses
from curses import KEY_RIGHT, KEY_LEFT, KEY_UP, KEY_DOWN
from random import randint

# 封装后可以交给wrapper()
curses.initscr()
win = curses.newwin(20, 60, 0, 0)
win.keypad(1)
curses.noecho()
curses.curs_set(0)
win.border(0)
win.nodelay(1)

# Initializing values
key = KEY_RIGHT
score = 0

# Initial snake co-ordinates
snake = [[4, 10], [4, 9], [4, 8]]
# First food co-ordinates
food = [10, 20]

# Prints the food
win.addch(food[0], food[1], '*')

# While Esc key is not pressed
while key != 27:
    # 绘制边框
    win.border(0)
    # Printing 'Score' and
    win.addstr(0, 2, 'Score : ' + str(score) + ' ')
    # 'SNAKE' strings
    win.addstr(0, 27, ' SNAKE ')

    # Previous key pressed
    prevKey = key
    event = win.getch()
    # 无输入，非阻塞输入返回-1
    key = key if event == -1 else event

    # Increases the speed of Snake as its length increases
    # 非阻塞输入，无输入时返回-1
    win.timeout(round(150 - (len(snake)/5 + len(snake)/10) % 120))

    # If SPACE BAR is pressed, wait for another
    if key == ord(' '):
        # one (Pause/Resume)
        key = -1
        while key != ord(' '):
            key = win.getch()
        key = prevKey
        continue

    # If an invalid key is pressed
    if key not in [KEY_LEFT, KEY_RIGHT, KEY_UP, KEY_DOWN, 27]:
        key = prevKey

    # If snake crosses the boundaries, make it enter from the other side
    if snake[0][0] == 0:
        snake[0][0] = 18
    if snake[0][1] == 0:
        snake[0][1] = 58
    if snake[0][0] == 19:
        snake[0][0] = 1
    if snake[0][1] == 59:
        snake[0][1] = 1

    # Exit if snake crosses the boundaries (Uncomment to enable)
    # if snake[0][0] == 0 or snake[0][0] == 19 or snake[0][1] == 0 or snake[0][1] == 59: break

    # If snake runs over itself
    if snake[0] in snake[1:]:
        if key == prevKey:
            break

    # Calculates the new coordinates of the head of the snake. NOTE: len(snake) increases.
    # This is taken care of later at [1].
    snake.insert(0, [snake[0][0] + (key == KEY_DOWN and 1) + (key == KEY_UP and -1),
                     snake[0][1] + (key == KEY_LEFT and -1) + (key == KEY_RIGHT and 1)])

    # When snake eats the food
    if snake[0] == food:
        food = []
        score += 1
        while food == []:
            # Calculating next food's coordinates
            food = [randint(1, 18), randint(1, 58)]
            if food in snake:
                food = []
        win.addch(food[0], food[1], '*')
    else:
        # [1] If it does not eat the food, length decreases
        last = snake.pop()
        win.addch(last[0], last[1], ' ')
    win.addch(snake[0][0], snake[0][1], '#')

curses.endwin()
print("\nScore - " + str(score))
print("http://bitemelater.in\n")