41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
import action
|
|
|
|
class Agent:
|
|
def __init__(self, name, stats,
|
|
idle_message="Idle",
|
|
action=None):
|
|
self.name = name
|
|
self.stats = stats
|
|
self.idle_message = idle_message
|
|
self.action = action
|
|
self.action_agent = None
|
|
def get_stat(self, stat_name):
|
|
return self.stats[stat_name]
|
|
def change_stat(self, stat_name, amt):
|
|
stat = self.stats[stat_name]
|
|
stat.change_index(amt)
|
|
def get_name(self):
|
|
return self.name
|
|
def get_action(self):
|
|
return self.action
|
|
def set_action(self, action):
|
|
self.action = action
|
|
def update_action(self, cur_time):
|
|
if self.action is not None:
|
|
if self.action.is_done(cur_time):
|
|
self.action = None
|
|
def is_idle(self):
|
|
return self.action is None
|
|
def get_idle_message(self):
|
|
return self.idle_message
|
|
def get_action_message(self):
|
|
if self.is_idle():
|
|
return self.idle_message
|
|
else:
|
|
return self.action.get_msg()
|
|
def set_action_agent(self, agent):
|
|
self.action_agent = agent
|
|
def get_action_agent(self):
|
|
return self.action_agent
|
|
|