Enforce Best Practices for all your Airflow DAGs. ⭐

• Use function-level imports instead of top-level imports¹² (see Top level Python Code)
• Use jinja macro instead of Variable.get (see Airflow Variables)

based on official Best Practices

To install it from PyPI run:

pip install airflint

Then just call it like this:

Alternatively you can add the following repo to your pre-commit-config.yaml:

  - repo: https://github.com/feluelle/airflint
    rev: v0.2.0-alpha
    hooks:
      - id: airflint
        args: ["-a"]  # Use -a for replacing inplace

To complete the UseFunctionlevelImports rule, please add the autoflake hook after the airflint hook, as below:

  - repo: https://github.com/pycqa/autoflake
    rev: v1.4
    hooks:
      - id: autoflake
        args: ["--remove-all-unused-imports", "--in-place"]

This will remove unused imports.