Python中的动态类

Python中的动态类

---

有这样一个需求，我有SegmentReader、PostagReader、ConllReader这三个Reader，他们都继承于一个Reader类。在程序运行中，由用户通过segment、postag或conll决定读入哪一种数据，所以XReader的实例化也由用户决定。当然，我们可以写一连串的if判断，但是这往往不够美观。这里有一个python动态创建类的问题。

Python中有个一函数globals()返回当前全局的符号表，如果已经import了这三个类，他们就会出现在全局的符号表中。

from corpusproc.io import SegmentReader, PostagReader, ConllReader

def createObject(className, * args):
    cls = globals()[className]

    if isinstance(cls, type) and isinstance(args[0], file):
        return cls(args[0])
    else:
        raise Exception("No such class")

测试一下

format = "segment"
reader = createObject(format[0].upper() + format[1:])

format = "unknown"
reader = createObject(format[0].upper() + format[1:])
# Raise Exception "No such class"