-
Notifications
You must be signed in to change notification settings - Fork 0
/
flora.py
32 lines (24 loc) · 870 Bytes
/
flora.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import random
from food import Food
from consts import Colors
class Flora(object):
def __init__(self, game, food_count=5):
self.food_list = {}
self.game = game
self.generate_food(food_count)
def generate_food(self, quantity=1):
for _ in range(quantity):
food = Food(self.game.random_vector(), self._random_color())
self.food_list[food.position] = food
def remove_food(self, xyVector):
food = self.food_list.get(xyVector)
if food is not None:
del self.food_list[food.position]
return food
def is_food_here(self, xyVector):
return xyVector in self.food_list
def draw(self):
for food in self.food_list.values():
food.draw(self.game)
def _random_color(self):
return random.choice([Colors.BROWN, Colors.CHOCOLATE])