从其字符串名称创建(实例化?)对内置类型的引用

时间:2013-06-05 20:53:45

标签: python-2.7 pyyaml voluptuous

我希望能够在yaml中定义架构,使用pyyaml读取它,然后使用voluptuous(或其他架构验证器!)进行验证。但是,正如问题标题中所述,我需要将内置类str实例化为丰富而不是字符串表示。

from voluptuous import Schema
import yaml


y = '''
a: str
b: int
c:
  d: float
  e: str
'''

yaml_schema = yaml.load(y,
                        Loader=yaml.CLoader)

schema1 = Schema(yaml_schema, required=True)

但是,此架构现在正在寻找字符串str作为a唯一可接受的值。使用直接pyyaml(例如'a':!! python / int)失败。相反,我想要下面的架构:

schema2 = Schema({'a': str,
                 'b': int,
                 'c': {'d': float,
                       'e': str}},
                required=True)

我很清楚eval不是生产解决方案,但下面的evaler函数会将schema1转换为schema2 ...

def evaler(d):
    out = {}
    for k, v in d.items():
        if isinstance(v, dict):
            out[k] = evaler(v)
        else:
            out[k] = eval(v)
    return out

## Tests:

## passing
v.Schema(evaler(yaml_schema), 
         required=True)({'a': 'foo', 
                         'b': 2, 
                         'c': {'d': 2.0,
                               'e': 'bar'}})

## failling
v.Schema(evaler(yaml_schema), 
         required=True)({'a': 3, 
                         'b': 2, 
                         'c': {'d': 2.0,
                               'e': 1}})

我也知道你可以实例化一个空类:

class foo: pass
globals()['foo']

但是使用内置功能这是不可能的:

globals()['int']
# KeyError: 'int'

我探索了newtype模块,但没有运气......

1 个答案:

答案 0 :(得分:1)

最安全,最简单,最清晰的解决方案是明确列出您关注的类型的映射:

TYPES = {
    'str': str,
    'int': int,
    ...
}

您可以通过从类型列表创建此dict来消除重复(以牺牲一些灵活性为代价):

TYPES = {cls.__name__: cls for cls in [str, int, ...]}

然后,您可以递归地遍历文档(就像在evaler中一样),并用TYPES[s]替换每个字符串。 如果您坚持按名称支持所有内置类型,而不单独列出,则可以使用builtins模块(在Python 2中称为__builtin__)。 getattr是你的朋友。您应该检查它是否是一种类型 - 有许多内置名称不是。

无论如何,你需要走完文件。从PyYAML的角度来看,用作映射值的字符串“str”与用作映射键的字符串“a”具有相同的标记,因此您无法通过为该标记指定不同的类来执行任何操作。虽然你可能会深入研究它的内容并引入一种不同处理标量映射值的hack,但这只是:一个hack。还有大量的额外工作要启动。不值得。