source: fedd/federation/experiment_info.py @ 9294673

#!/usr/local/bin/python

import copy

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 = { }

    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 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