python - pylint、协程、装饰器和类型推断

Question

我正在做一个 Google AppEngine 项目，我最近将我的 pylint 版本升级到:

No config file found, using default configuration
pylint 1.5.6, 
astroid 1.4.6
Python 2.7.10 (default, Oct 23 2015, 19:19:21)

这似乎破坏了一些类型推断。具体来说，GAE 的 ndb uses a decorator and a generator function to return a "Future" object像这样:

@ndb.tasklet
def coroutine_like(item_id):
    # do something here...
    item = yield EntityType.get_by_id_async(item_id)
    raise ndb.Return(item)

我可能会这样调用它:

future = coroutine_like('12345')
# Do other stuff
entity = future.get_result()

以前，我对这里的 linter 没有任何问题。现在我得到:

E: 42,17: Generator 'generator' has no 'get_result' member (no-member)
E: 48,17: Generator 'generator' has no 'get_result' member (no-member)
E: 60,25: Generator 'generator' has no 'get_result' member (no-member)
E: 74, 8: Generator 'generator' has no 'wait' member (no-member)
E: 88, 8: Generator 'generator' has no 'wait' member (no-member)
E: 95,17: Generator 'generator' has no 'get_result' member (no-member)

我意识到我可以 # pylint: disable=no-member 这些行单独但是那会很麻烦。我还意识到我可以通过在模块级别添加抑制代码来抑制模块级别的警告，并且可以通过修改我的 pylintrc 文件来全局抑制警告。我真的不想做那些事。我更愿意(以某种方式)告诉 pylint 用 @ndb.tasklet 装饰器装饰的东西返回 ndb.Future 实例。我看到有 ways to register type-inferencing helpers ¹ 用于 pylint，但我不确定如何让它们与我的生成器函数装饰器一起工作。

^{1请注意，这是一篇很老的博文...我认为 logilab.astng 已不再使用，现在您可以使用 astroid，但这并没有让我太接近我正在寻找的答案...}

Answer 1

那篇博文肯定很老了，现在已经发生了一段时间的变化。

您可以看看 astroid 的大脑模块是如何实现的(https://github.com/PyCQA/astroid/tree/master/astroid/brain)。它们通常是 AST 转换器，应用于特定的 AST，提供修改以便 pylint 了解您的代码到底发生了什么。

转换通常是一个函数，它接收一个节点并应该返回一个新节点或修改后的相同节点(请注意，尽管将来我们将删除对修改相同节点的支持，它们将变得不可变)

你可以通过以下方式注册一个

astroid.MANAGER.register_transform(type_of_node, transform_function)

但通常可以为 register_transform 提供一个过滤器，这样它就只应用于您感兴趣的特定节点。过滤器是 register_transform 的第三个参数，它是一个接收节点并返回一个节点的函数 bool 值，如果应该转换节点则为 true，否则为 false。您还可以将此转换作为推理提示，通过将第二个参数包装在 astroid.inference_tip(...) 中来代替正常的推理机制。这可能是您想要的，因为您希望帮助 pylint 正确推断此函数，而不是向 AST 本身添加构造。在这种特殊情况下，转换可以返回 ndb.Return 的实例，使用您在函数中拥有的屈服点进行初始化。另外，请注意，您可以从字符串构建 AST，仅使用代码表示，如:

ast = astroid.parse('''...'''
return ast

但是如果你想要更细粒度的方法，你可以自己构建 AST(粗略的例子):

from astroid import MANAGER
module = MANAGER.ast_from_module_name('ndb')
cls = next(module.igetattr('Return'))
instance = cls.instantiate_class()
node = astroid.Return(...)
node.value = ... node
return node

另外，请注意，虽然创建新节点会随着最新版本而改变，但通过使用正确的构造函数方法来构建它们，而不是手动添加属性。

希望这对您有所帮助。