41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from .base_agent import BaseAgent
|
|
class AutoProductionAgent(BaseAgent):
|
|
"""
|
|
Automaticaly produces commodity if in business inventory
|
|
"""
|
|
|
|
|
|
def __init__(self,sim,business,worker=1,employment_rate=0) -> None:
|
|
super().__init__(sim)
|
|
self.business=business
|
|
self.prod=business.production
|
|
self.worker=worker
|
|
self.employment_rate=employment_rate
|
|
self.employment_index=worker
|
|
|
|
def set_worker(self,workers):
|
|
if workers>1:
|
|
self.worker=workers
|
|
else:
|
|
self.worker=1
|
|
|
|
def can_produce(self):
|
|
# If can produce item
|
|
for k,cost in self.prod["craft"].items():
|
|
if cost > self.business.inventory[k]:
|
|
return False
|
|
return True
|
|
|
|
def tick(self,step,epi):
|
|
for i in range(self.worker):
|
|
if not self.can_produce():
|
|
self.employment_index-=self.employment_rate
|
|
continue
|
|
self.employment_index+=self.employment_rate
|
|
self.set_worker(int(self.employment_index))
|
|
# remove cost from inventory
|
|
for k,cost in self.prod["craft"].items():
|
|
self.business.inventory[k]-=cost
|
|
# add commodity
|
|
self.business.inventory[self.prod['name']]+=self.prod["amount"]
|