fabric decorator

decorators.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

__author__ = 'Krzysztof Szarek'

from functools import wraps
from fabric.decorators import _wrap_as_new
from darwin.ec2 import _cloud_hosts


def _list_annotating_decorator(attribute, *values):
    def attach_list(func):
        @wraps(func)
        def inner_decorator(*args, **kwargs):
            return func(*args, **kwargs)
        _values = values
        if len(_values) == 1 and not isinstance(_values[0], basestring):
            _values = _values[0]
        setattr(inner_decorator, attribute, list(_values))
        inner_decorator = _wrap_as_new(func, inner_decorator)
        return inner_decorator
    return attach_list


def ec2hosts(running=True):
    """
    Wraps the decorated function
    """
    return _list_annotating_decorator('hosts', [_cloud_hosts(running)])