腌制功能并包括进口

Question

My idea is this, so I want to allow users to send code to a web endpoint.

I would like to grab the modules/imports from the method. Is that possible?

import pickle
import inspect

def add(x,y):
    return x+y

def stringify_method(func):
    """"""
    return pickle.dumps(func)

print(pickle.loads(stringify_method(add))(1,2))
3

So it returns 3, which is expected.

Now let's say I have something more complicated:

import sys
import pickle
import inspect
import arcpy
import pandas as pd
import numpy as np

def create_array(array):
    return np.array(array)

def unpickle_method(func):
    """"""
    return pickle.dumps(func)

print(pickle.loads(stringify_method(create_array))([1,2,3,4]))

I get the method just fine, but the modules do not follow. How do I get my import numpy as np and pandas, etc..?

Answer 1

Not exactly sure what you are trying to do, but is this helpful?

>>> import sys
>>> import numpy as np
>>> import inspect
>>> 
>>> [x[0] for x in inspect.getmembers(sys.modules[__name__], inspect.ismodule)]
['__builtins__', 'inspect', 'np', 'sys']

this gets me 1/2 way there, I need to be able to then grab the import statements from the method: import numpy as np

You could probably reconstruct the statements from this:

>>> [(x, y.__name__) for x,y in inspect.getmembers(sys.modules[__name__], inspect.ismodule)]
[('__builtins__', 'builtins'), ('inspect', 'inspect'), ('np', 'numpy'), ('sys', 'sys')]

(note the ('np', 'numpy') element which tells you import numpy as np )