Можно сделать декоратор, вроде такого:
import sys
def decorator(n):
def out_changer(f):
sys.stdout = open(n,«w+»)
def res_func(*args, **kwargs):
return f(*args, **kwargs)
return res_func
return out_changer
@decorator('testfile.txt')
def your_func(a):
for i in range(a):
print i
func(10)