30 lines
862 B
Python
30 lines
862 B
Python
import random
|
|
|
|
from survival.generators.resource_type import ResourceType
|
|
|
|
|
|
class ResourceComponent:
|
|
def __init__(self, resource_type):
|
|
self.resource_type = resource_type
|
|
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
|