How to add function return value

[HYRY]

I need to write functions that return locals() as follows,
def func1():
a = 1
return locals()

def func2():
b = 2
return locals()

Can I write a decorator that it can automately do this conversion

def func1()
a = 1

--->

def func1():
a = 1
return locals()

[Raymond Hettinger]

Re: How to add function return value

On May 30, 6:21 pm, HYRY <ruoyu0...@gmai l.comwrote:

Can I write a decorator that it can automately do this conversion
>
def func1()
    a = 1
>
--->
>
def func1():
    a = 1
    return locals()

Not sure why you would want to do this, but there are several ways.

1. Make bytecode hack decorator that transforms the final "return
None" into "return locals()". A recipe that shows the basic technique
is at: http://aspn.activestate.com/ASPN/Coo.../Recipe/277940

2. Retrieve the source using inspect.getsour celines(f). Then, append a
"return locals()" to the end of the function and run it through exec.

3. Try hacking a tracing/debugging utility.

4. Run the sourcefile through tokenize, make the appropriate
insertion, and then untokenize.

. . .

Raymond

[George Sakkis]

Re: How to add function return value

On May 30, 10:16 pm, Raymond Hettinger <pyt...@rcn.com wrote:

On May 30, 6:21 pm, HYRY <ruoyu0...@gmai l.comwrote:
>

Can I write a decorator that it can automately do this conversion

>

def func1()
    a = 1

>

--->

>

def func1():
    a = 1
    return locals()

>
Not sure why you would want to do this, but there are several ways.
>
1. Make bytecode hack decorator that transforms the final "return
None" into "return locals()".  A recipe that shows the basic technique
is at:http://aspn.activestate.com/ASPN/Coo.../Recipe/277940
>
2. Retrieve the source using inspect.getsour celines(f). Then, append a
"return locals()" to the end of the function and run it through exec.
>
3. Try hacking a tracing/debugging utility.
>
4. Run the sourcefile through tokenize, make the appropriate
insertion, and then untokenize.

Here's an illustration of (3):

import sys
import functools

def withlocals(f):
@functools.wrap s(f)
def wrapper(*args, **kwds):
f_locals = {}
def probe(frame, event, arg):
if event == 'return':
f_locals.update (frame.f_locals )
return probe
sys.settrace(pr obe)
try: res = f(*args,**kwds)
finally: sys.settrace(No ne)
return (res, f_locals)
return wrapper

# example

@withlocals
def foo(x, y=0, *args, **kwds):
a = max(x,y)
b = len(args)
c = min(kwds.values ())
return a+b+c

r,locs = foo(1,2,3,4,a=5 ,b=6)
print locs

George