Is there any way to execute a statement before each return statement in python function?

Question

For example i have this piece of code:

def example():
    a = 'goodbye'
    if True:
        print a
        return 1
    else:
        print a
        return 0

I would like to know if there is any possible solution to write once "print a" and execute it before each "return" statement automaticaly. So that if I add more return statements I wouldn't need to add anything, but "print a" would execute. Result would look like something:

def example():
    a = "goodbye"
    """ some code to implement print a """
    if True:
        return 1
    else:
        return 0

Each time there is return statement it still would print a.

I tried to google, but don't know how word query, since all results are about returning multiple values.

UPDATE: My question was answered, thanks to all of you.
Although wrapping functions are correct answer, but I have chosen answer by GingerPlusPlus who suggested to use try...finally for simplicity.

Answer 1

try .. finally:

def example():
    try:
       if True:
           return 1
       else:
           return 0
    finally:
        print 'goodbye'

>>> example()
goodbye
1

A finally clause is always executed before leaving the try statement, whether an exception has occurred or not. Docs

Answer 2

You can use a context. Initialize it with the value you want to print. Then print when context exit, i.e. upon return.

class PrinterOnContextExit():
    def __init__( self, a ): self.a = a
    def __enter__( self ): pass
    def __exit__( self, exc_type, exc_value, traceback ): print( self.a )

def example():
    a = 'goodbye'
    with PrinterOnContextExit( a ):
        if True:
            return 1
        else:
            return 0

Note that you cannot print the returned value this way. If you ever wanted to print the returned value, then you should use a decorator.

class PrintOnReturn():
    def __init__( self, a ): self.a = a
    def __call__( self, func ): return lambda *args, **kwargs: self.callFunc( func, *args, **kwargs )
    def callFunc( self, func, *args, **kwargs ): r = func( *args, **kwargs ); print( self.a, r ); return r

@PrintOnReturn( "hello" )
def example():
    if True:
        return 1
    else:
        return 0

This will print whatever string you passed to the decorator, followed by the value returned from the decorated function. Here hello 1.