source: fedd/federation/experiment_info.py @ 6e33086

#!/usr/local/bin/python

import copy
from datetime import datetime, timedelta
from numbers import Number

class allocation_info:
    """
    An allocation identified by allocationID.  It includes the local name
    of the testbed on which it was allocated and the uri on which to
    contact it.  In addition, the proofs of access are included.
    """
    def __init__(self, allocID, tb, uri, proof=None):
        self.allocID = allocID
        self.tb = tb
        self.uri = uri
        self.proof = proof
        self.attrs = { }

    def set_attribute(self, a, v):
        a = a.lower()
        if a not in self.attrs: self.attrs[a] = v

    def get_attribute(self, a, default=None):
        a = a.lower()
        return self.attrs.get(a, default)

class experiment_info:
    """
    Information about an experiment in the eperiment controller.  It includes
    the fedid and localnames, the identity the experiment acts as, current
    status, topology allocation logs and the allocations themselves.
    """
    def __init__(self, fedid, localname, identity=None):
        self.fedid = fedid
        self.localname = localname
        self.identity = identity
        self.top = None
        self.status = 'empty'
        self.log = []
        self.alloc = { }
        self.last_update = datetime.now()

    def add_allocation(self, a):
        self.alloc[a.tb] = a

    def get_allocation(self, tb):
        return self.alloc.get(tb,None)

    def get_all_allocations(self):
        return self.alloc.values()

    def updated(self):
        self.last_update = datetime.now()

    def older_than(self, secs=None, dt=None):
        """
        If the last update of this info was more than secs seconds ago, or
        before dt (a datetime), return True.  If both secs and dt or neither is
        given return False.  If the last update time is completelt unknown
        (which should never happen) return True.
        """
        if self.last_update is None:
            return True
        elif dt is None and isinstance(secs, Number): 
            return self.last_update + timedelta(seconds=secs) < datetime.now()
        elif secs is None and isinstance(dt, datetime):
            return self.last_update < dt
        else:
            return False


    def get_info(self):
        """
        Return the information as a dict suitable for a soap return value.
        This is an infoResponseType in fedd_types.xsd.
        """
        # The state may be massaged by the service function that called
        # get_info (e.g., encoded for XMLRPC transport) so copy structured
        # fields.
        rv = {
                'experimentID': [
                    { 'fedid': copy.deepcopy(self.fedid) },
                    {'localname': self.localname },
                ],
                'experimentStatus': self.status,
            }
        if self.log:
            rv['allocationLog'] = "".join(self.log)
        if self.identity is not None:
            rv['experimentAccess'] = {'X509': copy.deepcopy(self.identity)}
        if self.top is not None:
            rv['experimentdescription'] = \
                { 'topdldescription': self.top.to_dict() }
        return rv