子类化模板以提供默认参数

时间:2019-04-24 09:40:51

标签: python python-3.x templates subclass

我正在从Template继承string来为其提供一些额外的默认功能。这个想法是为了使其查找范围超出传递的字典,首先扩展到locals(),然后扩展到globals(),最后扩展到默认值(例如,扩展到'-')。这就是我写的:

class MyTemplate(Template):
    def substitute_default(*args, **kws):
        if not args:
            raise TypeError("descriptor 'substitute' of 'Template' object needs an argument")
        self, *args = args  # allow the "self" keyword be passed
        if len(args) > 1:
            raise TypeError('Too many positional arguments')
        if not args:
            mapping = kws
        elif kws:
            mapping = ChainMap(kws, args[0])
        else:
            mapping = args[0]

        def convert(mo):
            named = mo.group('named') or mo.group('braced')
            if named is not None:
                val = mapping.get(named, locals().get(named, globals().get(named, '-')))
                return '%s' % (val,)
            if mo.group('escaped') is not None:
                return self.delimiter
            if mo.group('invalid') is not None:
                self._invalid(mo)
            raise ValueError('Unrecognized named group in pattern', self.pattern)

        return self.pattern.sub(convert, self.template)

果汁所在的行是这样的:

val = mapping.get(named, locals().get(named, globals().get(named, '-')))

我正在像这样测试它:

a = 'global_foo'

def f():
    b = 'local_foo'
    t = MyTemplate('''a=$a, b=$b, c=$c, d=$d''')
    text = t.substitute_default({'c': 'foo', 'd': 'bar'})
    print(text)

f()  # -> a=global_foo, b=-, c=foo, d=bar

如您所见,globals()查找有效,而locals()查找无效。

  • 有人对为什么会这样有想法吗?
  • 有更好的方法吗?

1 个答案:

答案 0 :(得分:1)

问题是locals()在您的convert函数中是本地的,当您希望它引用f的本地人时。

您必须以某种方式通过locals()字典,无论是在构造函数中还是在其他地方,它都可以正常工作。