__init__.py 文件源码

def register(action_klass):
    """
    Registers a chaos action class with Action class
    Args:
        action_klass:

    Returns:

    """
    def _register(chaos_klass):
        if chaos_klass.enabled:
            name = chaos_klass.__name__
            if isinstance(chaos_klass, (type, types.ClassType)):
                chaos_klass = chaos_klass()
            setattr(action_klass, name, chaos_klass)
            action_klass._actions.update([(name.upper(), chaos_klass)])
            return chaos_klass
    return _register