Как перезагрузить функцию модуля в Python?
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, где он живет в другом модуле.