Python: динамически импортировать код модуля из строки с помощью importlib

Я хочу динамически импортировать модуль в Python (3.7), при этом код модуля определяется в строке.

Ниже приведен рабочий пример, в котором используется модуль imp, который устарел в пользу importlib (начиная с версии 3.4):

import imp

def import_code(code, name):
    # create blank module
    module = imp.new_module(name)
    # populate the module with code
    exec(code, module.__dict__)
    return module

code = """
def testFunc():
    print('spam!')
"""

m = import_code(code, 'test')
m.testFunc()

В документации Python указано, что вместо imp.new_module() следует использовать importlib.util.module_from_spec(). Однако, похоже, нет способа создать пустой объект модуля с помощью модуля importlib, как я мог бы с imp.

Как я могу использовать importlib вместо imp для достижения того же результата?


python-3.x python dynamic-import python-importlib
person Josh    schedule 29.04.2019    source источник

Ответы (2)


arrow_upward
7
arrow_downward

Вы можете просто создать экземпляр types.Module:

import types
mod = types.ModuleType("mod")

Затем вы можете заполнить его exec так же, как и вы:

exec(code, mod.__dict__)
mod.testFunc() # will print 'spam!'

Итак, ваш код будет выглядеть так:

import types

def import_code(code, name):
    # create blank module
    module = types.ModuleType(name)
    # populate the module with code
    exec(code, module.__dict__)
    return module

code = """
def testFunc():
    print('spam!')
"""

m = import_code(code, 'test')
m.testFunc()

Как прокомментировал @Error - Syntactical Remorse, вы должны иметь в виду, что exec в основном выполняет любой код, содержащийся в строке, которую вы ему даете, поэтому вы должны использовать его с особой осторожностью. По крайней мере, проверьте, что вам дали, но было бы хорошо использовать исключительно предопределенные строки.

person Right leg    schedule 29.04.2019
comment
Вы можете добавить оговорку о том, что он должен использовать только предварительно указанные str или проверять их перед вызовом метода exec. - person Error - Syntactical Remorse; 29.04.2019
comment
@Error-SyntacticalRemorse Вы правы, я добавил примечание. - person Right leg; 29.04.2019

arrow_upward
4
arrow_downward

Согласно документации Python module_from_spec()

importlib.util.module_from_spec (спецификация)

...

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

Вот что я придумал, чтобы загрузить модуль из исходного кода, расположенного в репозитории github. Это способ без записи файла на диск.

import requests
url = "https://github.com/udacity/deep-learning-v2-pytorch/raw/master/intro-to-pytorch/helper.py"
r = requests.get(url)

import importlib.util
spec = importlib.util.spec_from_loader('helper', loader=None, origin=url)
helper = importlib.util.module_from_spec(spec)
exec(r.content, helper.__dict__)

helper.view_classify() # executes function from github file
person LVitya    schedule 04.02.2020
comment
именно то, что я искал - person naisanza; 08.08.2020