56 lines
1.7 KiB
Python
56 lines
1.7 KiB
Python
from settings import *
|
|
from point import Point
|
|
from food import Food
|
|
|
|
|
|
class Player:
|
|
def __init__(self) -> None:
|
|
self.direction = MAIN_DIRECTION
|
|
self.body = [Point(MAIN_X, MAIN_Y)]
|
|
|
|
self.food = Food(0, 0)
|
|
self.food.generate_new()
|
|
|
|
self.score = 1
|
|
|
|
def input(self) -> None:
|
|
if self.direction == D_UP:
|
|
self.body.append(Point(self.body[-1].x, self.body[-1].y - 1))
|
|
elif self.direction == D_DOWN:
|
|
self.body.append(Point(self.body[-1].x, self.body[-1].y + 1))
|
|
elif self.direction == D_LEFT:
|
|
self.body.append(Point(self.body[-1].x - 1, self.body[-1].y))
|
|
elif self.direction == D_RIGHT:
|
|
self.body.append(Point(self.body[-1].x + 1, self.body[-1].y))
|
|
|
|
def _check_collision(body: list[Point]) -> None:
|
|
for point in body:
|
|
if point.x < 0 or point.y < 0:
|
|
raise IndexError("Snake has collision with walls")
|
|
elif point.x >= WIDTH or point.y >= HEIGHT:
|
|
raise IndexError("Snake has collision with walls")
|
|
|
|
if len(set(body)) != len(body):
|
|
raise IndexError("Player has collision with self")
|
|
|
|
def update(self) -> None:
|
|
if self.body[-1] == self.food:
|
|
self.food.generate_new()
|
|
self.score += 1
|
|
else:
|
|
self.body.pop(0)
|
|
|
|
Player._check_collision(self.body)
|
|
|
|
def left(self) -> None:
|
|
self.direction = D_LEFT
|
|
|
|
def right(self) -> None:
|
|
self.direction = D_RIGHT
|
|
|
|
def up(self) -> None:
|
|
self.direction = D_UP
|
|
|
|
def down(self) -> None:
|
|
self.direction = D_DOWN
|