AI-Project/survival/components/resource_component.py

30 lines
862 B
Python
Raw Normal View History

2021-05-24 17:20:08 +02:00
import random
from survival.generators.resource_type import ResourceType
2021-05-24 13:10:10 +02:00
class ResourceComponent:
def __init__(self, resource_type):
self.resource_type = resource_type
2021-05-24 17:20:08 +02:00
w, e, t = self.generate_attributes(resource_type)
self.weight = w
self.eatable = e
self.toughness = t
@staticmethod
def generate_attributes(resource_type):
if resource_type == ResourceType.WOOD:
weight = random.randint(10, 15)
eatable = False
toughness = random.randint(10, 15)
elif resource_type == ResourceType.WATER:
weight = random.randint(1, 2)
eatable = True
toughness = 0
else:
weight = random.randint(1, 7)
eatable = True
toughness = random.randint(2, 5)
return weight, eatable, toughness