mastosnake/snake.py

81 lines
2.7 KiB
Python
Raw Normal View History

2021-05-24 00:37:28 -05:00
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))
2021-05-24 00:37:28 -05:00
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"
2021-05-24 22:51:46 -05:00
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
2021-05-24 00:37:28 -05:00
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')
2021-05-24 00:37:28 -05:00
else:
current.append('_')
return tmp