from nepi.execution.attribute import Attribute, Flags, Types
from nepi.execution.trace import Trace, TraceAttr
-from nepi.execution.resource import ResourceManager, clsinit, ResourceState, \
- ResourceAction
+from nepi.execution.resource import ResourceManager, clsinit_copy, \
+ ResourceState, ResourceAction
from nepi.util.sshfuncs import ProcStatus
import os
import tempfile
-@clsinit
+@clsinit_copy
class Collector(ResourceManager):
""" The collector is reponsible of collecting traces
of a same type associated to RMs into a local directory.
:type ec: ExperimentController
:param guid: guid of the RM
:type guid: int
- :param creds: Credentials to communicate with the rm (XmppClient)
- :type creds: dict
"""
_rtype = "Collector"
+ _help = "A Collector can be attached to a trace name on another " \
+ "ResourceManager and will retrieve and store the trace content " \
+ "in a local file at the end of the experiment"
+ _backend_type = "all"
@classmethod
def _register_attributes(cls):
@property
def store_path(self):
return self._store_path
-
- def provision(self):
+
+ def do_provision(self):
trace_name = self.get("traceName")
if not trace_name:
self.fail()
except OSError:
pass
- super(Collector, self).provision()
+ super(Collector, self).do_provision()
- def deploy(self):
- try:
- self.discover()
- self.provision()
- except:
- self.fail()
- raise
+ def do_deploy(self):
+ self.do_discover()
+ self.do_provision()
- super(Collector, self).deploy()
+ super(Collector, self).do_deploy()
- def release(self):
+ def do_release(self):
trace_name = self.get("traceName")
rename = self.get("rename") or trace_name
result = self.ec.trace(rm.guid, trace_name)
fpath = os.path.join(self.store_path, "%d.%s" % (rm.guid,
rename))
- f = open(fpath, "w")
- f.write(result)
- f.close()
-
- super(Collector, self).release()
+ try:
+ f = open(fpath, "w")
+ f.write(result)
+ f.close()
+ except:
+ msg = "Couldn't retrieve trace %s for %d at %s " % (trace_name,
+ rm.guid, fpath)
+ self.error(msg)
+ continue
+
+ super(Collector, self).do_release()
def valid_connection(self, guid):
# TODO: Validate!