How can I add behavior to all the methods in a module?

[Jesse Aldridge]

So I have the following code, which works. I have a log function that I use to redirect standard output from a function to a file. I then change the other functions in the module so that when they are called, they will pass themselves into this logging function. So the code below results in a file called "test.log" being created which contains the lines "foo" and "bar".

import sys
 
def log_function( function, *args ):  
  log = open( 'test.log', 'a' )        
  sys.stdout = log
 
  try:  
    return function( *args )
  finally:    
    sys.stdout = sys.__stdout__ 
    log.close()
 
 
def foo():
  print "foo"
 
def bar():
  print "bar"
 
foo_without_logging = foo
foo = lambda: log_function( foo_without_logging )
 
bar_without_logging = bar
bar = lambda: log_function( bar_without_logging )
 
foo()
bar()
 

What I want to do is somehow take all the functions in a given module and transform them so that they will pass themselves into the logging function. Something like this:

for function in current_module:
  function_without_logging = function
  function = lambda: log_function( function_without_logging )
 

Know what I mean? It's kind of hard to explain, so let me know if I need to clarify.

Thanks.