source: branches/UQ/funcIO.py @ 855

#!/usr/bin/env python
#
# Author: Mike McKerns (mmckerns @caltech and @uqfoundation)
# Copyright (c) 2009-2016 California Institute of Technology.
# License: 3-clause BSD.  The full license text is available at:
#  - http://mmckerns.github.io/project/mystic/browser/mystic/LICENSE

def parse_from_history(object):
    import readline, inspect
    lbuf = readline.get_current_history_length()
    code = [readline.get_history_item(i)+'\n' for i in range(1,lbuf)]
    lnum = 0
    codeblocks = []
    while lnum < len(code)-1:
       if code[lnum].startswith('def'):    
           block = inspect.getblock(code[lnum:])
           lnum += len(block)
           if block[0].startswith('def %s' % object.func_name):
               codeblocks.append(block)
       else:
           lnum +=1
    return codeblocks


def src(object):
    """
This is designed to work on simple functions (not general callables.)
The advantage is that it will work on functions defined interactively.
    """
    import inspect
    # no try/except (like the normal src function)
    if hasattr(object,'func_code') and object.func_code.co_filename == '<stdin>':
        # function is typed in at the python shell
        lines = parse_from_history(object)[-1]
    else:
        lines, lnum = inspect.getsourcelines(object)
    return ''.join(lines)


HOLD = []
def tmp_funcdump(func):
    """ write func source to a NamedTemporaryFile (instead of pickle.dump)
with a last line 'FUNC = <function>' to be included as module.FUNC
hence you can arbitrarily import and call the funciton w/o knowing the name

NOTE: HOLD the return value for as long as you want your file to exist
    """
    import tempfile
    file = tempfile.NamedTemporaryFile(suffix='.py', dir='.')
    file.write(''.join(src(func)))
    file.write('FUNC = %s\n' % func.func_name)
    file.flush()
    return file


def func_load(modulename):
    """ read function from a 'special' source file that has a line
'FUNC = <function>' included as module.FUNC
    """
    module = __import__(modulename)
    return module.FUNC


if __name__ == '__main__':
  pass


# EOF