Преобразовать строку в объект класса Python?

Учитывая строку в качестве пользовательского ввода в функцию Python, я хотел бы извлечь из нее объект класса, если в текущем определенном пространстве имен есть класс с таким именем. По сути, я хочу реализацию для функции, которая даст такой результат:

class Foo:
    pass

str_to_class("Foo")
==> <class __main__.Foo at 0x69ba0>

Это вообще возможно?

Предупреждение : eval()может использоваться для выполнения произвольного кода Python. Вы никогда не должны использовать eval()с ненадежными строками. (См. Безопасность eval () Python для ненадежных строк? )

Это кажется самым простым.

>>> class Foo(object):
...     pass
... 
>>> eval("Foo")
<class '__main__.Foo'>

Это может сработать:

import sys

def str_to_class(classname):
    return getattr(sys.modules[__name__], classname)

Вы могли бы сделать что-то вроде:

globals()[class_name]

Вы хотите класс Baz, который живет в модуле foo.bar. Вы хотите использовать Python 2.7, так importlib.import_module()как это облегчит переход на Python 3:

import importlib

def class_for_name(module_name, class_name):
    # load the module, will raise ImportError if module cannot be loaded
    m = importlib.import_module(module_name)
    # get the class, will raise AttributeError if class cannot be found
    c = getattr(m, class_name)
    return c

С Python <2.7:

def class_for_name(module_name, class_name):
    # load the module, will raise ImportError if module cannot be loaded
    m = __import__(module_name, globals(), locals(), class_name)
    # get the class, will raise AttributeError if class cannot be found
    c = getattr(m, class_name)
    return c

Использование:

loaded_class = class_for_name('foo.bar', 'Baz')

import sys
import types

def str_to_class(field):
    try:
        identifier = getattr(sys.modules[__name__], field)
    except AttributeError:
        raise NameError("%s doesn't exist." % field)
    if isinstance(identifier, (types.ClassType, types.TypeType)):
        return identifier
    raise TypeError("%s is not a class." % field)

Это точно обрабатывает классы как старого, так и нового стиля.

Я смотрел на то, как Django справляется с этим

django.utils.module_loading имеет это

def import_string(dotted_path):
    """
    Import a dotted module path and return the attribute/class designated by the
    last name in the path. Raise ImportError if the import failed.
    """
    try:
        module_path, class_name = dotted_path.rsplit('.', 1)
    except ValueError:
        msg = "%s doesn't look like a module path" % dotted_path
        six.reraise(ImportError, ImportError(msg), sys.exc_info()[2])

    module = import_module(module_path)

    try:
        return getattr(module, class_name)
    except AttributeError:
        msg = 'Module "%s" does not define a "%s" attribute/class' % (
            module_path, class_name)
        six.reraise(ImportError, ImportError(msg), sys.exc_info()[2])

Вы можете использовать его как import_string("module_path.to.all.the.way.to.your_class")

Да, ты можешь это сделать. Предполагая, что ваши классы существуют в глобальном пространстве имен, что-то вроде этого сделает это:

import types

class Foo:
    pass

def str_to_class(s):
    if s in globals() and isinstance(globals()[s], types.ClassType):
            return globals()[s]
    return None

str_to_class('Foo')

==> <class __main__.Foo at 0x340808cc>

С точки зрения выполнения произвольного кода или нежелательных имен, переданных пользователем, у вас может быть список допустимых имен функций / классов, и, если входные данные соответствуют одному в списке, это eval'd.

PS: я знаю ... немного поздно ... но это для всех, кто столкнется с этим в будущем.

Использование importlib работало лучше для меня.

import importlib

importlib.import_module('accounting.views') 

При этом используется нотация с точкой для модуля Python, который вы хотите импортировать.

Если вы действительно хотите получить классы, которые вы делаете со строкой, вы должны сохранить (или правильно сформулировать, ссылку ) их в словаре. В конце концов, это также позволит назвать ваши классы на более высоком уровне и избежать выявления нежелательных классов.

Например, из игры, в которой классы актеров определены в Python, и вы хотите избежать других общих классов, которые будут доступны при вводе пользователем.

Другой подход (как в примере ниже) состоит в создании целого нового класса, который содержит dictвышеупомянутое. Это будет:

• Разрешить создание нескольких владельцев классов для упрощения организации (например, один для актерских классов и другой для типов звука);
• Вносите изменения как в владельца, так и в классы, которые легче проводить;
• И вы можете использовать методы класса, чтобы добавить классы к dict. (Хотя приведенная ниже абстракция на самом деле не нужна, она просто для ... "иллюстрации" ).

Пример:

class ClassHolder(object):
    def __init__(self):
        self.classes = {}

    def add_class(self, c):
        self.classes[c.__name__] = c

    def __getitem__(self, n):
        return self.classes[n]

class Foo(object):
    def __init__(self):
        self.a = 0

    def bar(self):
        return self.a + 1

class Spam(Foo):
    def __init__(self):
        self.a = 2

    def bar(self):
        return self.a + 4

class SomethingDifferent(object):
    def __init__(self):
        self.a = "Hello"

    def add_world(self):
        self.a += " World"

    def add_word(self, w):
        self.a += " " + w

    def finish(self):
        self.a += "!"
        return self.a

aclasses = ClassHolder()
dclasses = ClassHolder()
aclasses.add_class(Foo)
aclasses.add_class(Spam)
dclasses.add_class(SomethingDifferent)

print aclasses
print dclasses

print "======="
print "o"
print aclasses["Foo"]
print aclasses["Spam"]
print "o"
print dclasses["SomethingDifferent"]

print "======="
g = dclasses["SomethingDifferent"]()
g.add_world()
print g.finish()

print "======="
s = []
s.append(aclasses["Foo"]())
s.append(aclasses["Spam"]())

for a in s:
    print a.a
    print a.bar()
    print "--"

print "Done experiment!"

Это возвращает меня:

<__main__.ClassHolder object at 0x02D9EEF0>
<__main__.ClassHolder object at 0x02D9EF30>
=======
o
<class '__main__.Foo'>
<class '__main__.Spam'>
o
<class '__main__.SomethingDifferent'>
=======
Hello World!
=======
0
1
--
2
6
--
Done experiment!

Еще один забавный эксперимент с ними - добавить метод, который перебирает все, ClassHolderчтобы вы никогда не потеряли все классы, которые вы сделали: ^)