Source code for contrast.environment.snapshots

"""
Module which contains classes that gather information which
needs to be dumped to every scan.
"""


class EmptySnapshot(object):
    """
    Define the API.
    """
    def capture(self):
        """
        Returns a dict of label-value pairs.
        """
        return {}


class MotorSnapshot(EmptySnapshot):
    """
    Snapshot which consists of all motor values.
    """
    def capture(self):
        from contrast.motors import Motor
        dct = {}
        for m in Motor.getinstances():
            try:
                dct[m.name] = m.position()
            except:
                print('Could not take snapshot of motor %s' % m.name)
        return dct