class AgentHandler(object):
def __init__(self, node=None):
self._network_type = node.network_type
self._network_version = node.network_version
self._node_type = node.type
self._agent_type = node.agent.type
node_dict = node.__dict__
node_dict.update(
{
"worker_api": node.agent.worker_api,
"agent_id": str(node.agent.id),
"compose_file": node.get_compose_file_path(),
"k8s_config_file": os.path.join(
os.path.dirname(node.agent.config_file.path),
".kube/config",
)
if node.agent.config_file
else "",
}
)
self._worker_api = node.agent.worker_api
self._node = node
if self._agent_type == HostType.Docker.name.lower():
self._agent = DockerAgent(node_dict)
elif self._agent_type == HostType.Kubernetes.name.lower():
self._agent = KubernetesAgent(node_dict)
@property
def node(self):
return self._node
@node.setter
def node(self, value):
self._node = value
@property
def config(self):
return self._agent.generate_config()
def create_node(self):
self._agent.create()
return True
def delete_node(self):
self._agent.delete()
return True
class AgentHandler(object):
def __init__(self, node=None):
self._network_type = node.get("network_type")
self._network_version = node.get("network_version")
self._node_type = node.get("type")
self._agent_type = node.get("agent_type")
self._node = node
if self._agent_type == HostType.Docker.name.lower():
self._agent = DockerAgent(node)
elif self._agent_type == HostType.Kubernetes.name.lower():
self._agent = KubernetesAgent(node)
@property
def node(self):
return self._node
@node.setter
def node(self, value):
self._node = value
@property
def config(self):
return self._agent.generate_config()
def create(self, info):
try:
cid = self._agent.create(info)
if cid:
return cid
else:
return None
except Exception as e:
raise e
def delete(self):
self._agent.delete()
return True
def start(self):
self._agent.start()
return True
def stop(self):
self._agent.stop()
return True