sqlmesh.core.config.linter

 1from __future__ import annotations
 2
 3import typing as t
 4
 5from sqlglot import exp
 6from sqlglot.helper import ensure_collection
 7
 8from sqlmesh.core.config.base import BaseConfig
 9
10from sqlmesh.utils.pydantic import field_validator
11
12
13class LinterConfig(BaseConfig):
14    """Configuration for model linting
15
16    Args:
17        enabled: Flag indicating whether the linter should run
18
19        rules: A list of error rules to be applied on model
20        warn_rules: A list of rules to be applied on models but produce warnings instead of raising errors.
21        ignored_rules: A list of rules to be excluded/ignored
22
23    """
24
25    enabled: bool = False
26
27    rules: t.Set[str] = set()
28    warn_rules: t.Set[str] = set()
29    ignored_rules: t.Set[str] = set()
30
31    @classmethod
32    def _validate_rules(cls, v: t.Any) -> t.Set[str]:
33        if isinstance(v, exp.Paren):
34            v = v.unnest().name
35        elif isinstance(v, (exp.Tuple, exp.Array)):
36            v = [e.name for e in v.expressions]
37        elif isinstance(v, exp.Expr):
38            v = v.name
39
40        return {name.lower() for name in ensure_collection(v)}
41
42    @field_validator("rules", "warn_rules", "ignored_rules", mode="before")
43    def rules_validator(cls, vs: t.Any) -> t.Set[str]:
44        return cls._validate_rules(vs)