81 lines
2.7 KiB
Python
81 lines
2.7 KiB
Python
import random
|
|
|
|
class Snake:
|
|
def __init__(self, bx=5, by=5):
|
|
self.snake = [
|
|
(2, 2),
|
|
(2, 1),
|
|
(2, 0)
|
|
]
|
|
self.gamestate = "playing"
|
|
self.boarddimensions = (bx, by)
|
|
self.directions = [
|
|
(-1, 0, "left"),
|
|
(0, -1, "down"),
|
|
(1, 0, "right"),
|
|
(0, 1, "up")
|
|
]
|
|
self.apple = (random.randrange(0, self.boarddimensions[0]),
|
|
random.randrange(0, self.boarddimensions[0]))
|
|
def moveset(self):
|
|
tmp = []
|
|
for i, j, text in self.directions:
|
|
tmp.append(
|
|
(self.snake[0][0] + i,
|
|
self.snake[0][1] + j,
|
|
text)
|
|
)
|
|
|
|
for i, j, text in tmp:
|
|
if (i, j) == self.snake[1]:
|
|
tmp.remove((i, j, text))
|
|
|
|
for i, j, text in tmp:
|
|
if i < 0 or i > self.boarddimensions[0] - 1:
|
|
tmp.remove((i, j, text))
|
|
break
|
|
if j < 0 or j > self.boarddimensions[1] - 1:
|
|
tmp.remove((i, j, text))
|
|
break
|
|
return tmp
|
|
def nodir(self):
|
|
tmp = []
|
|
for i, j, _ in self.moveset():
|
|
tmp.append((i, j))
|
|
return tmp
|
|
def move(self, direction):
|
|
for i in self.directions:
|
|
if direction == i[2]:
|
|
considered = (self.snake[0][0] + i[0],
|
|
self.snake[0][1] + i[1])
|
|
if considered in self.snake:
|
|
self.gamestate = "gameover"
|
|
print("gameover!")
|
|
if considered in self.nodir():
|
|
self.snake.insert(0, considered)
|
|
if self.apple != considered:
|
|
self.snake.pop()
|
|
else:
|
|
c = (random.randrange(0, self.boarddimensions[0] - 1), random.randrange(0, self.boarddimensions[1] - 1))
|
|
while c in self.snake:
|
|
c = (random.randrange(0, self.boarddimensions[0] - 1), random.randrange(0, self.boarddimensions[1] - 1))
|
|
self.apple = c
|
|
|
|
def render(self):
|
|
tmp = []
|
|
for j in range(self.boarddimensions[0] - 1, -1, -1):
|
|
tmp.append([])
|
|
current = tmp[-1]
|
|
for i in range(0, self.boarddimensions[1]):
|
|
if (i, j) in self.snake:
|
|
if (i, j) == self.snake[0]:
|
|
current.append('X')
|
|
else:
|
|
current.append('x')
|
|
elif (i, j) == self.apple:
|
|
current.append('a')
|
|
else:
|
|
current.append('_')
|
|
return tmp
|
|
|