Fabric decorator for build script interaction with TeamCity

teamcity_messages_decorator.py

from functools import wraps

def teamcity_messages(f):
    """
    Decorator for Build Script Interaction with TeamCity
    https://confluence.jetbrains.com/display/
    TCD9/Build+Script+Interaction+with+TeamCity
    """
    @wraps(f)
    def wrapper(*args, **kwds):
        msg = f.__name__.replace('_', ' ')
        print "##teamcity[blockOpened name='%s']" % msg
        print "##teamcity[progressStart '%s']" % msg
        fn = f(*args, **kwds)
        print "##teamcity[progressFinish '%s']" % msg
        print "##teamcity[blockClosed name='%s']" % msg
        return fn
    return wrapper
    
@task
@teamcity_messages
def actual_task():
    print 'actual_task'


@task
@teamcity_messages
def actual_task2():
    print 'actual_task2'


@task
def deploy():
    actual_task()
    actual_task2()