Pygame绘制rect跟随另一个rect基于他们的坐标(即蛇游戏)

import pygame
import sys

pygame.init()

window_width = 800
window_height = 600
window = pygame.display.set_mode((window_width, window_height))
pygame.display.set_caption("Snake Game")

snake_x = 100
snake_y = 100
snake_width = 20
snake_height = 20

snake_speed = 5
snake_direction = "right"

def draw_snake():
    pygame.draw.rect(window, (0, 255, 0), (snake_x, snake_y, snake_width, snake_height))

def draw_food():
    # 绘制食物的代码

while True:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            pygame.quit()
            sys.exit()

    # 处理用户输入的代码

    # 更新蛇的位置和状态的代码

    # 绘制游戏界面的代码
    window.fill((0, 0, 0))
    draw_snake()
    draw_food()
    pygame.display.update()

if snake_direction == "right":
    snake_x += snake_speed
elif snake_direction == "left":
    snake_x -= snake_speed
elif snake_direction == "up":
    snake_y -= snake_speed
elif snake_direction == "down":
    snake_y += snake_speed

def draw_snake():
    pygame.draw.rect(window, (0, 255, 0), (snake_x, snake_y, snake_width, snake_height))