visual-studio-code - VSCode 中 Airflow 插件的 Intellisense 支持

Question

我正在尝试找到一种方法让 VSCode Python intellisense 与 Airflow Plugins 一起工作.

在 code example 之后插件操作符的导入路径可以是:

from airflow.operators import MyPluginOperator

VSCode 无法解析这个导入，因为它只会在运行时通过 airflow 插件系统有效。

有什么方法可以配置 VSCode 来解决这个导入问题吗？

Answer 1

Airflow 通过在 airflow/plugins 文件夹中搜索 AirflowPlugin 子类动态加载插件，并在运行时将它们添加到 airflow 命名空间中。以下是来自 airflow/operators/__init__.py 的代码:

# Imports operators dynamically while keeping the package API clean,
# abstracting the underlying modules
...
def _integrate_plugins():
    """Integrate plugins to the context"""
    from airflow.plugins_manager import operators_modules
    for operators_module in operators_modules:
        sys.modules[operators_module.__name__] = operators_module
        globals()[operators_module._name] = operators_module

VS Code 无法处理它。甚至像 PyCharm 这样的“大型”Python IDE has problems with it . VS Code 不可能知道特定文件夹中的一段代码稍后会在 airflow.operator 中转换。 “python.autoComplete.extraPaths”也无济于事。你应该只希望有人会在某处为 Airflow 编写 VS Code 扩展:)