Как перезагрузить функцию модуля в Python?

на этот вопрос относительно перезагрузки модуля, как перезагрузить определенную функцию из измененного модуля?

псевдо-код:

from foo import bar

if foo.py has changed:
    reload bar

7 ответов


то, что вы хотите, возможно, но требует перезагрузки двух вещей... первый reload(foo), но тогда вы также должны reload(baz) (если baz - имя модуля, содержащего from foo import bar заявления).

, почему... Когда foo сначала загружается, a foo создается объект, содержащий


горячая перезагрузка-это не то, что вы можете сделать в Python надежно, не взрывая голову. Вы буквально не можете поддерживать перезагрузку без написания кода специальными способами, а попытка написать и поддерживать код, который поддерживает перезагрузку с любым здравомыслием, требует крайней дисциплины и слишком запутан, чтобы стоить усилий. Тестирование такого кода тоже непростая задача.

решение состоит в том, чтобы полностью перезапустить процесс Python, когда код изменился. Это возможно сделать плавно, но как зависит от конкретной проблемной области.


на сегодняшний день, правильный способ сделать это:

import sys, importlib
importlib.reload(sys.modules['foo'])
from foo import bar

протестировано на python 2.7, 3.5, 3.6.


в то время как перезагрузка функций не является особенностью reload функция, это все еще возможно. Я бы не рекомендовал делать это в производстве, но вот как это работает: Функции, которую вы хотите заменить объект где-то в памяти, и ваш код может содержать много ссылок на него (а не имя функции). Но то, что эта функция фактически делает при вызове, сохраняется не в этом объекте, а в другом объекте, на который, в свою очередь, ссылается объект функции, в его атрибуте __code__. Пока у вас есть ссылка на функцию, вы можете обновить ее код:

модуль mymod:

from __future__ import print_function
def foo():
    print("foo")

другой модуль / python сессии:

>>> import mymod
>>> mymod.foo()
foo
>>> old_foo_ref = mymod.foo
>>> # edit mymod.py to make function "foo" print "bar"
>>> reload(mymod)
<module 'mymod' from 'mymod.py'>
>>> old_foo_ref()   # old reference is running old code
foo
>>> mymod.foo()     # reloaded module has new code
bar
>>> old_foo_ref.__code__ = mymod.foo.__code__
>>> old_foo_ref()   # now the old function object is also running the new code
bar
>>> 

теперь, если у вас нет ссылки на старую функцию (т. е. a lambda передано в другую функцию), вы можете попытаться получить его с помощью gc модуль, путем поиска списка всех объектов:

>>> def _apply(f, x):
...     return lambda: f(x)
... 
>>> tmp = _apply(lambda x: x+1, 1)  # now there is a lambda we don't have a reference to
>>> tmp()
2
>>> import gc
>>> lambdas = [obj for obj in gc.get_objects() if getattr(obj,'__name__','') == '<lambda>'] # get all lambdas
[<function <lambda> at 0x7f315bf02f50>, <function <lambda> at 0x7f315bf12050>]     
# i guess the first one is the one i passed in, and the second one is the one returned by _apply
# lets make sure:
>>> lambdas[0].__code__.co_argcount
1  # so this is the "lambda x: ..."
>>> lambdas[1].__code__.co_argcount
0  # and this is the "lambda: ..."
>>> lambdas[0].__code__ = (lambda x: x+2).__code__  # change lambda to return arg + 2
>>> tmp()
3  # 
>>> 

вы не можете перезагрузить метод из модуля, но вы можете загрузить модуль снова с новым именем, скажем foo2 и сказал bar = foo2.bar для перезаписи текущей ссылки.

обратите внимание, что если bar имеет какие-либо зависимости от других вещей в foo или любые другие побочные эффекты, вы получите в неприятности. Поэтому, пока он работает, он работает только для самых простых случаев.


вам придется использовать reload перезагрузить модуль, так как вы не можете просто перезагрузить функции:

>>> import sys
>>> reload(sys)
<module 'sys' (built-in)>
>>>

Если вы являетесь автором foo.py, вы можете написать:

with open('bar.py') as f: bar_code=compile(f.read(),'bar.py','exec')

def bar(a,b):
    exec(bar_code)

def reload_bar():
    global bar_code
    with open('bar.py') as f: bar_code=compile(f.read(),'bar.py','exec') 

затем, в вашем псевдокоде, перезагрузите if bar.py изменилось. Этот подход особенно хорош, когда bar живет в том же модуле, что и код, который хочет его перезагрузить, а не в случае OP, где он живет в другом модуле.