用python写一个贪吃蛇小游戏

import random
import curses

# 初始化窗口
stdscr = curses.initscr()
curses.curs_set(0)
sh, sw = stdscr.getmaxyx()
w = curses.newwin(sh, sw, 0, 0)
w.keypad(1)
w.timeout(100)

# 初始化蛇
snake = [[sh//2, sw//2]]
food = [sh//2, sw//2]
key = curses.KEY_RIGHT

# 游戏逻辑
while True:
    next_key = w.getch()
    key = key if next_key == -1 else next_key

    # 计算蛇头的下一个位置
    new_head = [snake[0][0], snake[0][1]]

    # 根据按键更新蛇头位置
    if key == curses.KEY_DOWN:
        new_head[0] += 1
    elif key == curses.KEY_UP:
        new_head[0] -= 1
    elif key == curses.KEY_LEFT:
        new_head[1] -= 1
    elif key == curses.KEY_RIGHT:
        new_head[1] += 1

    # 判断是否吃到食物
    if snake[0] == food:
        food = None
        while food is None:
            nf = [
                random.randint(1, sh-1),
                random.randint(1, sw-1)
            ]
            food = nf if nf not in snake else None
        w.addch(food[0], food[1], curses.ACS_PI)
    else:
        # 移动蛇尾，实现蛇的移动
        tail = snake.pop()
        w.addch(tail[0], tail[1], ' ')

    # 移动蛇头
    snake.insert(0, new_head)
    w.addch(snake[0][0], snake[0][1], curses.ACS_CKBOARD)

    # 游戏结束条件
    if (
        snake[0][0] in [0, sh] or
        snake[0][1] in [0, sw] or
        snake[0] in snake[1:]
    ):
        curses.endwin()
        quit()